Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
The authors investigate the temporal decoupling between generalization and memorization in diffusion models by analyzing the eigenspectrum of the score-matching Neural Tangent Kernel (NTK) Gram matrix under multiple noise realizations. They prove analytically in the high-dimensional lazy limit that training on repeated noisy copies splits the spectrum into a dominant bulk encoding global distribution features and a low-lying bulk aligned with sample-specific noise that parametrically delays memorization. Crucially, this two-bulk spectral bias extends to practical, finite-width U-Nets, allowing targeted regularization of the second spectral bulk to suppress training set memorization without degrading generative fidelity.
Multi-noise score matching mathematically splits the neural tangent kernel into two distinct spectral bulks, explaining why diffusion models master data distributions long before collapsing into training set memorization.
Diffusion models trained on a finite dataset first learn to generate novel, high-quality samples and only much later collapse onto their training set. We identify the mechanism behind this separation of timescales and the object that probes it. The training dynamics of the score function are governed---exactly, and at any width---by the Gram matrix of the Neural Tangent Kernel (NTK) evaluated on the noisy training data, so the timescales of generalization and of memorization must be encoded in its spectrum. We show that they are, and that the structure responsible has no analogue in standard kernel settings. The use of multiple noise realizations per sample ($m$ noised copies at a fixed noise level) in the score-matching loss is what restructures the Gram matrix spectrum into two distinct parts. The first, of large eigenvalues, carries the global features of the target distribution and is present already for $m=1$. The second, which the repeated noising creates, consists of the smallest eigenvalues and is supported on eigenvectors aligned with the sample-specific noise directions; it sets a memorization timescale parametrically larger in the training set size $n$. We establish this picture on two fronts. Analytically, we solve the spectrum in the lazy high-dimensional limit for both linear ($n \asymp d$) and polynomial ($n \asymp d^k$) sample complexities, and prove through a bias--variance decomposition that the first bulk minimizes the approximation error while the second drives the error associated with memorization. Empirically, we show the same two-bulk structure in Convolutional NTKs on CelebA and in finite-width U-Nets trained well beyond the lazy regime, and we make the link causal: truncating the Gram matrix at rank $r$ tunes the generalization--memorization transition, and an $L_2$ penalty targeting the second bulk suppresses memorization in feature-learning U-Nets.