Abstract:Deep neural networks (DNNs) trained for image denoising are able to generate high-quality samples with score-based reverse diffusion algorithms. These impressive capabilities seem to imply an escape from the curse of dimensionality, but recent reports of memorization of the training set raise the question of whether these networks are learning the "true" continuous density of the data. Here, we show that two DNNs trained on non-overlapping subsets of a dataset learn nearly the same score function, and thus the same density, when the number of training images is large enough. In this regime of strong generalization, diffusion-generated images are distinct from the training set, and are of high visual quality, suggesting that the inductive biases of the DNNs are well-aligned with the data density. We analyze the learned denoising functions and show that the inductive biases give rise to a shrinkage operation in a basis adapted to the underlying image. Examination of these bases reveals oscillating harmonic structures along contours and in homogeneous regions. We demonstrate that trained denoisers are inductively biased towards these geometry-adaptive harmonic bases since they arise not only when the network is trained on photographic images, but also when it is trained on image classes supported on low-dimensional manifolds for which the harmonic basis is suboptimal. Finally, we show that when trained on regular image classes for which the optimal basis is known to be geometry-adaptive and harmonic, the denoising performance of the networks is near-optimal.

Towards a Mechanistic Explanation of Diffusion Model Generalization

On the Generalization Properties of Diffusion Models

Understanding Generalizability of Diffusion Models Requires Rethinking the Hidden Gaussian Structure

On the Generalization of Diffusion Model

Interpreting and Improving Diffusion Models from an Optimization Perspective

Generalization in diffusion models arises from geometry-adaptive harmonic representations

From Denoising Diffusions to Denoising Markov Models

Soft Mixture Denoising: Beyond the Expressive Bottleneck of Diffusion Models

An overview of diffusion models for generative artificial intelligence

The Emergence of Reproducibility and Generalizability in Diffusion Models

Diffusion Model for Generative Image Denoising

On Inductive Biases That Enable Generalization of Diffusion Transformers

gDDIM: Generalized denoising diffusion implicit models

Where to Diffuse, How to Diffuse, and How to Get Back: Automated Learning for Multivariate Diffusions

An Overview of Diffusion Models: Applications, Guided Generation, Statistical Rates and Optimization

Non-Denoising Forward-Time Diffusions

Diffusion with Forward Models: Solving Stochastic Inverse Problems Without Direct Supervision

Physics-Informed Diffusion Models

From memorization to generalization: a theoretical framework for diffusion-based generative models

Diffusion models as plug-and-play priors