Generalization, memorization, and overfitting for diffusion models trained in the lazy high-dimensional regime
On the map: Diffusion models → Diffusion foundations and sampling
Abstract (arXiv)
Modern score-based generative models have achieved remarkable empirical success in high-dimensional tasks such as image, audio, and video synthesis. These models reduce distribution learning to a sequence of regression problems that, if solved exactly on finite data, would ultimately reproduce the training samples. Their ability to generalize must therefore arise from the implicit or explicit regularization during training. In this work, we develop a generative counterpart to the theory of benign overfitting and algorithmic regularization for overparameterized neural networks in the supervised lazy-training regime. We study denoising score matching in a vector-valued reproducing kernel Hilbert space with an inner-product kernel. In the proportional high-dimensional regime nasymp d, we derive exact risk trajectories under gradient flow training. These trajectories exhibit three phases governed by qualitatively distinct estimators: a spectral estimator that generalizes, a pure-noise score with localized peaks that interpolate the training objective, and an empirical Bayes estimator that memorizes the data. We then analyze how these estimators combine along the reverse-time SDE and characterize the distribution of the resulting samples. The analysis reveals familiar mechanisms from supervised learning, including kernel linearization and self-induced regularization from the nonlinear part of the kernel, but also reveals a distinct phenomenology specific to generative modeling.
Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 13 of its 111 references that are in the map. First posted on arXiv 25 Aug 2026.
Builds on
13 of 111 references.
Closest ancestors · 9
Same direction, newest first
- 2026Benign Overfitting Does Not Occur in Diffusion Models2 Jul
- 2026Local Coverage Governs Memorization in Diffusion Models12 Jun
- 2026Diffusion Models Memorize in Training -- and Generalize in Inference12 Mar
- 2026A Random Matrix Theory Perspective on the Consistency of Diffusion Models2 Feb
- 2025Bigger Isn't Always Memorizing: Early Stopping Overparameterized Diffusion Models22 May
- 2025Memorization and Regularization in Generative Diffusion Models27 Jan
- 2024Losing dimensions: Geometric memorization in generative diffusion11 Oct
- 2024From optimal score matching to optimal sampling11 Sep
- 2023The Hidden Linear Structure in Score-Based Models and its Application17 Nov
Other references · 4
- 2024An analytic theory of creativity in convolutional diffusion modelscited by 103
- 2024A non-asymptotic theory of Kernel Ridge Regression: deterministic equivalents, test error, and GCV estimatorcited by 22
- 2022Quantitative deterministic equivalent of sample covariance matrices with a general dependence structurecited by 9
- 2026Anisotropic local law for non-separable sample covariance matricescited by 1
Built on by
Citing papers per year: 2026: 3.
3 papers.
Similar, not cited
Close in wording, no citation either way
- 2025Taking a Big Step: Large Learning Rates in Denoising Score Matching Prevent Memorization
- 2025Denoising Score Matching with Random Features: Insights on Diffusion Models from Precise Learning Curves
- 2021Heavy-tailed denoising score matching
- 2021A Variational Perspective on Diffusion-Based Generative Models and Score Matching
- 2025Implicit Regularisation in Diffusion Models: An Algorithm-Dependent Generalisation Analysis
- 2024Neural Network-Based Score Estimation in Diffusion Models: Optimization and Generalization
- 2024Nonlinear denoising score matching for enhanced learning of structured distributions
- 2026Generalization Properties of Score-matching Diffusion Models for Intrinsically Low-dimensional Data