Measurement-Consistent Langevin Corrector for Stabilizing Latent Diffusion Inverse Problem Solvers
About
While latent diffusion models (LDMs) have emerged as powerful priors for inverse problems, existing LDM-based solvers frequently suffer from instability. In this work, we first identify the instability as a discrepancy between the solver dynamics and stable reverse diffusion dynamics learned by the diffusion model, and show that reducing this gap stabilizes the solver. Building on this, we introduce \textit{Measurement-Consistent Langevin Corrector (MCLC)}, a theoretically grounded plug-and-play stabilization module that remedies the LDM-based inverse problem solvers through measurement-consistent Langevin updates. Compared to prior approaches that rely on linear manifold assumptions, which often fail to hold in latent space, MCLC provides a principled stabilization mechanism, leading to more stable and reliable behavior in latent space.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Inpainting | FFHQ | LPIPS0.169 | 32 | |
| HDR | FFHQ | PSNR25.55 | 25 | |
| HDR | ImageNet | PSNR24.79 | 21 | |
| Nonlinear Deblur | FFHQ | PSNR24.84 | 20 | |
| Gaussian deblur | ImageNet | PSNR25.89 | 19 | |
| Motion Deblur | ImageNet | PSNR24.94 | 17 | |
| Motion Deblur | FFHQ | PSNR27.45 | 17 | |
| Nonlinear Deblur | ImageNet | PSNR22.96 | 13 | |
| Gaussian deblur | FFHQ | PSNR28.14 | 12 | |
| Super-Resolution | ImageNet | PSNR26.25 | 12 |