LLM2D
MRS:基于ODE和SDE求解器的快速平均回复扩散采样器
MRS: A Fast Sampler for Mean Reverting Diffusion based on ODE and SDE Solvers
作者: Ao Li, Wei Fang, Hongbo Zhao, Le Lu, Ge Yang, Minfeng Xu
发布日期: 2/13/2025
arXiv ID: oai:arXiv.org:2502.07856v2

摘要

arXiv:2502.07856v2 控制生成类型: 交叉 摘要:在扩散模型的应用中,可控生成具有实际意义,但也颇具挑战性。当前的可控生成方法主要集中在修改扩散模型的分数函数上,而反向收敛(MR)扩散直接修改随机微分方程(SDE)的结构,使得图像条件的融入更简单且更自然。然而,当前的无训练快速采样器不直接适用于MR扩散。因此,MR扩散需要数百次NFE(函数评估次数)才能获得高质量样本。本文中,我们提出了一种新算法MRS(MR采样器),以减少MR扩散的采样NFEs。我们解决了与MR扩散相关的逆时SDE和概率流常微分方程(PF-ODE),并推导出了半解析解。解由一个解析函数和一个由神经网络参数化的积分组成。基于此解,我们可以在更少的步骤中生成高质量样本。我们的方法不需要训练,并支持所有主流参数化,包括噪声预测、数据预测和速度预测。广泛实验表明,在十个不同的图像恢复任务中,MR采样器保持了高质量的采样,速度快了10到20倍。我们的算法加速了MR扩散的采样过程,使其在可控生成中更加实用。