Diffusion probabilistic models (DPMs) are a powerful class of generative models known for their ability to generate high-fidelity image samples. A major challenge in the implementation of DPMs is the slow sampling process. In this work, we bring a high-efficiency sampler for DPMs. Specifically, we propose a score-based exact solution paradigm for the diffusion ODEs corresponding to the sampling process of DPMs, which introduces a new perspective on developing numerical algorithms for solving diffusion ODEs. To achieve an efficient sampler, we propose a recursive derivative estimation (RDE) method to reduce the estimation error. With our proposed solution paradigm and RDE method, we propose the score-integrand solver with the convergence order guarantee as efficient solver (SciRE-Solver) for solving diffusion ODEs. The SciRE-Solver attains state-of-the-art (SOTA) sampling performance with a limited number of score function evaluations (NFE) on both discrete-time and continuous-time DPMs in comparison to existing training-free sampling algorithms. Such as, we achieve $3.48$ FID with $12$ NFE and $2.42$ FID with $20$ NFE for continuous-time DPMs on CIFAR10, respectively. Different from other samplers, SciRE-Solver has the promising potential to surpass the FIDs achieved in the original papers of some pre-trained models with a small NFEs. For example, we reach SOTA value of $2.40$ FID with $100$ NFE for continuous-time DPM and of $3.15$ FID with $84$ NFE for discrete-time DPM on CIFAR-10, as well as of $2.17$ ($2.02$) FID with $18$ ($50$) NFE for discrete-time DPM on CelebA 64$\times$64.
翻译:扩散概率模型(DPMs)是一类强大的生成模型,以其生成高保真图像样本的能力而闻名。DPMs实现中的主要挑战在于其采样过程速度缓慢。本文提出了一种面向DPMs的高效采样器。具体而言,我们为对应DPMs采样过程的扩散ODE提出了一种基于分数的精确求解范式,这为开发求解扩散ODE的数值算法提供了新视角。为实现高效采样器,我们提出了一种递归导数估计(RDE)方法来降低估计误差。基于所提出的求解范式与RDE方法,我们提出了具有收敛阶保证的分数积分求解器(SciRE-Solver),作为求解扩散ODE的高效求解器。与现有免训练采样算法相比,SciRE-Solver在离散时间和连续时间DPMs上均能以有限次评分函数评估(NFE)实现最先进的采样性能。例如,在CIFAR-10数据集上,对于连续时间DPMs,我们在12次NFE下达到3.48的FID,在20次NFE下达到2.42的FID。与其他采样器不同,SciRE-Solver具有显著潜力:在少量NFE下即可超越某些预训练模型原始论文报告的FID值。例如,在CIFAR-10数据集上,我们以100次NFE对连续时间DPM达到2.40的FID最先进值,以84次NFE对离散时间DPM达到3.15的FID;在CelebA 64×64数据集上,以18次(50次)NFE对离散时间DPM达到2.17(2.02)的FID。