Universal Local Error and Realized Amplification for the First-Order EDM Predictor
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.
ProofPaper ↗
Key points
- 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.
- At high noise levels, we exploit the network parametrization of EDM to derive an explicit contraction criterion.
- This analysis yields an $O(e^{ΛK}/K)$ global discretization error for $K$ sampling steps, where $Λ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.
Sources (1)
- [1]Universal Local Error and Realized Amplification for the First-Order EDM PredictorarXiv (AI, ML, NLP, CV, robotics, multi-agent) · Oct 7, 02:54 PM
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.
Extractive summary: sentences quoted from the sources.