Score-Based Diffusion Models (SGMs)
- Score-based diffusion models (SGMs) are generative models that reverse a stochastic diffusion process via score function gradients to transform simple reference measures into complex data distributions.
- They leverage advanced tools like Malliavin and Γ-calculus to rigorously extend analysis from finite-dimensional spaces to infinite-dimensional Hilbert spaces and manifold-valued data.
- Dimension-independent entropic convergence bounds and an L2 regression formulation ensure robust denoising and optimal filtering in modeling complex random fields.
Score-based diffusion models (SGMs) are a class of generative models that transform a simple reference distribution into a complex data distribution by reversing a stochastic diffusion process, guided by the score function—the gradient of the log-density of the evolving distribution. Recent advances have placed the theory on increasingly rigorous mathematical ground, extending the analysis and modeling capabilities from finite-dimensional data to random fields and infinite-dimensional Hilbert spaces through the application of Malliavin and Γ (“Gamma”) calculus. This generalization enables SGMs to model complex functional data, including random fields on manifolds such as the sphere, and provides robust theoretical guarantees for both the forward and reverse processes.
1. Infinite-Dimensional Hilbert Space Formalism for SGMs
SGMs can be formulated directly on an infinite-dimensional separable Hilbert space (e.g., for random fields on the sphere) by defining both the forward diffusion (noising) and the reverse (denoising) processes using tools from functional analysis.
A Gaussian reference measure with trace-class covariance operator on is specified, with spectral decomposition for an orthonormal basis . The associated Cameron–Martin space is
with inner product .
Cylindrical functions , where and is smooth with polynomial growth, serve as test functions for the development of the infinite-dimensional calculus. Here, maps the Cameron–Martin element to a Paley–Wiener integral.
The gradient (Malliavin derivative) of in the direction of the Cameron–Martin space is
This is used to define the carré du champ operator (Γ operator) as . The associated Dirichlet form is
The forward diffusion process is then given by the infinite-dimensional Ornstein–Uhlenbeck dynamics,
where is an -valued Brownian motion with covariance given by .
2. Malliavin Derivative as the Score Function: Conditional Expectation Representation
In this functional framework, the reverse SGM process is driven by a score function defined as a suitable Malliavin (gradient) operator. Specifically, for the time-marginal density , the critical quantity is
where is a test function.
A principal result (see Theorem 3.1) identifies this quantity as a conditional expectation: This formalizes the Malliavin derivative as the abstract infinite-dimensional analog of the finite-dimensional score function . The identification with a conditional expectation positions learning the score in SGMs as a form of optimal regression in Hilbert space.
This conditional expectation property is essential for training, as it implies that the best estimator of the initial condition given the observed influx of noise is precisely the conditional mean—an underpinning of optimal filtering and denoising.
3. Entropic Convergence in Infinite Dimensions and Fisher Information
The established framework allows for rigorous extension of finite-dimensional entropic convergence results (e.g., Kullback–Leibler divergence bounds) to infinite-dimensional settings. If denotes a projection onto a finite set of coordinates (subsystem), the convergence bound holds uniformly in : Here is the Fisher information relative to , with the gradient taken in the Cameron–Martin norm, and quantifies the score approximation error. These bounds are dimension-free in the sense that they persist under arbitrary finite-dimensional projections, highlighting the centrality of the Cameron–Martin norm for regularity properties and control of the KL divergence. Γ-calculus and functional inequalities underlie the proof, leveraging the structure of infinite-dimensional Dirichlet forms.
4. Specialization to Random Fields on the Sphere and the Whittle–Matérn Construction
Taking , the machinery specializes to random fields over the sphere. An isotropic random field admits the Karhunen–Loève expansion
with the spherical harmonics and uncorrelated random coefficients.
For Whittle–Matérn fields (reference noise model), one uses the stochastic PDE
where is spherical white noise, , . The eigenfunctions are the spherical harmonics, with eigenvalues .
The Cameron–Martin space becomes
Each harmonic coefficient undergoes an independent infinite-dimensional Ornstein–Uhlenbeck process: Thus, modeling and generation can be performed in the spectral domain, with denoising implemented via neural predictors for the conditional expectation of the coefficients given noisy observations.
5. Malliavin–Gamma Calculus and Dirichlet Forms in the Theory of SGMs
The synthesis of Malliavin calculus and Γ-calculus (the paper of Dirichlet forms) allows for the expression of both the forward and reverse SDEs, the score function, and information-theoretic quantities in a dimension-independent, analytically robust language.
- The Dirichlet form, essentially capturing energy dissipation or the “diffusive” nature of the process, is used to define the generator of the forward SDE.
- The Malliavin derivative, acting as the proper “score” in the infinite-dimensional space, is simultaneously used to define optimal regression predictors and to link SGMs with conditional expectation-based denoising schemes.
- The carré du champ operator and Fisher information in the Cameron–Martin norm replace classical finite-dimensional analytic quantities, enabling generalization to abstract Wiener spaces and manifold-valued data.
A critical implication is that key convergence bounds (e.g., for relative entropy, marginal law matching) no longer scale with the ambient dimension, but only with regularity properties of the data as measured in the Cameron–Martin space—a phenomenon crucial for modeling high-dimensional functional data and complex random fields.
6. Summary and Significance
This Malliavin–Gamma calculus framework robustly generalizes SGMs:
- Lifting the SGM paradigm from to infinite-dimensional Hilbert spaces enables modeling of challenging data types such as random fields and functional observations.
- The score function is characterized as a Malliavin derivative, with a concrete conditional expectation representation directly connected to regression.
- Entropic (KL) convergence bounds uniform over finite-dimensional projections are achieved, governed by Fisher information in the Cameron–Martin space.
- For random fields, particularly on manifolds such as the sphere, specialization to Whittle–Matérn fields and spectral decompositions provides a concrete computational pathway.
- This approach paves the way for theoretically grounded, dimension-independent generative modeling of infinite-dimensional data and augments the deployment of SGMs in scientific, functional, and geometric data settings (Greco, 19 May 2025).