arXiv:2410.03463v5 公告类型: replace-cross 摘要：最近在扩散模型方面的进展在学习数据先验以解决逆问题方面非常有效。它们利用扩散采样步骤来诱导数据先验，在每个步骤中使用测量指导梯度来施加数据的一致性。对于一般的逆问题，在使用未有条件训练的扩散模型时，由于难以计算测量似然性，因此需要进行近似，这会导致后验采样的不准确性。换句话说，由于这些方法的近似，它们无法在由扩散先验定义的数据流形上保留生成过程，从而在诸如图像恢复等应用中产生伪影。为了增强扩散模型在解决逆问题时的表现和鲁棒性，我们提出了扩散状态引导投影梯度（DiffStateGrad），该方法将测量梯度投影到一个低秩近似的中间扩散过程状态子空间中。作为模块，DiffStateGrad 可以添加到广泛的基于扩散的逆求解器中，以提高先验流形上扩散过程的保留并滤除导致伪影的成分。我们强调，DiffStateGrad 在测量指导步骤大小和噪声方面增强了扩散模型的选择鲁棒性，同时改进了最坏情况下的性能。最后，我们展示了 DiffStateGrad 在线性和非线性图像恢复逆问题中优于最新方法。我们的代码可在 https://github.com/Anima-Lab/DiffStateGrad 获取。