Accelerated Bayesian imaging by relaxed proximal-point Langevin sampling

T Klatzer, P Dobson, Y Altmann, M Pereyra…�- SIAM Journal on Imaging�…, 2024 - SIAM
SIAM Journal on Imaging Sciences, 2024SIAM
This paper presents a new accelerated proximal Markov chain Monte Carlo methodology to
perform Bayesian inference in imaging inverse problems with an underlying convex
geometry. The proposed strategy takes the form of a stochastic relaxed proximal-point
iteration that admits two complementary interpretations. For models that are smooth or
regularized by Moreau–Yosida smoothing, the algorithm is equivalent to an implicit midpoint
discretization of an overdamped Langevin diffusion targeting the posterior distribution of�…
Abstract
This paper presents a new accelerated proximal Markov chain Monte Carlo methodology to perform Bayesian inference in imaging inverse problems with an underlying convex geometry. The proposed strategy takes the form of a stochastic relaxed proximal-point iteration that admits two complementary interpretations. For models that are smooth or regularized by Moreau–Yosida smoothing, the algorithm is equivalent to an implicit midpoint discretization of an overdamped Langevin diffusion targeting the posterior distribution of interest. This discretization is asymptotically unbiased for Gaussian targets and shown to converge in an accelerated manner for any target that is -strongly log-concave (i.e., requiring in the order of iterations to converge, similar to accelerated optimization schemes), comparing favorably to Pereyra, Vargas Mieles, and Zygalakis [SIAM J. Imaging Sci., 13 (2020), pp.�905–935], which is only provably accelerated for Gaussian targets and has bias. For models that are not smooth, the algorithm is equivalent to a Leimkuhler–Matthews discretization of a Langevin diffusion targeting a Moreau–Yosida approximation of the posterior distribution of interest and hence achieves a significantly lower bias than conventional unadjusted Langevin strategies based on the Euler–Maruyama discretization. For targets that are -strongly log-concave, the provided nonasymptotic convergence analysis also identifies the optimal time step, which maximizes the convergence speed. The proposed methodology is demonstrated through a range of experiments related to image deconvolution with Gaussian and Poisson noise with assumption-driven and data-driven convex priors. Source codes for the numerical experiments of this paper are available from https://github.com/MI2G/accelerated-langevin-imla.
Society for Industrial and Applied Mathematics
Showing the best result for this search. See all results