揭示扩散模型后验采样失败的根源,提供精准诊断工具
When, why, and how do diffusion posterior samplers fail? A finite-sample lens
- 从有限样本视角出发,构建可任意逼近真实后验的采样方法
- 发现现有方法常高估或低估后验分布宽度,导致伪影与模式误判
- 无需非线性模型或多模态后验,仅需多模态先验即可引发错误
扩散模型在建模自然数据复杂分布方面表现优异,已广泛用于成像逆问题中的后验采样。现有方法可在推理时灵活适配任意测量模型,但为保证计算可行性,必须在中间时间步使用似然函数的近似。尽管这些近似在实践中常有效,其对最终采样后验的影响尚不明确,可能导致无法解释的失败。本文提出一种有限样本视角下的后验采样分析框架,当训练集规模趋于无穷时,可将后验近似至任意精度,适用于任意前向模型和先验分布。基于此框架,我们发现主流后验采样近似在中间步骤常低估或高估后验分布的扩散程度,进而导致对早期停止时间敏感、后验模式相对权重不准,以及生成不属于后验的先验模式或不受先验支持的似然模式等现象。更重要的是,这些错误的根源并非依赖非线性测量模型或多模态后验,仅由多模态先验与中间采样阶段的后验扩散估计不准即可引发。该有限样本后验采样方法对似然近似类型和前向模型类型(线性或非线性)均无偏,可作为即插即用的诊断工具,评估现有及未来后验采样器的准确性和失效模式。
原文摘要 · Abstract (English)
Diffusion models have excellent capacity to model complex distributions of natural data, which has made them a popular and effective choice for posterior sampling in imaging inverse problems. Existing methods can incorporate any measurement model at inference time but must use an inexact approximation for the likelihood at intermediate timesteps for computational tractability. Although these approximations can often work well empirically, their downstream effect on the sampled posterior is poorly understood and can result in unexplained failures. To understand when, why, and how these likelihood approximations propagate to erroneous posterior distributions, we introduce a finite-sample perspective on posterior sampling that approximates the posterior to arbitrary precision as training set size tends towards infinity, for any forward model and prior distribution. Using this finite-sample lens, we observe that popular posterior sampling approximations tend to under- or over-estimate the spread of the posterior at intermediate timesteps, causing downstream consequences including sensitivity to early stopping time, inaccurate relative weighting of posterior modes, and hallucination, both of prior modes that are not in the posterior and likelihood modes that are not supported by the prior. Moreover, we find that the cause of these posterior errors requires neither a nonlinear measurement model nor a multimodal posterior, but can arise solely due to a multimodal prior and inaccurate posterior spread at intermediate sampling times. Our finite-sample posterior sampling approach is agnostic to the type of likelihood approximation and the type of (linear or nonlinear) forward model, and can thus serve as a drop-in diagnostic to evaluate the accuracy and failure modes of existing and future posterior samplers.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。