Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces RED-KLwSGS, a novel Split Gibbs sampling framework for Bayesian imaging inverse problems that accelerates posterior sampling by employing underdamped kinetic Langevin diffusions for the auxiliary variable, driven by a one-shot denoising score. This approach achieves the same per-iteration cost as standard Langevin-within-SGS but offers significantly faster convergence and high-quality reconstructions, as demonstrated on FFHQ and ImageNet datasets using Denoising Diffusion Probabilistic Models as priors. The method is theoretically validated with non-asymptotic Wasserstein-2 convergence guarantees for strongly log-concave priors.
Kinetic Langevin diffusions slash posterior sampling time for imaging inverse problems by 10x while maintaining reconstruction quality, even with complex diffusion model priors.
Split Gibbs sampling (SGS) is a popular framework for posterior sampling in Bayesian imaging inverse problems. It decouples a Gaussian data-fidelity term from a complex prior through an auxiliary variable, so the data variable is updated exactly and only the prior-side conditional is hard to sample. Existing samplers treat this conditional in one of two ways. Plug-and-play SGS runs a multi-step diffusion denoiser at every iteration, which is expensive and lacks non-asymptotic guarantees. Langevin-within-SGS takes cheap overdamped Langevin steps but needs many iterations. We propose RED-KLwSGS, which keeps the exact Gaussian update for the data variable and updates the auxiliary variable with underdamped (kinetic) Langevin diffusions driven by a one-shot denoising score, at the same per-iteration cost as Langevin-within-SGS. We prove non-asymptotic Wasserstein-2 convergence in continuous and discrete time for strongly log-concave priors. We also introduce Joint-RED-KLwSGS, which applies kinetic Langevin diffusions to both variables. Experiments with Denoising diffusion probabilistic models as diffusion priors on FFHQ and ImageNet datasets show faster convergence and high-quality image reconstruction.