LLM2D
扩散状态引导投影梯度方法用于反问题
Diffusion State-Guided Projected Gradient for Inverse Problems
作者: Rayhan Zirvi, Bahareh Tolooshams, Anima Anandkumar
发布日期: 10/7/2024
arXiv ID: oai:arXiv.org:2410.03463v1

摘要

近年来,扩散模型在学习数据先验以解决逆问题方面取得了显著进展。它们利用扩散采样步骤来诱导数据先验,同时在每个步骤中使用测量引导梯度来强加数据一致性。对于一般的逆问题,当使用无条件训练的扩散模型时,由于测量似然性难以处理,需要进行近似,导致后验采样不准确。换句话说,由于这些方法的近似,它们无法在扩散先验定义的数据流形上保留生成过程,从而导致图像恢复等应用中的伪影。为了提高扩散模型在解决逆问题方面的性能和鲁棒性,我们提出了扩散状态引导投影梯度 (DiffStateGrad),它将测量梯度投影到一个子空间,该子空间是扩散过程中间状态的低秩近似。DiffStateGrad 作为一个模块,可以添加到各种基于扩散的逆解算器中,以改善扩散过程在先验流形上的保留,并滤除产生伪影的成分。我们强调,DiffStateGrad 提高了扩散模型在测量引导步长和噪声选择方面的鲁棒性,同时提高了最坏情况下的性能。最后,我们证明了 DiffStateGrad 在线性​​和非线性图像恢复逆问题上优于现有技术。