Model-Aware Schedules Improve Generation via Fiberwise Optimal Transport

Luyi Jia, Boyan Zhang, Yilun Liu +1

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 coeff…