On the Memorization of Consistency Distillation for Diffusion Models
On the map: Diffusion models → Diffusion foundations and sampling
Abstract (arXiv)
Diffusion models are central to modern generative modeling, and understanding how they balance memorization and generalization is critical for reliable deployment. Recent work has shown that memorization in diffusion models is shaped by training dynamics, with generalization and memorization emerging at different stages of training. However, deployed diffusion models are often further distilled, introducing an additional training phase whose impact on memorization is not well understood. In this work, we analyze how distillation reshapes memorization behavior in diffusion models, taking consistency distillation as a representative framework. Empirically, we show that when applied to a teacher model that has memorized data, consistency distillation significantly reduces transferred memorization in the student while preserving, and sometimes improving, sample quality. To explain this behavior, we provide a theoretical analysis using a random feature neural network model [Bonnaire et al., 2025], showing that consistency distillation suppresses unstable feature directions associated with memorization while preserving stable, generalizable modes. Our findings suggest that distillation can serve not only as an acceleration tool, but also as a mechanism for improving the memorization-generalization trade-off.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 43 of its 52 references that are in the map. First posted on arXiv 26 Apr 2026.
Builds on
43 of 52 references.
Closest ancestors · 38
Same direction, newest first
- 2025Generalization of Diffusion Models Arises with a Balanced Representation Space24 Dec
- 2025On the Edge of Memorization in Diffusion Models25 Aug
- 2025When Diffusion Models Memorize: Inductive Biases in Probability Flow of Minimum-Norm Shallow Neural Nets23 Jun
- 2025Memorization to Generalization: Emergence of Diffusion Models from Associative Memory27 May
- 2025Why Diffusion Models Don't Memorize: The Role of Implicit Dynamical Regularization in Training23 May
- 2025Convergence Of Consistency Model With Multistep Sampling Under General Data Assumptions6 May
- 2025Scale-wise Distillation of Diffusion Models20 Mar
- 2025Can Diffusion Models Learn Hidden Inter-Feature Rules Behind Images?7 Feb
- 2025Denoising Score Matching with Random Features: Insights on Diffusion Models from Precise Learning Curves1 Feb
- 2025Memorization and Regularization in Generative Diffusion Models27 Jan
- 2024Inference-Time Diffusion Model Distillation12 Dec
- 2024On the Feature Learning in Diffusion Models2 Dec
26 more
- 2024Stable Consistency Tuning: Understanding and Improving Consistency Models24 Oct
- 2024One-Step Diffusion Distillation through Score Implicit Matching22 Oct
- 2024DKDM: Data-Free Knowledge Distillation for Diffusion Models with Any Architecture5 Sep
- 2024Detecting, Explaining, and Mitigating Memorization in Diffusion Models31 Jul
- 2024Consistency Models Made Easy20 Jun
- 2024Phased Consistency Models28 May
- 2024Improved Distribution Matching Distillation for Fast Image Synthesis23 May
- 2024Towards a mathematical theory for consistency training in diffusion models12 Feb
- 2023One-Step Diffusion Distillation via Deep Equilibrium Models12 Dec
- 2023One-step Diffusion with Distribution Matching Distillation30 Nov
- 2023On the Generalization Properties of Diffusion Models3 Nov
- 2023Latent Consistency Models: Synthesizing High-Resolution Images with Few-Step Inference6 Oct
- 2023On Memorization in Diffusion Models4 Oct
- 2023Consistency Trajectory Models: Learning Probability Flow ODE Trajectory of Diffusion1 Oct
- 2023SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesis4 Jul
- 2023On the Equivalence of Consistency-Type Models: Consistency Models, Consistent Diffusion Models, and Fokker-Planck Regularization1 Jun
- 2023Understanding and Mitigating Copying in Diffusion Models31 May
- 2023Diff-Instruct: A Universal Approach for Transferring Knowledge From Pre-trained Diffusion Models29 May
- 2023Consistency Models2 Mar
- 2022Elucidating the Design Space of Diffusion-Based Generative Models1 Jun
- 2021High-Resolution Image Synthesis with Latent Diffusion Models20 Dec
- 2021Diffusion Models Beat GANs on Image Synthesis11 May
- 2020Score-Based Generative Modeling through Stochastic Differential Equations26 Nov
- 2020Denoising Diffusion Implicit Models6 Oct
- 2020Denoising Diffusion Probabilistic Models19 Jun
- 2020Improved Techniques for Training Score-Based Generative Models16 Jun
Other references · 5
- 2018The Unreasonable Effectiveness of Deep Features as a Perceptual Metriccited by 11,029
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,067
- 2023Extracting Training Data from Diffusion Modelscited by 723
- 2024Physics Informed Distillation for Diffusion Modelscited by 13
- 2024SIDE: Surrogate Conditional Data Extraction from Diffusion Modelscited by 7
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025Provable Separations between Memorization and Generalization in Diffusion Models
- 2026Local Coverage Governs Memorization in Diffusion Models
- 2026Understanding diffusion models requires rethinking (again) generalization
- 2025Bigger Isn't Always Memorizing: Early Stopping Overparameterized Diffusion Models
- 2026Diffusion Models Preferentially Memorize Prototypical Examples or: Why Does My Diffusion Model Love Slop?
- 2025How Diffusion Models Memorize
- 2025Does Generation Require Memorization? Creative Diffusion Models using Ambient Diffusion
- 2026Characterizing Memorization in Diffusion Language Models: Generalized Extraction and Sampling Effects