Plug-and-Play Split Gibbs Sampler: Embedding Deep Generative Priors in Bayesian Inference - Signal et Communications
Article Dans Une Revue IEEE Transactions on Image Processing Année : 2024

Plug-and-Play Split Gibbs Sampler: Embedding Deep Generative Priors in Bayesian Inference

Résumé

This paper introduces a stochastic plug-and-play (PnP) sampling algorithm that leverages variable splitting to efficiently sample from a posterior distribution. The algorithm based on split Gibbs sampling (SGS) draws inspiration from the alternating direction method of multipliers (ADMM). It divides the challenging task of posterior sampling into two simpler sampling problems. The first problem depends on the likelihood function, while the second is interpreted as a Bayesian denoising problem that can be readily carried out by a deep generative model. Specifically, for an illustrative purpose, the proposed method is implemented in this paper using state-of-the-art diffusion-based generative models. Akin to its deterministic PnP-based counterparts, the proposed method exhibits the great advantage of not requiring an explicit choice of the prior distribution, which is rather encoded into a pre-trained generative model. However, unlike optimization methods (e.g., PnP-ADMM) which generally provide only point estimates, the proposed approach allows conventional Bayesian estimators to be accompanied by confidence intervals at a reasonable additional computational cost. Experiments on commonly studied image processing problems illustrate the efficiency of the proposed sampling strategy. Its performance is compared to recent state-of-the-art optimization and sampling methods.
Fichier principal
Vignette du fichier
Coeurdoux_IEEE_Trans_IP_2024.pdf (5.3 Mo) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-04602765 , version 1 (05-06-2024)

Identifiants

Citer

Florentin Coeurdoux, Nicolas Dobigeon, Pierre Chainais. Plug-and-Play Split Gibbs Sampler: Embedding Deep Generative Priors in Bayesian Inference. IEEE Transactions on Image Processing, 2024, 33, pp.3496-3507. ⟨10.1109/TIP.2024.3404338⟩. ⟨hal-04602765⟩
166 Consultations
85 Téléchargements

Altmetric

Partager

More