Model-Aware Schedules Improve Generation via Fiberwise Optimal Transport
Diffusion and flow-matching schedules control the signal and noise coefficients that mix data and noise along affine probability paths. Minimizing a kinetic action defined on coefficient paths, motivated by optimal transport, helps explain strong baselines but remains model-agnostic and ignores prediction error. Here we introduce a model-aware schedule construction based on fiberwise optimal…
We haven't written up this one. arXiv cs.AI has the full story — the link below goes straight to it.
