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.
Original language | English |
---|---|
Journal | Transactions on Machine Learning Research |
Volume | 2024 |
Publication status | Published - 2024 |
Bibliographical note
Publisher Copyright:© 2024, Transactions on Machine Learning Research. All rights reserved.
Funding
This project was supported with funding from the Cambridge Centre for Data-Driven Discovery and Accelerate Programme for Scientific Discovery, made possible by a donation from Schmidt Futures. YC is funded by an AstraZeneca studentship and a Google studentship. The work of RHC was partially supported by HKRGC GRF grants CityU11309922, CRF grant C1013-21GF and HKITF MHKJFS Grant MHP/054/22. CBS acknowledges support from the Philip Leverhulme Prize, the Royal Society Wolfson Fellowship, the EPSRC advanced career fellowship EP/V029428/1, EPSRC grants EP/S026045/1 and EP/T003553/1, EP/N014588/1, EP/T017961/1, the Wellcome Innovator Awards 215733/Z/19/Z and 221633/Z/20/Z, the European Union Horizon 2020 research and innovation programme under the Marie Skodowska-Curie grant agreement No. 777826 NoMADS, the Cantab Capital Institute for the Mathematics of Information and the Alan Turing Institute. AIAR acknowledges support from CMIH (EP/T017961/1) and CCIMI, University of Cambridge. This work was supported in part by Oracle Cloud credits and related resources provided by Oracle for Research. Also, EPSRC Digital Core Capability.