Passer à l'échelle les minibatches d'optimal transport en ligne pour le flow matching

Le papier développe une approximation scalable de l'optimal transport (OT) entropique en ligne pour le flow matching, autorisant des couplages source-cible sur des minibatches nettement plus grands. Les auteurs prouvent, pour l'OT non régularisé et entropique, une réduction stricte de la variance du gradient stochastique et de l'erreur d'intégration par rapport à l'appariement indépendant, d'autant plus marquée que le minibatch grandit. Des expériences sur l'interpolation de gènes en cellule unique et l'entraînement de flows d'images confirment une meilleure stabilité et de meilleures performances de génération.