Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Divide-and-Conquer Posterior Sampling for Denoising Diffusion Priors

About

Recent advancements in solving Bayesian inverse problems have spotlighted denoising diffusion models (DDMs) as effective priors. Although these have great potential, DDM priors yield complex posterior distributions that are challenging to sample. Existing approaches to posterior sampling in this context address this problem either by retraining model-specific components, leading to stiff and cumbersome methods, or by introducing approximations with uncontrolled errors that affect the accuracy of the produced samples. We present an innovative framework, divide-and-conquer posterior sampling, which leverages the inherent structure of DDMs to construct a sequence of intermediate posteriors that guide the produced samples to the target posterior. Our method significantly reduces the approximation error associated with current techniques without the need for retraining. We demonstrate the versatility and effectiveness of our approach for a wide range of Bayesian inverse problems. The code is available at \url{https://github.com/Badr-MOUFAD/dcps}

Yazid Janati, Badr Moufad, Alain Durmus, Eric Moulines, Jimmy Olsson• 2024

Related benchmarks

TaskDatasetResultRank
Gaussian DeblurringImageNet
PSNR26.08
16
Super-ResolutionImageNet
PSNR25.17
15
Motion DeblurringImageNet
PSNR33.03
10
Gaussian DeblurringFFHQ
PSNR27.5
10
Motion DeblurringFFHQ
PSNR25.08
10
Inpainting (Random)FFHQ
PSNR30.69
8
Inpainting (Random)ImageNet
PSNR20.55
8
Showing 7 of 7 rows

Other info

Follow for update