SA-Solver: Stochastic Adams Solver for Fast Sampling of Diffusion Models

arXiv:2309.05019v2 Announce Type: replace
Abstract: Diffusion Probabilistic Models (DPMs) have achieved considerable success in generation tasks. As sampling from DPMs is equivalent to solving diffusion SDE or ODE which is time-consuming, numerous fast sampling methods built upon improved differential equation solvers are proposed. The majority of such techniques consider solving the diffusion ODE due to its superior efficiency. However, stochastic sampling could offer additional advantages in generating diverse and high-quality data. In this work, we engage in a comprehensive analysis …

abstract arxiv cs.lg differential differential equation diffusion diffusion models equation sampling solver stat.ml stochastic success tasks type