Stabilizing Consistency Training: A Flow Map Analysis and Self-Distillation
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
- 2025Diffusion Transformers with Representation Autoencoders13 Oct
- 2025Transition Models: Rethinking the Generative Learning Objective4 Sep
- 2025Unified Continuous Generative Models12 May
- 2025Inductive Moment Matching10 Mar
- 2025VCT: Training Consistency Models with Variational Noise Coupling25 Feb
- 2025Reconstruction vs. Generation: Taming Optimization Dilemma in Latent Diffusion Models2 Jan
- 2024One Step Diffusion via Shortcut Models16 Oct
- 2024Simplifying, Stabilizing and Scaling Continuous-Time Consistency Models14 Oct
- 2024Consistency Models Made Easy20 Jun
- 2024Improving Consistency Models with Generator-Augmented Flows13 Jun
- 2024Improved Distribution Matching Distillation for Fast Image Synthesis23 May
- 2024Score identity Distillation: Exponentially Fast Distillation of Pretrained Diffusion Models for One-Step Generation5 Apr
14 more
- 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesis5 Mar
- 2023One-step Diffusion with Distribution Matching Distillation30 Nov
- 2023Improved Techniques for Training Consistency Models22 Oct
- 2023Consistency Trajectory Models: Learning Probability Flow ODE Trajectory of Diffusion1 Oct
- 2023SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesis4 Jul
- 2023Consistency Models2 Mar
- 2022Scalable Diffusion Models with Transformers19 Dec
- 2022Classifier-Free Diffusion Guidance26 Jul
- 2022Elucidating the Design Space of Diffusion-Based Generative Models1 Jun
- 2021High-Resolution Image Synthesis with Latent Diffusion Models20 Dec
- 2021Tackling the Generative Learning Trilemma with Denoising Diffusion GANs15 Dec
- 2020Score-Based Generative Modeling through Stochastic Differential Equations26 Nov
- 2020Denoising Diffusion Probabilistic Models19 Jun
- 2020Improved Techniques for Training Score-Based Generative Models16 Jun
Other references · 18
- 2022Flow Matching for Generative Modelingcited by 5,426
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,067
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,550
- 2017Progressive Growing of GANs for Improved Quality, Stability, and Variationcited by 2,967
- 2016Improved Techniques for Training GANscited by 2,938
- 2023Diffusion Policy: Visuomotor Policy Learning via Action Diffusioncited by 2,730
- 2019Root Mean Square Layer Normalizationcited by 1,425
- 2022Building Normalizing Flows with Stochastic Interpolantscited by 964
- 2025Mean Flows for One-step Generative Modelingcited by 524
- 2019PyHessian: Neural Networks Through the Lens of the Hessiancited by 291
- 2020Query-Key Normalization for Transformerscited by 266
- 2023Flow Matching on General Geometriescited by 260
6 more
- 2025Improved Mean Flows: On the Challenges of Fastforward Generative Modelscited by 102
- 2024Consistency Flow Matching: Defining Straight Flows with Velocity Consistencycited by 101
- 2024Flow map matching with stochastic interpolants: A mathematical framework for consistency modelscited by 99
- 2025How to build a consistency model: Learning flow maps via self-distillationcited by 96
- 2025Align Your Flow: Scaling Continuous-Time Flow Map Distillationcited by 81
- 2025Transition Matching: Scalable and Flexible Generative Modelingcited by 10
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2026Dual-End Consistency Model
- 2024Towards Faster Training of Diffusion Models: An Inspiration of A Consistency Phenomenon
- 2026Consistent Distribution Matching for Data-Free Diffusion Distillation
- 2024Stable Consistency Tuning: Understanding and Improving Consistency Models
- 2025CMT: Mid-Training for Efficient Learning of Consistency, Mean Flow, and Flow Map Models
- 2023Consistency Models as a Rich and Efficient Policy Class for Reinforcement Learning
- 2024Truncated Consistency Models
- 2025Beyond and Free from Diffusion: Invertible Guided Consistency Training