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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Motion Deblur | FFHQ | -- | 56 | |
| Gaussian Deblurring | FFHQ 256x256 (val) | LPIPS0.272 | 48 | |
| Phase Retrieval | FFHQ | -- | 30 | |
| 4x super-resolution | FFHQ 256x256 (val) | LPIPS0.233 | 23 | |
| Motion Deblur | ImageNet 256x256 (val) | PSNR27.119 | 18 | |
| Inpainting (Random) | FFHQ | -- | 17 | |
| Box Inpainting | ImageNet 256 x 256 (val) | LPIPS0.342 | 15 | |
| High Dynamic Range | ImageNet 256 x 256 (val) | PSNR28.885 | 13 | |
| Nonlinear Deblurring | FFHQ 256 x 256 (val) | PSNR29.243 | 13 | |
| Random Inpainting | FFHQ 256x256 (val) | PSNR31.433 | 12 |