mlsanityData to 14 Oct 2026

Order-Optimal Sample Complexity of Rectified Flows

arXiv 2601.20250 ↗First posted 28 Jan 2026cs.LG cs.AI cs.ITcited by 1

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

Abstract (arXiv)

Recently, flow-based generative models have shown superior efficiency compared to diffusion models. In this paper, we study rectified flow models, which constrain transport trajectories to be linear from the base distribution to the data distribution. This structural restriction greatly accelerates sampling, often enabling high-quality generation with a single Euler step. Under standard assumptions on the neural network classes used to parameterize the velocity field and data distribution, we prove that rectified flows achieve sample complexity tildeO(varepsilon-2). This improves on the best known O(varepsilon-4) bounds for flow matching model and matches the optimal rate for mean estimation. Our analysis exploits the particular structure of rectified flows: because the model is trained with a squared loss along linear paths, the associated hypothesis class admits a sharply controlled localized Rademacher complexity. This yields the improved, order-optimal sample complexity and provides a theoretical explanation for the strong empirical performance of rectified flow models.

Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Placed in Flow matching, time series and science, one of 14 directions in Diffusion models. Builds on 17 of its 30 references that are in the map. First posted on arXiv 28 Jan 2026.

Builds on

17 of 30 references.

Closest ancestors · 4

Same direction, newest first

  1. 2025Generative Modeling with Continuous Flows: Sample Complexity of Flow Matching1 Dec
  2. 2024Improving the Training of Rectified Flows30 May
  3. 2022Flow Matching for Generative Modeling6 Oct
  4. 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flow7 Sep

Other references · 13

  1. 2020Denoising Diffusion Probabilistic Modelscited by 21,076
  2. 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
  3. 2018A Convergence Theory for Deep Learning via Over-Parameterizationcited by 787
  4. 2018Gradient Descent Finds Global Minima of Deep Neural Networkscited by 574
  5. 2022Sampling is as easy as learning the score: theory for diffusion models with minimal data assumptionscited by 445
  6. 2023Nearly d-Linear Convergence Bounds for Diffusion Models via Stochastic Localizationcited by 235
  7. 2020Loss landscapes and optimization in over-parameterized non-linear systems and neural networkscited by 233
  8. 2020Generative Modeling with Denoising Auto-Encoders and Langevin Samplingcited by 119
  9. 2024An Overview of Diffusion Models: Applications, Guided Generation, Statistical Rates and Optimizationcited by 73
  10. 2025Efficient Diffusion Models: A Surveycited by 35
  11. 2023Improved Sample Complexity Bounds for Diffusion Model Trainingcited by 14
  12. 2025Generative AI in Depth: A Survey of Recent Advances, Model Variants, and Real-World Applicationscited by 5
1 more
  1. 2025GeneFlow: Translation of Single-cell Gene Expression to Histopathological Images via Rectified Flowcited by 2

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 1.

1 paper.

Similar, not cited

Close in wording, no citation either way

  1. 2025On the Relation between Rectified Flows and Optimal Transport
  2. 2024On the Convergence and Straightness of Rectified Flow
  3. 2025Theoretical Guarantees for High Order Trajectory Refinement in Generative Flows
  4. 2025Flow Diverse and Efficient: Learning Momentum Flow Matching via Stochastic Velocity Field Sampling
  5. 2025Pathway to O(√d) Complexity bound under Wasserstein metric of flow-based models
  6. 2026Limit Points of Reflow with Minibatch Optimal Transport
  7. 2025Elucidating Rectified Flow with Deterministic Sampler: Polynomial Discretization Complexity for Multi and One-step Models
  8. 2024Rectified Diffusion: Straightness Is Not Your Need in Rectified Flow