LLM2D
ambient 扩散后验采样:使用受污染数据训练的扩散模型求解逆问题
Ambient Diffusion Posterior Sampling: Solving Inverse Problems with Diffusion Models Trained on Corrupted Data
作者: Asad Aali, Giannis Daras, Brett Levac, Sidharth Kumar, Alexandros G. Dimakis, Jonathan I. Tamir
发布日期: 4/23/2025
arXiv ID: oai:arXiv.org:2403.08728v2

摘要

arXiv:2403.08728v2 逆问题类型: 替换-交叉 摘要: 我们提供了一种框架,用于解决来自线性受腐蚀数据中学习的扩散模型的逆问题。首先,我们将Ambient Diffusion框架扩展为能够直接从Fourier域中受腐蚀的数据中进行训练。其次,我们仅使用采样速率为R=2,4,6,8的Fourier子采样多线圈测量数据来训练MRI的扩散模型。随后,我们提出了Ambient Diffusion后验采样(A-DPS),这是一种利用在一种类型腐蚀(如图像填充)上预先训练的生成模型来对另一种前向过程(如图像模糊)的测量数据进行后验采样的重建算法。对于高加速率下的MRI重建,我们观察到,训练于子采样数据上的A-DPS模型比训练于全采样数据上的模型更适合解决逆问题。我们还在自然图像数据集(CelebA, FFHQ, AFHQ)上测试了A-DPS的有效性,并展示了在速度和性能方面,A-DPS有时可以优于清洁数据上训练的模型,完成多项图像恢复任务。