mlsanityData to 7 Oct 2026

Stabilizing Consistency Training: A Flow Map Analysis and Self-Distillation

arXiv 2601.22679 ↗First posted 30 Jan 2026cs.LG cs.CVcited by 2

On the map: Diffusion models → Diffusion foundations and sampling

Abstract (arXiv)

Consistency models have been proposed for fast generative modeling, achieving results competitive with diffusion and flow models. However, these methods exhibit inherent instability and limited reproducibility when training from scratch, motivating subsequent work to explain and stabilize these issues. While these efforts have provided valuable insights, the explanations remain fragmented, and the theoretical relationships remain unclear. In this work, we provide a theoretical examination of consistency models by analyzing them from a flow map-based perspective. This joint analysis clarifies how training stability and convergence behavior can give rise to degenerate solutions. Building on these insights, we revisit self-distillation as a practical remedy for certain forms of suboptimal convergence and reformulate it to avoid excessive gradient norms for stable optimization. We demonstrate that our strategy extends beyond image generation to diffusion-based policy learning, without reliance on pretrained diffusion models for initialization, illustrating its broader applicability.

Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 44 of its 49 references that are in the map. First posted on arXiv 30 Jan 2026.

Builds on

44 of 49 references.

Closest ancestors · 26

Same direction, newest first

  1. 2025Diffusion Transformers with Representation Autoencoders13 Oct
  2. 2025Transition Models: Rethinking the Generative Learning Objective4 Sep
  3. 2025Unified Continuous Generative Models12 May
  4. 2025Inductive Moment Matching10 Mar
  5. 2025VCT: Training Consistency Models with Variational Noise Coupling25 Feb
  6. 2025Reconstruction vs. Generation: Taming Optimization Dilemma in Latent Diffusion Models2 Jan
  7. 2024One Step Diffusion via Shortcut Models16 Oct
  8. 2024Simplifying, Stabilizing and Scaling Continuous-Time Consistency Models14 Oct
  9. 2024Consistency Models Made Easy20 Jun
  10. 2024Improving Consistency Models with Generator-Augmented Flows13 Jun
  11. 2024Improved Distribution Matching Distillation for Fast Image Synthesis23 May
  12. 2024Score identity Distillation: Exponentially Fast Distillation of Pretrained Diffusion Models for One-Step Generation5 Apr
14 more
  1. 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesis5 Mar
  2. 2023One-step Diffusion with Distribution Matching Distillation30 Nov
  3. 2023Improved Techniques for Training Consistency Models22 Oct
  4. 2023Consistency Trajectory Models: Learning Probability Flow ODE Trajectory of Diffusion1 Oct
  5. 2023SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesis4 Jul
  6. 2023Consistency Models2 Mar
  7. 2022Scalable Diffusion Models with Transformers19 Dec
  8. 2022Classifier-Free Diffusion Guidance26 Jul
  9. 2022Elucidating the Design Space of Diffusion-Based Generative Models1 Jun
  10. 2021High-Resolution Image Synthesis with Latent Diffusion Models20 Dec
  11. 2021Tackling the Generative Learning Trilemma with Denoising Diffusion GANs15 Dec
  12. 2020Score-Based Generative Modeling through Stochastic Differential Equations26 Nov
  13. 2020Denoising Diffusion Probabilistic Models19 Jun
  14. 2020Improved Techniques for Training Score-Based Generative Models16 Jun

Other references · 18

  1. 2022Flow Matching for Generative Modelingcited by 5,426
  2. 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,067
  3. 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,550
  4. 2017Progressive Growing of GANs for Improved Quality, Stability, and Variationcited by 2,967
  5. 2016Improved Techniques for Training GANscited by 2,938
  6. 2023Diffusion Policy: Visuomotor Policy Learning via Action Diffusioncited by 2,730
  7. 2019Root Mean Square Layer Normalizationcited by 1,425
  8. 2022Building Normalizing Flows with Stochastic Interpolantscited by 964
  9. 2025Mean Flows for One-step Generative Modelingcited by 524
  10. 2019PyHessian: Neural Networks Through the Lens of the Hessiancited by 291
  11. 2020Query-Key Normalization for Transformerscited by 266
  12. 2023Flow Matching on General Geometriescited by 260
6 more
  1. 2025Improved Mean Flows: On the Challenges of Fastforward Generative Modelscited by 102
  2. 2024Consistency Flow Matching: Defining Straight Flows with Velocity Consistencycited by 101
  3. 2024Flow map matching with stochastic interpolants: A mathematical framework for consistency modelscited by 99
  4. 2025How to build a consistency model: Learning flow maps via self-distillationcited by 96
  5. 2025Align Your Flow: Scaling Continuous-Time Flow Map Distillationcited by 81
  6. 2025Transition Matching: Scalable and Flexible Generative Modelingcited by 10

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 2.

2 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Dual-End Consistency Model
  2. 2024Towards Faster Training of Diffusion Models: An Inspiration of A Consistency Phenomenon
  3. 2026Consistent Distribution Matching for Data-Free Diffusion Distillation
  4. 2024Stable Consistency Tuning: Understanding and Improving Consistency Models
  5. 2025CMT: Mid-Training for Efficient Learning of Consistency, Mean Flow, and Flow Map Models
  6. 2023Consistency Models as a Rich and Efficient Policy Class for Reinforcement Learning
  7. 2024Truncated Consistency Models
  8. 2025Beyond and Free from Diffusion: Invertible Guided Consistency Training