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

Geometry-Correct Diffusion Posterior Sampling with Denoiser-Pullback Curvature Guidance and Manifold-Aligned Damping

About

Diffusion posterior sampling conditions diffusion priors on measurements, but data-consistency updates are typically scaled by hand-tuned guidance weights and can destabilize sampling under stiff, operator-dependent curvature. We replace scalar guidance with a per-noise-level damped Gauss--Newton correction computed in diffusion-state coordinates. The correction pulls likelihood gradients back through the denoiser, uses a one-sided curvature model that avoids forward denoiser Jacobians, and applies diffusion-calibrated rank-one damping aligned with the denoiser residual. Each correction is solved with matrix-free GMRES using automatic differentiation, and sampling proceeds with a variance-preserving Langevin transition with a closed-form drift/noise split. On FFHQ and ImageNet across inverse problems, it achieves competitive PSNR/SSIM/LPIPS while running markedly faster than most of the compared baselines; on accelerated MRI reconstruction, it achieves the best PSNR/SSIM among the compared baselines.

Seunghyeok Shin, Minwoo Kim, Dabin Kim, Hongki Lim• 2026

Related benchmarks

TaskDatasetResultRank
Motion DeblurFFHQ--
56
Gaussian DeblurringFFHQ 256x256 (val)
LPIPS0.272
48
Phase RetrievalFFHQ--
30
4x super-resolutionFFHQ 256x256 (val)
LPIPS0.233
23
Motion DeblurImageNet 256x256 (val)
PSNR27.119
18
Inpainting (Random)FFHQ--
17
Box InpaintingImageNet 256 x 256 (val)
LPIPS0.342
15
High Dynamic RangeImageNet 256 x 256 (val)
PSNR28.885
13
Nonlinear DeblurringFFHQ 256 x 256 (val)
PSNR29.243
13
Random InpaintingFFHQ 256x256 (val)
PSNR31.433
12
Showing 10 of 32 rows

Other info

Follow for update