Fast ODE-based Sampling for Diffusion Models in Around 5 Steps

arXiv:2312.00094v2 Announce Type: replace
Abstract: Sampling from diffusion models can be treated as solving the corresponding ordinary differential equations (ODEs), with the aim of obtaining an accurate solution with as few number of function evaluations (NFE) as possible. Recently, various fast samplers utilizing higher-order ODE solvers have emerged and achieved better performance than the initial first-order one. However, these numerical methods inherently result in certain approximation errors, which significantly degrades sample quality with extremely small NFE (e.g., around 5). In …

arxiv cs.ai cs.cv diffusion diffusion models sampling type