papersSEP 12 04:00 UTC
Model-Aware Diffusion Schedules Derived via Optimal Transport
A new arXiv paper argues that the schedules controlling how signal and noise are mixed along diffusion and flow-matching paths can be optimized by minimizing a kinetic action borrowed from optimal transport theory. The authors show that making these schedules depend on the specific model, rather than using fixed hand-tuned coefficients, improves generation quality. The work offers a theoretical framing for why certain noise schedules perform better than others.