Exact Posterior Score Estimation for Solving Linear Inverse Problems
About
Diffusion and flow-based models learn powerful data priors by training a denoiser to reverse Gaussian corruption. To use this prior to solve a linear inverse problem, one needs to sample from the posterior, but the score that the prior provides is the unconditional score, not the posterior score. Existing methods either steer a fixed pretrained denoiser with approximate measurement-matching corrections, or train a conditional restoration model that abandons the denoising structure of the prior. We derive the exact posterior score in closed form for linear Gaussian inverse problems under general Gaussian interpolants, and show that posterior sampling reduces to a denoising problem at an operator-dependent shifted pivot under an anisotropic noise covariance. We turn this identity into Exact Posterior Score (EPS), a denoising training objective that preserves the input/output structure of standard pretraining and can therefore be trained from scratch or fine-tuned from a pretrained denoiser. At inference, EPS uses the same sampler as the underlying backbone, with no likelihood gradients or projections. We evaluate EPS on five linear inverse problems across FFHQ and ImageNet, where it outperforms training-free and training-based baselines on fidelity, perceptual, and distributional metrics, while using roughly an order of magnitude fewer denoiser evaluations than gradient-based posterior samplers.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Gaussian Deblurring | ImageNet 256 x 256 | PSNR25.9 | 34 | |
| Super-res (4x) | ImageNet 256x256 | PSNR22.65 | 20 | |
| Inpaint (box) | ImageNet 256x256 | PSNR22.13 | 17 | |
| Inpaint (random) | ImageNet 256x256 | PSNR29.73 | 16 | |
| Motion Deblur | ImageNet 256x256 | PSNR24.39 | 15 | |
| Box inpaint | FFHQ 64 | PSNR26.01 | 9 | |
| Box Inpainting | ImageNet 64 | PSNR23.6 | 9 | |
| Gaussian deblur | FFHQ 64 | PSNR31.47 | 9 | |
| Gaussian Deblurring | ImageNet 64 | PSNR29.68 | 9 | |
| Motion Deblurring | ImageNet 64 | PSNR28.17 | 9 |