Jun, 2024

基于补丁扩散模型学习图像先验用于解决逆问题

TL;DR本文提出了一种通过仅对图像的补丁进行扩散模型训练来学习整个图像的高效数据先验的方法,该方法通过分数和位置编码获取整个图像的分数函数,并将其用作解决逆问题的先验。该方法在提高内存效率和数据效率的同时,仍然能够通过位置编码生成整个图像,并且可以与不同的扩散逆求解器(DIS)配合使用,适用于自然和医学图像域中的各种逆问题解决,包括 CT 重建,去模糊和超分辨率,仅使用基于补丁的先验方法优于基于整个图像先验训练的先前 DIS 方法在训练数据有限的情况下表现出了数据效率。