English

Beckmann Transport Models: From Autonomous Flows to One-Step Maps

Machine Learning 2026-08-03 v1

Abstract

We propose an instantiation of flow matching that relies on a time-independent velocity field (an \emph{autonomous flow}) to exactly map between two distributions, so long as the target is singular, i.e.\ supported on a lower-dimensional data manifold. We also show that the one-step generative map associated with this flow is the unique solution of a simple conservation equation, which can be used to learn the map directly from samples. These autonomous flows and maps give a dynamical meaning to the flux constraint of Beckmann's transportation problem. Their construction provides a unifying framework that recovers, for instance, the closed-form Poisson-flow generative model and equilibrium matching with a quadratic flow-matching regression loss. We illustrate how this theory corrects inconsistencies in existing methods and demonstrate the effectiveness of the autonomous flow and the one-step map on ImageNet 256x256.

Cite

@article{arxiv.2608.01692,
  title  = {Beckmann Transport Models: From Autonomous Flows to One-Step Maps},
  author = {Lee Cheuk-Kit and Florentin Coeurdoux and Peter Potaptchik and Yilun Du and Michael Samuel Albergo and Eric Vanden-Eijnden},
  journal= {arXiv preprint arXiv:2608.01692},
  year   = {2026}
}