Scaling Online Optimal Transport Minibatches for Flow Matching
This work develops a scalable online entropic OT approximation that enables substantially larger source-target couplings while preserving the benefits of OT, and proves strict reductions in full stochastic-gradient variance and integration error relative to independent pairing.
P. Chen, Sergey Levine
· 0 citations