Publication · 2026
Geometry-aware discretization error of diffusion models
arXiv preprint arXiv:2605.08392v2, 2026
Abstract
Practical diffusion sampling requires simulating a reverse-time ODE or SDE with a limited number of denoising steps, making the choice of sampling parameters crucial for minimizing discretization error. We derive asymptotically exact small-stepsize expansions of Euler-Maruyama weak and Fréchet errors for general smooth reverse diffusions, with explicit formulas for Gaussian data. The resulting objectives optimize the noise and rescaling schedules and the stochasticity coefficient according to the target covariance spectrum.