An Online Plug-and-Play Algorithm for Regularized Image Reconstruction

Abstract

Plug-and-play priors (PnP) is a powerful framework for regularizing imaging inverse problems by using advanced denoisers within an iterative algorithm. Recent experimental evidence suggests that PnP algorithms achieve state-of-the-art performance in a range of imaging applications. In this paper, we introduce a new online PnP algorithm based on the iterative shrinkage/thresholding algorithm (ISTA). The proposed algorithm uses only a subset of measurements at every iteration, which makes it scalable to very large datasets. We present a new theoretical convergence analysis, for both batch and online variants of PnP-ISTA, for denoisers that do not necessarily correspond to proximal operators. We also present simulations illustrating the applicability of the algorithm to image reconstruction in diffraction tomography. The results in this paper have the potential to expand the applicability of the PnP framework to very large and redundant datasets.

References in corpus (16)

• Message Passing Algorithms for Compressed Sensing
• Trainable Nonlinear Reaction Diffusion: A Flexible Framework for Fast and Effective Image Restoration
• Fast Image Recovery Using Variable Splitting and Constrained Optimization
• The dynamics of message passing on dense graphs, with applications to compressed sensing
• BM3D Frames and Variational Image Deblurring
• Plug-and-Play Priors for Bright Field Electron Tomography and Sparse Interpolation
• Learning Proximal Operators: Using Denoising Networks for Regularizing Inverse Imaging Problems
• Convergence Rates of Inexact Proximal-Gradient Methods for Convex Optimization
• Convolutional Dictionary Learning: A Comparative Review and New Algorithms
• Deconvolving Images with Unknown Boundaries Using the Alternating Direction Method of Multipliers
• Learning Deep CNN Denoiser Prior for Image Restoration
• On Convergence of Approximate Message Passing
• Plug-in Estimation in High-Dimensional Linear Inverse Problems: A Rigorous Analysis
• Image Restoration and Reconstruction using Variable Splitting and Class-adapted Image Priors
• Regularization by Denoising: Clarifications and New Interpretations
• signProx: One-Bit Proximal Algorithm for Nonconvex Stochastic Optimization

Cited by in corpus (55)

• Plug and play methods for magnetic resonance imaging (long version)
• Plug-and-Play Methods for Integrating Physical and Learned Models in Computational Imaging
• RARE: Image Reconstruction using Deep Priors Learned without Ground Truth
• Fixed Point Strategies in Data Science
• Learning to Optimize: A Primer and A Benchmark
• Super-Resolution via Image-Adapted Denoising CNNs: Incorporating External and Internal Learning
• Provable Convergence of Plug-and-Play Priors with MMSE denoisers
• SIMBA: Scalable Inversion in Optical Tomography using Deep Denoising Priors
• Convolutional Proximal Neural Networks and Plug-and-Play Algorithms
• Back-Projection based Fidelity Term for Ill-Posed Linear Inverse Problems
• Fixed-Point and Objective Convergence of Plug-and-Play Algorithms
• Parseval Proximal Neural Networks
• Plug-and-play ISTA converges with kernel denoisers
• Biomedical Image Reconstruction: From the Foundations to Deep Neural Networks
• Regularized Fourier Ptychography using an Online Plug-and-Play Algorithm
• On Plug-and-Play Regularization using Linear Denoisers
• The Practicality of Stochastic Optimization in Imaging Inverse Problems
• SGD-Net: Efficient Model-Based Deep Learning with Theoretical Guarantees
• Effective New Methods for Automated Parameter Selection in Regularized Inverse Problems
• Gradient Step Denoiser for convergent Plug-and-Play
• A New Recurrent Plug-and-Play Prior Based on the Multiple Self-Similarity Network
• CoIL: Coordinate-based Internal Learning for Imaging Inverse Problems
• On Maximum-a-Posteriori estimation with Plug & Play priors and stochastic gradient descent
• Learned convex regularizers for inverse problems
• On the Proof of Fixed-Point Convergence for Plug-and-Play ADMM
• Tuning-free Plug-and-Play Proximal Algorithm for Inverse Imaging Problems
• Provable Preconditioned Plug-and-Play Approach for Compressed Sensing MRI Reconstruction
• Solving Linear Inverse Problems Using the Prior Implicit in a Denoiser
• Deep Learning for Biomedical Image Reconstruction: A Survey
• End-to-end reconstruction meets data-driven regularization for inverse problems
• Plug-and-Play Regularization using Linear Solvers
• Designing Stable Neural Networks using Convex Analysis and ODEs
• Regularization by Denoising Sub-sampled Newton Method for Spectral CT Multi-Material Decomposition
• An ${\ell }^1$-Plug-and-Play Approach for MPI Using a Zero Shot Denoiser with Evaluation on the 3D Open MPI Dataset
• Monotone Lipschitz-Gradient Denoiser: Explainability of Operator Regularization Approaches Free From Lipschitz Constant Control
• SNIPS: Solving Noisy Inverse Problems Stochastically
• Recovery Analysis for Plug-and-Play Priors using the Restricted Eigenvalue Condition
• Async-RED: A Provably Convergent Asynchronous Block Parallel Stochastic Method using Deep Denoising Priors
• Scalable Plug-and-Play ADMM with Convergence Guarantees
• Joint Reconstruction and Calibration using Regularization by Denoising
• TFPnP: Tuning-free Plug-and-Play Proximal Algorithm with Applications to Inverse Imaging Problems
• A Fast Stochastic Plug-and-Play ADMM for Imaging Inverse Problems
• Plug-In Stochastic Gradient Method
• Regularization by Denoising: Clarifications and New Interpretations
• Inversion of Integral Models: a Neural Network Approach
• New Computational Techniques for a Faster Variation of BM3D Image Denoising
• Interpretation of Plug-and-Play (PnP) algorithms from a different angle
• Online Regularization by Denoising with Applications to Phase Retrieval
• Instabilities in Plug-and-Play (PnP) algorithms from a learned denoiser
• Plug-and-Play Priors for Reconstruction-based Placental Image Registration
• Joint ptycho-tomography with deep generative priors
• Boosting the Performance of Plug-and-Play Priors via Denoiser Scaling
• Infusing Learned Priors into Model-Based Multispectral Imaging
• A Machine-learning Based Initialization for Joint Statistical Iterative Dual-energy CT with Application to Proton Therapy
• Investigating Customization Strategies and Convergence Behaviors of Task-specific ADMM