Fiberwise Optimal Transport Schedules Cut Flow Matching FID by 38.6%

MaxUnfried · x · 2026-09-15

A new preprint introduces model-aware diffusion/flow-matching schedules based on fiberwise optimal transport: compatible signal/noise decompositions form affine 'fibers,' and a fiberwise prediction risk combined with a kinetic action yields a closed-form optimal time allocation, estimable from an early checkpoint.

Evaluated across DDPMs and flow matching over prediction targets, datasets, and architectures, the schedules consistently beat strong baselines — including a 38.6% relative FID reduction for flow matching on CIFAR-10 at 16 function evaluations. Risk profiles from independently trained models align closely in kinetic reference coordinates.

Original post →

More from Multimodal

Multimodal channel →