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