Goto

Collaborating Authors

 nfe


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

Neural Information Processing Systems

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.




Improving the Training of Rectified Flows

Neural Information Processing Systems

One approach for tackling this problem is rectified flows, which iteratively learn smooth ODE paths that are less susceptible to truncation error. However, rectified flows still require a relatively large number of function evaluations (NFEs). In this work, we propose improved techniques for training rectified flows, allowing them to compete with knowledge distillation methods even in the low NFE setting.







Solver

Neural Information Processing Systems

Based onourformulation, weproposeDPM-Solver,afastdedicated high-order solver for diffusion ODEs with the convergence order guarantee.