Constrained Plug-and-Play Priors for Image Restoration

J Imaging. 2024 Feb 19;10(2):50. doi: 10.3390/jimaging10020050.

Abstract

The Plug-and-Play framework has demonstrated that a denoiser can implicitly serve as the image prior for model-based methods for solving various inverse problems such as image restoration tasks. This characteristic enables the integration of the flexibility of model-based methods with the effectiveness of learning-based denoisers. However, the regularization strength induced by denoisers in the traditional Plug-and-Play framework lacks a physical interpretation, necessitating demanding parameter tuning. This paper addresses this issue by introducing the Constrained Plug-and-Play (CPnP) method, which reformulates the traditional PnP as a constrained optimization problem. In this formulation, the regularization parameter directly corresponds to the amount of noise in the measurements. The solution to the constrained problem is obtained through the design of an efficient method based on the Alternating Direction Method of Multipliers (ADMM). Our experiments demonstrate that CPnP outperforms competing methods in terms of stability and robustness while also achieving competitive performance for image quality.

Keywords: constrained formulation; discrepancy principle; image restoration; inverse problems; plug-and-play priors; regularization by denoising.

Grants and funding

This work has been conducted under the activities of Gruppo Nazionale Calcolo Scientifico (GNCS) of Istituto Nazionale di Alta Matematica (INDAM) and under the activities of the 2022 PRIN project “Sustainable Tomographic Imaging with Learning and rEgularization”. This study was carried out within the MICS (Made in Italy—Circular and Sustainable) Extended Partnership and received funding from the European Union Next-GenerationEU (Piano Nazionale di ripresa e resilienza (PNRR)—Missione 4 Componente 2, Investimento 1.3—D.D. 1551.11-10-2022, PE00000004).