面向逆成像问题的免调参即插即用近端算法
图像与视频处理
2020-11-19 v2 计算机视觉与模式识别
摘要
即插即用(Plug-and-play, PnP)是一种将 ADMM 或其他近端算法与先进去噪器先验相结合的非凸框架。近年来,PnP 取得了巨大的实证成功,尤其是集成了基于深度学习的去噪器。然而,基于 PnP 的方法的一个关键问题是它们需要手动调节参数。为了在成像条件高度差异和场景内容多变的情况下获得高质量结果,这是必要的。在这项工作中,我们提出了一种免调参的 PnP 近端算法,它能够自动确定内部参数,包括惩罚参数、去噪强度和终止时间。我们方法的一个关键部分是开发一个策略网络用于参数的自动搜索,该网络可通过混合无模型与基于模型的深度强化学习有效学习。我们通过数值和视觉实验证明,学习到的策略可以为不同状态定制不同参数,并且通常比现有手工准则更高效、更有效。此外,我们讨论了所插入去噪器的实际考量,其与我们学习到的策略结合产生了最先进的结果。这在线性和非线性示例逆成像问题上均很普遍,特别地,我们在压缩感知 MRI 和相位恢复上展示了有前景的结果。
引用
@article{arxiv.2002.09611,
title = {Tuning-free Plug-and-Play Proximal Algorithm for Inverse Imaging Problems},
author = {Kaixuan Wei and Angelica Aviles-Rivero and Jingwei Liang and Ying Fu and Carola-Bibiane Schönlieb and Hua Huang},
journal= {arXiv preprint arXiv:2002.09611},
year = {2020}
}