Single-Shot Plug-and-Play Methods for Inverse Problems

Abstract

The utilisation of Plug-and-Play (PnP) priors in inverse problems has become increasingly prominent in recent years. This preference is based on the mathematical equivalence between the general proximal operator and the regularised denoiser, facilitating the adaptation of various off-the-shelf denoiser priors to a wide range of inverse problems. However, existing PnP models predominantly rely on pre-trained denoisers using large datasets. In this work, we introduce Single-Shot PnP methods (SS-PnP), shifting the focus to solving inverse problems with minimal data. First, we integrate Single-Shot proximal denoisers into iterative methods, enabling training with single instances. Second, we propose implicit neural priors based on a novel function that preserves relevant frequencies to capture fine details while avoiding the issue of vanishing gradients. We demonstrate, through extensive numerical and visual experiments, that our method leads to better approximations.

Cite

Text

Cheng et al. "Single-Shot Plug-and-Play Methods for Inverse Problems." Transactions on Machine Learning Research, 2024.

Markdown

[Cheng et al. "Single-Shot Plug-and-Play Methods for Inverse Problems." Transactions on Machine Learning Research, 2024.](https://mlanthology.org/tmlr/2024/cheng2024tmlr-singleshot/)

BibTeX

@article{cheng2024tmlr-singleshot,
  title     = {{Single-Shot Plug-and-Play Methods for Inverse Problems}},
  author    = {Cheng, Yanqi and Zhang, Lipei and Shen, Zhenda and Wang, Shujun and Yu, Lequan and Chan, Raymond H. and Schönlieb, Carola-Bibiane and Aviles-Rivero, Angelica I},
  journal   = {Transactions on Machine Learning Research},
  year      = {2024},
  url       = {https://mlanthology.org/tmlr/2024/cheng2024tmlr-singleshot/}
}