摘要
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有时可以优于清洁数据上训练的模型,完成多项图像恢复任务。