Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Lee Hyoseok, Sohwi Lim, Eunju Cha, Tae-Hyun Oh• 2026

Related benchmarks

TaskDatasetResultRank
InpaintingFFHQ
LPIPS0.169
32
HDRFFHQ
PSNR25.55
25
HDRImageNet
PSNR24.79
21
Nonlinear DeblurFFHQ
PSNR24.84
20
Gaussian deblurImageNet
PSNR25.89
19
Motion DeblurImageNet
PSNR24.94
17
Motion DeblurFFHQ
PSNR27.45
17
Nonlinear DeblurImageNet
PSNR22.96
13
Gaussian deblurFFHQ
PSNR28.14
12
Super-ResolutionImageNet
PSNR26.25
12
Showing 10 of 12 rows

Other info

Follow for update