Denoising Score Matching with Random Features: Insights on Diffusion Models from Precise Learning Curves

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: George, Anand Jerry, Veiga, Rodrigo, Macris, Nicolas
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912634004045824
author George, Anand Jerry
Veiga, Rodrigo
Macris, Nicolas
author_facet George, Anand Jerry
Veiga, Rodrigo
Macris, Nicolas
contents We theoretically investigate the phenomena of generalization and memorization in diffusion models. Empirical studies suggest that these phenomena are influenced by model complexity and the size of the training dataset. In our experiments, we further observe that the number of noise samples per data sample ($m$) used during Denoising Score Matching (DSM) plays a significant and non-trivial role. We capture these behaviors and shed insights into their mechanisms by deriving asymptotically precise expressions for test and train errors of DSM under a simple theoretical setting. The score function is parameterized by random features neural networks, with the target distribution being $d$-dimensional Gaussian. We operate in a regime where the dimension $d$, number of data samples $n$, and number of features $p$ tend to infinity while keeping the ratios $ψ_n=\frac{n}{d}$ and $ψ_p=\frac{p}{d}$ fixed. By characterizing the test and train errors, we identify regimes of generalization and memorization as a function of $ψ_n,ψ_p$, and $m$. Our theoretical findings are consistent with the empirical observations.
format Preprint
id arxiv_https___arxiv_org_abs_2502_00336
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Denoising Score Matching with Random Features: Insights on Diffusion Models from Precise Learning Curves
George, Anand Jerry
Veiga, Rodrigo
Macris, Nicolas
Machine Learning
We theoretically investigate the phenomena of generalization and memorization in diffusion models. Empirical studies suggest that these phenomena are influenced by model complexity and the size of the training dataset. In our experiments, we further observe that the number of noise samples per data sample ($m$) used during Denoising Score Matching (DSM) plays a significant and non-trivial role. We capture these behaviors and shed insights into their mechanisms by deriving asymptotically precise expressions for test and train errors of DSM under a simple theoretical setting. The score function is parameterized by random features neural networks, with the target distribution being $d$-dimensional Gaussian. We operate in a regime where the dimension $d$, number of data samples $n$, and number of features $p$ tend to infinity while keeping the ratios $ψ_n=\frac{n}{d}$ and $ψ_p=\frac{p}{d}$ fixed. By characterizing the test and train errors, we identify regimes of generalization and memorization as a function of $ψ_n,ψ_p$, and $m$. Our theoretical findings are consistent with the empirical observations.
title Denoising Score Matching with Random Features: Insights on Diffusion Models from Precise Learning Curves
topic Machine Learning
url https://arxiv.org/abs/2502.00336