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