arXiv:2602.20549cs.LGcs.CV2026-02被引 2

提出高效估计扩散模型先验证据的方法,用于图像逆问题的模型选择。

Sample-efficient evidence estimation of score based priors for model selection

  • 利用反向扩散采样过程中的中间样本,集成时间边际估计证据。
  • 仅需20个后验样本即可准确估算模型证据,显著降低计算开销。
  • 适用于高病态非线性逆问题,如黑洞成像,可识别先验不匹配。

先验的选择对解决病态成像逆问题至关重要,需选择与观测数据 $y$ 一致的先验以避免严重偏差。在贝叶斯逆问题中,可通过评估不同模型 $M$ 下的模型证据 $p(y mid M)$ 来选择最优先验,其中 $M$ 定义了先验结构。当前最先进的数据驱动先验方法是扩散模型,但直接计算扩散先验的模型证据是不可行的。现有证据估计算法通常需要大量未归一化先验密度点评估或精确的干净先验得分。本文提出 DiME,通过积分后验采样方法的时间边际来估计扩散先验的模型证据。该方法利用反向扩散采样过程中自然产生的大量中间样本,仅需少量后验样本(如20个)即可获得高精度证据估计。我们还展示了如何与最新的扩散后验采样方法协同实现。实验表明,当模型证据可解析计算时,我们的估计结果与其一致;且在多种高度病态、非线性逆问题中,能正确选择扩散模型先验并诊断先验不匹配,包括一个真实的黑洞成像问题。

原文摘要 · Abstract (English)

The choice of prior is central to solving ill-posed imaging inverse problems, making it essential to select one consistent with the measurements $y$ to avoid severe bias. In Bayesian inverse problems, this could be achieved by evaluating the model evidence $p(y \mid M)$ under different models $M$ that specify the prior and then selecting the one with the highest value. Diffusion models are the state-of-the-art approach to solving inverse problems with a data-driven prior; however, directly computing the model evidence with respect to a diffusion prior is intractable. Furthermore, most existing model evidence estimators require either many pointwise evaluations of the unnormalized prior density or an accurate clean prior score. We propose DiME, an estimator of the model evidence of a diffusion prior by integrating over the time-marginals of posterior sampling methods. Our method leverages the large amount of intermediate samples naturally obtained during the reverse diffusion sampling process to obtain an accurate estimation of the model evidence using only a handful of posterior samples (e.g., 20). We also demonstrate how to implement our estimator in tandem with recent diffusion posterior sampling methods. Empirically, our estimator matches the model evidence when it can be computed analytically, and it is able to both select the correct diffusion model prior and diagnose prior misfit under different highly ill-conditioned, non-linear inverse problems, including a real-world black hole imaging problem.

逆问题扩散模型模型选择证据估计

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