mlsanityData to 7 Oct 2026

On the Relation between Rectified Flows and Optimal Transport

arXiv 2505.19712 ↗First posted 26 May 2025cs.LG math.PR stat.MLcited by 15

On the map: Diffusion models → Flow matching, time series and science

Abstract (arXiv)

This paper investigates the connections between rectified flows, flow matching, and optimal transport. Flow matching is a recent approach to learning generative models by estimating velocity fields that guide transformations from a source to a target distribution. Rectified flow matching aims to straighten the learned transport paths, yielding more direct flows between distributions. Our first contribution is a set of invariance properties of rectified flows and explicit velocity fields. In addition, we also provide explicit constructions and analysis in the Gaussian (not necessarily independent) and Gaussian mixture settings and study the relation to optimal transport. Our second contribution addresses recent claims suggesting that rectified flows, when constrained such that the learned velocity field is a gradient, can yield (asymptotically) solutions to optimal transport problems. We study the existence of solutions for this problem and demonstrate that they only relate to optimal transport under assumptions that are significantly stronger than those previously acknowledged. In particular, we present several counterexamples that invalidate earlier equivalence results in the literature, and we argue that enforcing a gradient constraint on rectified flows is, in general, not a reliable method for computing optimal transport maps.

Key facts. Cited by 15 papers in the map (data through 7 Oct 2026). Placed in Flow matching, time series and science, one of 14 directions in Diffusion models. Builds on 40 of its 55 references that are in the map. First posted on arXiv 26 May 2025.

Builds on

40 of 55 references.

Closest ancestors · 12

Same direction, newest first

  1. 2025Flow Matching: Markov Kernels, Stochastic Processes and Transport Plans28 Jan
  2. 2024Flow Matching Guide and Code9 Dec
  3. 2024On the Convergence and Straightness of Rectified Flow19 Oct
  4. 2024Improving the Training of Rectified Flows30 May
  5. 2024Conditional Wasserstein Distances with Applications in Bayesian OT Flow Matching27 Mar
  6. 2024Optimal Flow Matching: Learning Straight Trajectories in Just One Step19 Mar
  7. 2023InstaFlow: One Step is Enough for High-Quality Diffusion-Based Text-to-Image Generation12 Sep
  8. 2023Multisample Flow Matching: Straightening Flows with Minibatch Couplings28 Apr
  9. 2023Improving and generalizing flow-based generative models with minibatch optimal transport1 Feb
  10. 2022Flow Matching for Generative Modeling6 Oct
  11. 2022Rectified Flow: A Marginal Preserving Approach to Optimal Transport29 Sep
  12. 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flow7 Sep

Other references · 28

  1. 2020Denoising Diffusion Probabilistic Modelscited by 20,851
  2. 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
  3. 2018Neural Ordinary Differential Equationscited by 3,733
  4. 2013Sinkhorn Distances: Lightspeed Computation of Optimal Transportation Distancescited by 2,277
  5. 2022Building Normalizing Flows with Stochastic Interpolantscited by 964
  6. 2023Stochastic Interpolants: A Unifying Framework for Flows and Diffusionscited by 897
  7. 2021Diffusion Schr"odinger Bridge with Applications to Score-Based Generative Modelingcited by 594
  8. 2018FFJORD: Free-form Continuous Dynamics for Scalable Reversible Generative Modelscited by 584
  9. 2015Optimal Transport for Domain Adaptationcited by 457
  10. 2013A survey of the Schr"odinger problem and some of its connections with optimal transportcited by 286
  11. 2020Stochastic Normalizing Flowscited by 156
  12. 2023Diffusion Schr"odinger Bridge Matchingcited by 151
16 more
  1. 2020OT-Flow: Fast and Accurate Continuous Normalizing Flows via Optimal Transportcited by 143
  2. 2021Solving Schr"odinger Bridges via Maximum Likelihoodcited by 140
  3. 2022Neural Optimal Transportcited by 123
  4. 2024PnP-Flow: Plug-and-Play Image Restoration with Flow Matchingcited by 92
  5. 2020Convex Potential Flows: Universal Probability Distributions with Optimal Transport and Convex Optimizationcited by 89
  6. 2023Error Bounds for Flow Matching Methodscited by 88
  7. 2023Minimizing Trajectory Curvature of ODE-based Generative Modelscited by 87
  8. 2023Diffusion Bridge Mixture Transports, Schr"odinger Bridge Problems and Generative Modelingcited by 80
  9. 2017Multilevel Clustering via Wasserstein Meanscited by 48
  10. 2024Schr"odinger Bridge Flow for Unpaired Data Translationcited by 47
  11. 2022Normalizing flow neural networks by JKO schemecited by 39
  12. 2021Stochastic Normalizing Flows for Inverse Problems: a Markov Chains Viewpointcited by 32
  13. 2022Wasserstein Iterative Networks for Barycenter Estimationcited by 23
  14. 2018Bayesian Learning with Wasserstein Barycenterscited by 12
  15. 2018Variational Wasserstein Clusteringcited by 11
  16. 2024Importance Corrected Neural JKO Samplingcited by 7

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2025: 6; 2026: 9.

15 papers.

Sort:

By area:

Similar, not cited

Close in wording, no citation either way

  1. 2026Order-Optimal Sample Complexity of Rectified Flows
  2. 2026Beckmann Transport Models: From Autonomous Flows to One-Step Maps
  3. 2025On the Equivalence of Optimal Transport Problem and Action Matching with Optimal Vector Fields
  4. 2025Generative Modeling with Continuous Flows: Sample Complexity of Flow Matching
  5. 2026The Fundamental Limits of Valid Transport Map Estimation
  6. 2026Path-independent Flow Matching for Multi-parameter Generative Dynamics
  7. 2025Theoretical Guarantees for High Order Trajectory Refinement in Generative Flows
  8. 2026VTV-FM: Flow Matching through Variational Terminal-Velocity Closure