mlsanityData to 14 Oct 2026

Gradient Variance Reveals Failure Modes in Flow-Based Generative Models

arXiv 2510.18118 ↗First posted 20 Oct 2025cs.LGcited by 6

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

Abstract (arXiv)

Rectified Flows learn ODE vector fields whose trajectories are straight between source and target distributions, enabling near one-step inference. We show that this straight-path objective conceals fundamental failure modes: under deterministic training, low gradient variance drives memorization of arbitrary training pairings, even when interpolant lines between pairs intersect. To analyze this mechanism, we study Gaussian-to-Gaussian transport and use the loss gradient variance across stochastic and deterministic regimes to characterize which vector fields optimization favors in each setting. We then show that, in a setting where all interpolating lines intersect, applying Rectified Flow yields the same specific pairings at inference as during training. More generally, we prove that a memorizing vector field exists even when training interpolants intersect, and that optimizing the straight-path objective converges to this ill-defined field. At inference, deterministic integration reproduces the exact training pairings. We validate our findings empirically on the CelebA dataset, confirming that deterministic interpolants induce memorization, while the injection of small noise restores generalization.

Key facts. Cited by 6 papers 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 35 of its 48 references that are in the map. First posted on arXiv 20 Oct 2025.

Builds on

35 of 48 references.

Closest ancestors · 7

Same direction, newest first

  1. 2025Carr'e du champ flow matching: better quality-generalisation tradeoff in generative models7 Oct
  2. 2024Improving the Training of Rectified Flows30 May
  3. 2022Flow Matching for Generative Modeling6 Oct
  4. 2022Rectified Flow: A Marginal Preserving Approach to Optimal Transport29 Sep
  5. 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flow7 Sep
  6. 2022Equivariant Diffusion for Molecule Generation in 3D31 Mar
  7. 2022GeoDiff: a Geometric Diffusion Model for Molecular Conformation Generation6 Mar

Other references · 28

  1. 2020Denoising Diffusion Probabilistic Modelscited by 21,076
  2. 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
  3. 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
  4. 2022Video Diffusion Modelscited by 2,305
  5. 2023Consistency Modelscited by 1,822
  6. 2020DiffWave: A Versatile Diffusion Model for Audio Synthesiscited by 1,354
  7. 2021Cascaded Diffusion Models for High Fidelity Image Generationcited by 1,163
  8. 2023Stochastic Interpolants: A Unifying Framework for Flows and Diffusionscited by 939
  9. 2022eDiff-I: Text-to-Image Diffusion Models with an Ensemble of Expert Denoiserscited by 931
  10. 2022AudioLM: a Language Modeling Approach to Audio Generationcited by 616
  11. 2023AudioLDM: Text-to-Audio Generation with Latent Diffusion Modelscited by 599
  12. 2023Consistency Trajectory Models: Learning Probability Flow ODE Trajectory of Diffusioncited by 459
16 more
  1. 2022Diffusion Art or Digital Forgery? Investigating Data Replication in Diffusion Modelscited by 424
  2. 2022MAGVIT: Masked Generative Video Transformercited by 346
  3. 2022AudioGen: Textually Guided Audio Generationcited by 295
  4. 2019Optimal transport mapping via input convex neural networkscited by 206
  5. 2023Exposing flaws of generative model evaluation metrics and their unfair treatment of diffusion modelscited by 205
  6. 2023Diffusion Schr"odinger Bridge Matchingcited by 153
  7. 2021Do Neural Optimal Transport Solvers Work? A Continuous Wasserstein-2 Benchmarkcited by 96
  8. 2020Convex Potential Flows: Universal Probability Distributions with Optimal Transport and Convex Optimizationcited by 90
  9. 2023Differentially Private Diffusion Models Generate Useful Synthetic Imagescited by 86
  10. 2023Diffusion Bridge Mixture Transports, Schr"odinger Bridge Problems and Generative Modelingcited by 80
  11. 2024Unveiling and Mitigating Memorization in Text-to-image Diffusion Models through Cross Attentioncited by 58
  12. 2024Schr"odinger Bridge Flow for Unpaired Data Translationcited by 47
  13. 2024Vidu4D: Single Generated Video to High-Fidelity 4D Reconstruction with Dynamic Gaussian Surfelscited by 38
  14. 2025On the Edge of Memorization in Diffusion Modelscited by 22
  15. 2024Frame by Familiar Frame: Understanding Replication in Video Diffusion Modelscited by 11
  16. 2024Replication in Visual Diffusion Models: A Survey and Outlookcited by 11

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2025: 1; 2026: 5.

6 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026On Variance Reduction in Learning Mean Flows
  2. 2026Universality and Convergence of Generative Flows
  3. 2026ODE-free Neural Flow Matching for One-Step Generative Modeling
  4. 2025Theoretical Guarantees for High Order Trajectory Refinement in Generative Flows
  5. 2026A Unified Measure-Theoretic View of Diffusion, Score-Based, and Flow Matching Generative Models
  6. 2026Path-independent Flow Matching for Multi-parameter Generative Dynamics
  7. 2025A Theoretical Analysis of Discrete Flow Matching Generative Models
  8. 2022A Variational Perspective on Generative Flow Networks