🔗: github.com/khaledkah/tv...
📄: www.arxiv.org/abs/2502.08598
Thanks to Khaled, Winnie, Oliver, Klaus, and Shin for the cool collaboration as well as @bifold.berlin, TU Berlin, RIKEN, and DeepMind
🔗: github.com/khaledkah/tv...
📄: www.arxiv.org/abs/2502.08598
Thanks to Khaled, Winnie, Oliver, Klaus, and Shin for the cool collaboration as well as @bifold.berlin, TU Berlin, RIKEN, and DeepMind
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)
No fancy time grids like in EDM needed! VP-ISSNR on CIFAR-10/FFHQ ≈ EDM but with fewer hyperparameters!
No fancy time grids like in EDM needed! VP-ISSNR on CIFAR-10/FFHQ ≈ EDM but with fewer hyperparameters!
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!