首页 > AI前沿 > Forward-Value Discretization: Faster Stochastic Diffusion Sampling Without High-Order Approximation

Forward-Value Discretization: Faster Stochastic Diffusion Sampling Without High-Order Approximation

arXiv机器学习 2026-11-11 00:53 11 阅读 查看原文

Stochastic diffusion samplers often require substantially more neural function evaluations (NFEs) than deterministic samplers to achieve comparable sample quality. Although higher-order discretization accelerates deterministic sampling, its benefits for stochastic sampling remains unclear. We establish matching upper and worst-case lower bounds showing that stochastic samplers with higher-order approximations remain first-order convergent, even with exact score functions. Thus, increasing the approximation order does not generally improve the convergence order. Motivated by this limitation, we introduce forward-value discretization, a training-free framework that evaluates the data/noise prediction model at a lookahead estimate constructed from previous predictions.

Deterministic sampling: Forward DPMSolver (F-DPMSolver)

For deterministic sampling, the resulting Forward DPMSolver (F-DPMSolver), improves sample quality at low NFE budgets through its DDIM-based variant, despite remaining first-order convergent.

Stochastic sampling: Forward DDPM (F-DDPM)

For stochastic sampling, we develop Forward DDPM (F-DDPM), which evaluates the model at a cleaner lookahead point adjusted for subsequent noise injection. This estimate reuses both the previous model prediction and the previously injected noise.

  • Both variants admit first-order convergence guarantees with one model evaluation per iteration.

Experiments

Across CIFAR-10, ImageNet, FFHQ, and LSUN, F-DDPM consistently achieves lower FID than the evaluated stochastic baselines at matched NFE budgets. With 1.5 times as many NFEs as deterministic baselines, it achieves competitive sample quality, narrowing the practical efficiency gap between stochastic and deterministic sampling.

Code is available at https://github.com/Na-Li66/F-DDPM.