English

Asymptotic-Preserving A Posteriori Analysis of Diffusion and Flow-Matching Samplers

Machine Learning 2026-07-05 v1 Numerical Analysis

Abstract

Diffusion and flow-matching samplers integrate a learned probability-flow ODE from a large noise scale down to a small terminal floor σmin\sigma_{\min}, at which the score is stiff and the flow develops a boundary layer. We treat σmin\sigma_{\min} as a singular-perturbation parameter and determine which fixed-step samplers are asymptotic-preserving (AP), that is, stable and uniformly accurate as σmin0\sigma_{\min}\to0, casting the criteria as an a posteriori audit: residual functionals with σmin\sigma_{\min}-uniform coefficients, computable on a pretrained checkpoint without ground-truth scores or exact trajectories. On the terminal layer, Euler in the σ\sigma-clock, the deterministic DDIM update, is the unique layer-exact discretization up to affine reparameterization, with rectified flow its flow-matching counterpart; the λ\lambda-clock is stable only for steps hh=1+W(1/e)h\le h_\star=1+W(1/e), and the uniform-σ2\sigma^2 heat clock stalls a σmin\sigma_{\min}-independent distance from the data. On two solvable models (rank-deficient Gaussian, symmetric two-point mixture), deterministic samplers remain first-order uniformly accurate with no log(1/σmin)\log(1/\sigma_{\min}) factor, even across a symmetric posterior-switching interface whose distributional budget is a universal constant; the logarithm is charged entirely to the It\^o term of stochastic samplers, whose path-KL scales as Λ2/N\Lambda^2/N against the ODE's O(Λ2/N2)O(\Lambda^2/N^2) budget, with Λ=log(σmax/σmin)\Lambda=\log(\sigma_{\max}/\sigma_{\min}). On the EDM CIFAR-10 checkpoint, spectra measured once predict held-out residual budgets across step count, schedule, and noise level against pre-specified gates with no per-configuration refitting, and calibrate the It\^o coefficient at M1=1.00±0.01M_1=1.00\pm0.01. The clock decides stability; the noise, not the geometry, charges the logarithm.

Keywords

Cite

@article{arxiv.2607.04113,
  title  = {Asymptotic-Preserving A Posteriori Analysis of Diffusion and Flow-Matching Samplers},
  author = {Shiheng Zhang},
  journal= {arXiv preprint arXiv:2607.04113},
  year   = {2026}
}