MLCVLGIVOCJan 16, 2022

On Maximum-a-Posteriori estimation with Plug & Play priors and stochastic gradient descent

arXiv:2201.06133v129 citations
Originality Incremental advance
AI Analysis

This work addresses theoretical gaps in PnP methods for imaging inverse problems, offering more practical assumptions for researchers in computational imaging.

The paper tackles the theoretical analysis of Plug & Play (PnP) priors in Bayesian imaging, providing a convergence proof for MAP estimation using PnP stochastic gradient descent under realistic assumptions, with experimental demonstrations.

Bayesian methods to solve imaging inverse problems usually combine an explicit data likelihood function with a prior distribution that explicitly models expected properties of the solution. Many kinds of priors have been explored in the literature, from simple ones expressing local properties to more involved ones exploiting image redundancy at a non-local scale. In a departure from explicit modelling, several recent works have proposed and studied the use of implicit priors defined by an image denoising algorithm. This approach, commonly known as Plug & Play (PnP) regularisation, can deliver remarkably accurate results, particularly when combined with state-of-the-art denoisers based on convolutional neural networks. However, the theoretical analysis of PnP Bayesian models and algorithms is difficult and works on the topic often rely on unrealistic assumptions on the properties of the image denoiser. This papers studies maximum-a-posteriori (MAP) estimation for Bayesian models with PnP priors. We first consider questions related to existence, stability and well-posedness, and then present a convergence proof for MAP computation by PnP stochastic gradient descent (PnP-SGD) under realistic assumptions on the denoiser used. We report a range of imaging experiments demonstrating PnP-SGD as well as comparisons with other PnP schemes.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes