1. Straight trajectories near data (t ≈ 0) are important (see in the inset plot)
2. Broad support of pₜ(𝐱) early on → robust to errors (note how SMLD goes from small to huge range instead of staying the same)
1. Straight trajectories near data (t ≈ 0) are important (see in the inset plot)
2. Broad support of pₜ(𝐱) early on → robust to errors (note how SMLD goes from small to huge range instead of staying the same)
VP-ISSNR achieves 74% stability with 8 steps, 95% with 64 (SDE). Beats all baselines!
VP-ISSNR achieves 74% stability with 8 steps, 95% with 64 (SDE). Beats all baselines!
Exponential inverse sigmoid SNR (ISSNR)→ rapid decay at start/end. Generalizes Optimal Transport Flow Matching.
Exponential inverse sigmoid SNR (ISSNR)→ rapid decay at start/end. Generalizes Optimal Transport Flow Matching.
Take SMLD/EDM (exploding TV) → force TV=1. Result: +30% stability for molecules with 8 steps
(x-axis is NFE=number of function evals).
Take SMLD/EDM (exploding TV) → force TV=1. Result: +30% stability for molecules with 8 steps
(x-axis is NFE=number of function evals).
We show constant TV (variance preserving, VP) + optimized SNR works better (ISSNR)!
(it's a wild table, sorry, but notice our VP variants I circled)
We show constant TV (variance preserving, VP) + optimized SNR works better (ISSNR)!
(it's a wild table, sorry, but notice our VP variants I circled)
🔑Insight: control Total Variance (TV) and signal-to-noise-ratio (SNR) independently!
🔑Insight: control Total Variance (TV) and signal-to-noise-ratio (SNR) independently!
Faster diffusion models with total variance/signal-to-noise ratio disentanglement! ⚡️
Our new work shows how to generate stable molecules in sometimes as little 8 steps and match EDM’s image quality with a uniform time grid. 🧵
Faster diffusion models with total variance/signal-to-noise ratio disentanglement! ⚡️
Our new work shows how to generate stable molecules in sometimes as little 8 steps and match EDM’s image quality with a uniform time grid. 🧵