等变即插即用图像重建
图像与视频处理
2024-05-24 v2 计算机视觉与模式识别
摘要
即插即用算法构成了一个流行的框架，用于解决逆成像问题，其依赖于通过去噪器对图像先验的隐式定义。这些算法可以利用强大的预训练去噪器来解决广泛的成像任务，从而避免针对每个任务训练模型的必要。不幸的是，即插即用方法常表现出不稳定行为，阻碍了其通用性的前景并导致重建图像质量次优。在本工作中，我们展示了对去噪器强制施加对某些变换群（旋转、反射和/或平移）的等变性，能强烈改善算法的稳定性及其重建质量。我们提供了理论分析，阐明了等变性对更好性能和稳定性的作用。我们提出了一种简单算法，通过在算法每次迭代时对去噪器输入施加随机变换并对输出施加逆变换，从而对任意现有去噪器强制等变性。在多种成像模态和去噪网络上的实验表明，等变即插即用算法相比非等变对应方法，同时改善了重建性能和稳定性。
引用
@article{arxiv.2312.01831,
title = {Equivariant plug-and-play image reconstruction},
author = {Matthieu Terris and Thomas Moreau and Nelly Pustelnik and Julian Tachella},
journal= {arXiv preprint arXiv:2312.01831},
year = {2024}
}