Share your thoughts, 1 month free Claude Pro on usSee more
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 DeblurringFFHQ
PSNR27.5
34
Gaussian DeblurringImageNet
SSIM0.727
32
Motion DeblurringImageNet
SSIM0.931
27
Super-ResolutionImageNet
PSNR25.17
25
Motion DeblurringFFHQ
PSNR25.08
22
Inpainting (Random)FFHQ
PSNR30.69
17
Inpainting (Random)ImageNet
PSNR20.55
17
Showing 7 of 7 rows

Other info

Follow for update