摘要
现有的基于扩散的反问题方法使用分数函数从后验分布中采样,并将生成的随机样本作为解。在需要后验均值的应用中,需要从后验分布中生成多个样本,这非常耗时。本文通过分析条件反向扩散过程的概率密度演变,证明了后验均值可以通过跟踪每个反向扩散步骤的均值来实现。基于此,我们建立了一个名为反向均值传播 (RMP) 的框架,该框架直接针对后验均值。我们证明了 RMP 可以通过解决一个变分推断问题来实现,该问题可以进一步分解为最小化每个反向步骤的反向 KL 散度。我们进一步开发了一种算法,该算法使用分数函数用自然梯度下降优化反向 KL 散度,并在每个反向步骤传播均值。实验验证了我们框架理论的有效性,并表明我们的算法在各种反问题中以更低的计算复杂度优于最先进的算法,在重建性能方面表现出色。