On the Contractivity of Plug-and-Play Operators (2309.16899v3)
Abstract: In plug-and-play (PnP) regularization, the proximal operator in algorithms such as ISTA and ADMM is replaced by a powerful denoiser. This formal substitution works surprisingly well in practice. In fact, PnP has been shown to give state-of-the-art results for various imaging applications. The empirical success of PnP has motivated researchers to understand its theoretical underpinnings and, in particular, its convergence. It was shown in prior work that for kernel denoisers such as the nonlocal means, PnP-ISTA provably converges under some strong assumptions on the forward model. The present work is motivated by the following questions: Can we relax the assumptions on the forward model? Can the convergence analysis be extended to PnP-ADMM? Can we estimate the convergence rate? In this letter, we resolve these questions using the contraction mapping theorem: (i) for symmetric denoisers, we show that (under mild conditions) PnP-ISTA and PnP-ADMM exhibit linear convergence; and (ii) for kernel denoisers, we show that PnP-ISTA and PnP-ADMM converge linearly for image inpainting. We validate our theoretical findings using reconstruction experiments.
- A. Ribes and F. Schmitt. Linear inverse problems in imaging. IEEE Signal Process. Mag., 25(4):84–99, 2008.
- Charles A Bouman. Foundations of Computational Imaging: A Model-Based Approach. SIAM, 2022.
- T. F. Chan and J. Shen. Image Processing and Analysis: Variational, PDE, Wavelet, and Stochastic Methods. SIAM, 2005.
- Nonlinear total variation based noise removal algorithms. Physica D, 60:259–268, 1992.
- D. Zoran and Y. Weiss. From learning models of natural image patches to whole image restoration. Proc. IEEE Intl. Conf. Comp. Vis., pages 479–486, 2011.
- Learning deep CNN denoiser prior for image restoration. Proc. IEEE Conf. Comp. Vis. Pattern Recognit., pages 3929–3938, 2017.
- Deep convolutional neural network for inverse problems in imaging. IEEE Trans. Image Process., 26(9):4509–4522, 2017.
- Beyond a Gaussian denoiser: Residual learning of deep CNN for image denoising. IEEE Trans. Image Process., 26(7):3142–3155, 2017.
- The little engine that could: Regularization by denoising (RED). SIAM J. Imaging Sci., 10(4):1804–1844, 2017.
- Plug-and-play priors for bright field electron tomography and sparse interpolation. IEEE Trans. Comput. Imag., 2(4):408–423, 2016.
- Plug-and-play methods for magnetic resonance imaging: Using denoisers for image recovery. IEEE Signal Process. Mag., 37(1):105–116, 2020.
- Image restoration and reconstruction using targeted plug-and-play priors. IEEE Trans. Comput. Imag., 5(4):675–686, 2019.
- Convex Analysis and Monotone Operator Theory in Hilbert Spaces. Springer, 2017.
- A convergent image fusion algorithm using scene-adapted Gaussian-mixture-based denoising. IEEE Trans. Image Process., 28(1):451–463, 2019.
- Plug-and-play ISTA converges with kernel denoisers. IEEE Signal Process. Lett., 27:610–614, 2020.
- On plug-and-play regularization using linear denoisers. IEEE Trans. Image Process., 30:4802–4813, 2021.
- Plug-and-play ADMM for image restoration: Fixed-point convergence and applications. IEEE Trans. Comput. Imag., 3(1):84–98, 2017.
- Interpolation and denoising of graph signals using plug-and-play ADMM. Proc. IEEE Int. Conf. Acoust. Speech Signal Process., pages 5431–5435, 2019.
- On exact and robust recovery for plug-and-play compressed sensing. Signal Processing, 211:109100, 2023.
- Graph signal denoising using nested-structured deep algorithm unrolling. Proc. IEEE Int. Conf. Acoust. Speech Signal Process., pages 5280–5284, 2021.
- Plug-and-play methods for integrating physical and learned models in computational imaging: Theory, algorithms, and applications. IEEE Signal Processing Magazine, 40(1):85–97, 2023.
- A non-local algorithm for image denoising. Proc. IEEE Conf. Comp. Vis. Pattern Recognit., 2:60–65, 2005.
- Plug-and-play regularization using linear solvers. IEEE Trans. Image Process., 31:6344–6355, 2022.
- Regularization by denoising via fixed-point projection (RED-PRO). SIAM J. Imaging Sci., 14(3):1374–1406, 2021.
- Fixed-point and objective convergence of plug-and-play algorithms. IEEE Trans. Comput. Imag., 7:337–348, 2021.
- Provable convergence of plug-and-play priors with MMSE denoisers. IEEE Signal Processing Letters, 27:1280–1284, 2020.
- Plug-and-play methods provably converge with properly trained denoisers. Proc. Intl. Conf. Mach. Learn., 97:5546–5557, 2019.
- T. Tirer and R. Giryes. Image restoration by iterative denoising and backward projections. IEEE Trans. Image Process., 28(3):1220–1234, 2018.
- Denoising prior driven deep neural network for image restoration. IEEE Trans. Pattern Anal. Mach. Intell., 41(10):2305–2318, 2018.
- Walter Rudin. Principles of Mathematical Analysis. McGraw-Hill, 1976.
- A. Beck and M. Teboulle. A fast iterative shrinkage-thresholding algorithm for linear inverse problems. SIAM J. Imaging Sci., 2(1):183–202, 2009.
- L. P. Yaroslavsky. Digital Picture Processing. Springer-Verlag, 1985.
- C. Tomasi and R. Manduchi. Bilateral filtering for gray and color images. Proc. IEEE Intl. Conf. Comp. Vis., pages 839–846, 1998.
- Kernel regression for image processing and reconstruction. IEEE Trans. Image Process., 16(2):349–366, 2007.
- H. Talebi and P. Milanfar. Global image denoising. IEEE Trans. Image Process., 23(2):755–768, 2013.
- C. D. Meyer. Matrix Analysis and Applied Linear Algebra. SIAM, 2000.
- S. Boyd and L. Vandenberghe. Convex Optimization. CU Press, 2004.
- Numerical Optimization: Theoretical and Practical Aspects. Springer Science & Business Media, 2006.
- P. Milanfar. Symmetrizing smoothing filters. SIAM J. Imaging Sci., 6(1):263–284, 2013.
- J-S. Lee. Digital image smoothing and the sigma filter. Comp. Vis. Graph. Image Process., 24(2):255–269, 1983.
- P. Milanfar. A tour of modern image filtering: new insights and methods, both practical and theoretical. IEEE Signal Process. Mag., 30(1):106–128, 2013.
- Gene H. Golub and Charles F. Van Loan. Matrix Computations. Johns Hopkins University Press, 1983.