stat.MLOct 7, 2026

Universal Local Error and Realized Amplification for the First-Order EDM Predictor

Authors: Nicolas Brosse, Arnak S. Dalalyan

Organizations: CREST, ENSAE Paris, Institut Polytechnique de Paris

Abstract

We analyze the first-order deterministic diffusion sampler of Karras et al. (2022), termed EDM, in 2-Wasserstein distance by separating two sources of error: local discretization error and its amplification by subsequent learned steps. We prove that local error admits a universal bound: for any data distribution with finite second moment, the one-step discretization error is quadratic in the step size, with an explicit constant that does not depend on the data distribution. Error propagation, in contrast, depends on the learned network. At high noise levels, we exploit the network parametrization of EDM to derive an explicit contraction criterion. At low noise levels, we measure propagation through the amplification realized on the distributions transported by the sampler; this realized amplification can be arbitrarily smaller than the worst-case Lipschitz constant. This analysis yields an O(eΛK/K)O(e^{Λ_K}/K) global discretization error for KK sampling steps, where ΛKΛ_K is the low-noise log-amplification. Experiments on a one-dimensional Gaussian mixture show how measured amplification accounts for slower error decay on finite sampling grids. Diagnostics on a pretrained CIFAR-10 model illustrate related stability mechanisms without certifying the global assumptions.

Figures & tables

Appendix figures & tables14 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. Geometry-Aware Discretization Error of Diffusion Models

    May 8, 2026Samuel Hurault, Thomas Moreau, Gabriel PeyréDiffusion SamplingFew-Step Diffusion Sampling

  2. Diffusion Removes Langevin's Conditioning Dependence: A Sharp Gaussian Analysis

    Oct 8, 2026Adam Perbost, Francis Bach, Pierre Marion