Context. Song et al. (2021) gave diffusion models a standard procedure: learn the score of the OU-noised data by regression, plug it into the reverse-time SDE, and discretize. Whether this procedure provably samples from the data distribution was open under realistic assumptions. Earlier analyses needed an -accurate score, which score matching does not deliver because the network sees no data where the density is small, or a log-Sobolev inequality for the data, which rules out well-separated modes, or gave bounds exponential in dimension. This paper shows that an -accurate score suffices for essentially any data distribution, with step counts polynomial in , the smoothness and the accuracy. The reverse process never has to mix, because the forward OU process did all the mixing; it only has to follow the forward marginals backward. Girsanov bounds the error in following them as the time-integrated squared drift error under the true reverse process, whose marginals are the forward marginals, which is exactly where score matching controls the error, and the error of starting from noise instead of can only shrink by data processing.
Setting. Data on . The forward OU process , , has marginals , and its reversal solves . The algorithm (DDPM with an exponential integrator) uses estimates on a grid and freezes the score on each step, a linear SDE with Gaussian transitions; it outputs . Assumptions: (A1) is -Lipschitz for every , with ; (A2) ; (A3) at every grid time. Nothing like convexity or a functional inequality is assumed on , and need not be Lipschitz.
Theorem 2 (DDPM). Under (A1)-(A3) and , the three terms being convergence of the forward process, discretization, and score error. If and , the choices and give after steps, so suffices. This matches the best known complexity of Langevin Monte Carlo under a log-Sobolev inequality, here without one.
Corollaries 3-5 (bounded support, early stopping). Let , , with no density assumed, so manifold data are allowed, and stop the reverse process at time with . Then is -close in TV to , which is -close in to , provided and After projecting the output onto a ball of radius , the distance to is at most with , polynomial where the earlier manifold bound was exponential. A TV bound to itself is impossible here, since has full support.
Theorems 6-7 (critically damped Langevin). The CLD forward process is , , and its reverse needs only the velocity score . Under the analogous assumptions Theorem 6 gives the bound of Theorem 2, now on and with first term . Theorem 7 shows the technique cannot do better: for , exact scores and , the path-space divergence satisfies , so the Girsanov route needs , the same dimension dependence as DDPM.
The error splits into a part from starting at instead of and a part from following the reverse path, both measured in information divergences. Compare the algorithm with the same algorithm started at , whose path law is , and that one with the true reverse path law . Data processing (the same Markov kernel acts on both starts, and marginals are images of paths) and Pinsker give where the first bound is the exponential decay of KL along OU from the log-Sobolev inequality of . Information divergences are used because the reverse dynamics can be expansive: for the reverse drift near the end is , which pushes trajectories apart when , so a coupling would need log-concavity or dissipativity and even then the error bound grows exponentially in time, while data processing in TV or KL holds for any dynamics.
Girsanov turns the tracking error into the score error along the true process. The two path laws share the noise and the start , so Girsanov's formula involves only the drift gap , and since the stochastic integral has mean zero under , The direction of the divergence matters here. The expectation is under the true reverse process, whose time- marginal is , so at grid points the score error is literally (A3), and the error under the algorithm's own law never has to be bounded. The earlier interpolation method differentiates the divergence along the algorithm, so it needs the score error under the algorithm's marginals, which forced a log-Sobolev assumption.
Each step contributes , and the dominant term is the Brownian movement over one step, of size , times from the Lipschitz score. On split the integrand into estimation error, the change of the score in time, and the change of its argument: The first term is (A3), and the last is times the path's squared movement over a time , which is . The middle term compares two forward marginals apart: is contracted by , which contributes an term, and smoothed by . The score perturbation lemma handles the smoothing by writing the change of score as the gap between at a point and its average over a small Gaussian tilt there: with and , For the tilt is strongly log-concave, so it concentrates within of a mode that lies within of , and with the middle term is . Bounding the moments ( by integration by parts) and summing gives , and Step 1 gives Theorem 2. The term is the of Brownian movement over one step passed through the Lipschitz score, which is why and is linear in .
For bounded data, stop early and treat the slightly noised data as the target. Without a smooth density, (A1) fails at and may be infinite, but the OU marginal at a small time satisfies every hypothesis of Theorem 2, though with a Lipschitz constant that grows like . The explicit OU solution gives , at most for the stated ; Otto-Villani short-time regularization makes the start-up term finite, ; and the second-order Tweedie identity bounds the Hessian through the posterior covariance of given , which is at most : so every , , has . Theorem 2 with target gives Corollary 3, and is where comes from. For , project onto a ball: there is at most the radius times plus an exponentially small tail term, and the projection is 1-Lipschitz.
CLD gains nothing, because its score is taken in the velocity, which moves like Brownian motion. Theorem 6 repeats Steps 2-3 with , replacing OU's KL decay by entropic hypocoercivity. For Theorem 7 take : the joint law stays , the exact velocity score is , Novikov holds directly, and Girsanov is an identity whose integrand is the velocity's own increment, using for . The velocity carries the Brownian increment. Underdamped Langevin gets its speedup in log-concave sampling because its gradient depends only on the smooth position, whose increment is ; the reverse CLD drift depends on , so the speedup is lost. This is a lower bound for the technique only; the authors conjecture, without proof, that CLD has no better dimension dependence in general.
Consequences and limits. Theorem 2 is a black-box reduction: any score estimator with small error under the forward marginals yields a sampler at polynomial cost, so all the difficulty of diffusion models is in learning the score, and known hardness of learning Gaussian mixtures or shallow-ReLU pushforwards transfers to score estimation, the first information-computation gaps for that task. A network that fits the empirical objective exactly learns the score of the Gaussian-smoothed training set, and Corollary 3 applied to the empirical measure (an application the paper does not make) says the sampler then reproduces training points; whether the network instead approximates the population score is the generalization question (Kadkhodaie et al.), which (A3) assumes away. The limits are that the score must be uniformly -Lipschitz at all times, which manifold data violate near , and the early-stopping fix puts large powers of into the bound; that (A3) is assumed at every grid time with one ; and that the score error enters as , so it must already be at the target accuracy. Later work removed the Lipschitz assumption and reached step counts nearly linear in (Benton et al., 2024).
Terms
- forward OU; with decaying like
- Reverse SDE
- drift , noise , started at
- Exponential integrator
- score frozen at on each step; linear SDE with Gaussian transitions
- (A1)-(A3)
- -Lipschitz scores, finite second moment, score error
- Girsanov bound
- Theorem 2
- steps for TV , with and logarithmic
- Score perturbation lemma
- via a strongly log-concave tilt
- Localization
- stopped exponential martingales plus lower semicontinuity of KL replace Novikov
- Early stopping
- target with ; ,
- CLD lower bound
- for Gaussian data and exact score; the velocity moves by order per step
Quiz
- Why does the analysis avoid Wasserstein coupling of the reverse process?For the reverse drift near the end is , expansive when , and contraction in generally needs log-concavity or dissipativity and gives bounds that grow exponentially in time. TV and KL only shrink under a common Markov kernel.
- Why does the total time need to be only logarithmic in the target accuracy?Starting at instead of adds , which the log-Sobolev inequality of makes exponentially small. The other two terms grow like , so is set just large enough for the first.
- The data are a mixture of well-separated modes, a case where Langevin Monte Carlo mixes exponentially slowly. Why does Theorem 2 still give a polynomial number of steps?The reverse process starts near and follows the forward marginals backward, and the Girsanov bound involves only the drift error along that path, so no log-Sobolev inequality or convexity is assumed on . Langevin must cross between modes on its own, which is where it is slow.
- A network fits the empirical denoising objective exactly on training points. What does Theorem 2 (through Corollary 3) say about its samples?The reduction applies to whatever score is supplied, and the exact empirical score is the score of the smoothed empirical measure, so the sampler targets that measure. (A3) against the population fails badly at small , and generalization depends on the network not reaching this minimizer.
Sources
- Sitan Chen, Sinho Chewi, Jerry Li, Yuanzhi Li, Adil Salim, and Anru R. Zhang. Sampling is as easy as learning the score: theory for diffusion models with minimal data assumptions. ICLR, 2023.
- Yang Song, Jascha Sohl-Dickstein, Diederik P. Kingma, Abhishek Kumar, Stefano Ermon, and Ben Poole. Score-Based Generative Modeling through Stochastic Differential Equations. ICLR, 2021.
- Joe Benton, Valentin De Bortoli, Arnaud Doucet, and George Deligiannidis. Nearly d-Linear Convergence Bounds for Diffusion Models via Stochastic Localization. ICLR, 2024.