arXiv:2606.17048cs.LGcs.CV2026-06被引 1

提出精确后验得分估计方法,用预训练去噪器高效求解线性逆问题。

Exact Posterior Score Estimation for Solving Linear Inverse Problems

论文配图:Exact Posterior Score Estimation for Solving Linear Inverse Problems
图 1 · 摘自论文原文
  • 推导出线性高斯逆问题的闭式后验得分表达式。
  • 在多个数据集上优于基线方法,且推理时仅需少量去噪器调用。
  • 兼容现有去噪结构,可从零训练或微调预训练模型。

扩散和流模型通过训练去噪器逆转高斯扰动来学习强大的数据先验。为用此先验解决线性逆问题,需从后验采样,但先验提供的是无条件得分,而非后验得分。现有方法或以近似测量匹配修正固定预训练去噪器,或训练放弃去噪结构的条件恢复模型。本文推导了在一般高斯插值下线性高斯逆问题的闭式后验得分,并表明后验采样等价于在算子相关偏移点处进行带各向异性噪声协方差的去噪问题。据此提出精确后验得分(EPS),一种保留标准预训练输入输出结构的去噪训练目标,可从头训练或微调预训练去噪器。推理时,EPS 使用与底层骨干相同的采样器,无需似然梯度或投影。在 FFHQ 与 ImageNet 上五个线性逆问题上的评估显示,其在保真度、感知和分布指标上均优于训练无关与训练依赖基线,且去噪器调用次数比梯度基采样器少约一个数量级。

原文摘要 · Abstract (English)

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.

去噪器逆问题扩散模型后验采样

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。