Beyond Sample Copying: Structural Memorization in Diffusion Models
cs.LG
Submitted: 2026-03-12
Updated: 2026-10-08
Comments: 25 pages and 21 figures
License: http://creativecommons.org/licenses/by/4.0/
The gist: Diffusion models generalize well in practice.
Terminology
Abstract
Diffusion models generalize well in practice. Paradoxically, an optimal diffusion model fully memorizes the training data and therefore fails to generalize, raising the question of what induces generalization in a real diffusion model. We show that diffusion models progressively overfit the denoising training objective, creating a generalization gap between validation and training performance at intermediate noise levels. In a fully analytic 2D toy model with a controlled denoising error, we trace this gap to the interaction between model error and the density of the data distribution's support. The optimal denoising flow field localizes sharply around individual training points, whereas model error suppresses exact recall of training points, yielding a smooth, generalizing flow field. Finally, we examine how training-time overfitting manifests along inference trajectories. We find that predictions made from intermediate trajectory states occupy a distinct feature-space regime from predictions made from noised training and validation images. As training progresses and model size increases, these predictions develop greater relative affinity to the training data, despite the absolute similarity to the validation data not decreasing. Together, these findings show that the denoising objective and inference trajectories express structural memorization differently.
Sources
- Losing dimensions: Geometric memorization in generative diffusion
- Guiding a diffusion model using sliding windows
- Self-Rectifying Diffusion Sampling with Perturbed-Attention Guidance
- On Inductive Biases That Enable Generalization of Diffusion Transformers
- Stable Video Diffusion: Scaling Latent Video Diffusion Models to Large Datasets
- Why Diffusion Models Don't Memorize: The Role of Implicit Dynamical Regularization in Training
- On the Edge of Memorization in Diffusion Models
- Extracting Training Data from Diffusion Models
- On the Interpolation Effect of Score Smoothing in Diffusion Models
- Diffusion Models Beat GANs on Image Synthesis
- Two Calm Ends and the Wild Middle: A Geometric Picture of Memorization in Diffusion Models
- LCMem: A Universal Model for Robust Image Memorization Detection
- Diffusion Models and the Manifold Hypothesis: Log-Domain Smoothing is Geometry Adaptive
- Implicit Regularisation in Diffusion Models: An Algorithm-Dependent Generalisation Analysis
- Bigger Isn't Always Memorizing: Early Stopping Overparameterized Diffusion Models
- How Do Flow Matching Models Memorize and Generalize in Sample Data Subspaces?
- Biased Generalization in Diffusion Models
- Denoising Score Matching with Random Features: Insights on Diffusion Models from Precise Learning Curves
- On Memorization in Diffusion Models
- A solvable generative model with a linear, one-step denoiser
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks