Nov, 2022
使用预训练的2D扩散模型解决3D反问题
Solving 3D Inverse Problems using Pre-trained 2D Diffusion Models
TL;DR本文结合传统的基于模型的迭代重建方法和现代扩散模型，提出了一种高度有效的方法来解决3D医学图像重建任务，包括稀疏视图断层扫描、有限角度断层扫描、预训练2D扩散模型压缩感知MRI等。我们在测试时提出了2D扩散先验的增强模型先验，从而实现了所有维度上的一致重建。该方法可以在单个普通GPU上运行，并且在最极端情况下（例如2视的3D断层扫描）表现出高保真度和准确度的重建。我们进一步揭示了该方法的泛化能力出乎意料地高，并且可以用于重建与训练数据集完全不同的体积。