arXiv:2606.12710cs.LGmath.OC2026-06被引 1

提出稳定路径空间方法,提升扩散模型在逆问题中的采样精度与鲁棒性。

A Stabilized Path-Space Approach to Diffusion-Based Posterior Sampling

论文配图:A Stabilized Path-Space Approach to Diffusion-Based Posterior Sampling
图 1 · 摘自论文原文
  • 基于路径空间控制框架,将后验采样转化为学习可控随机过程。
  • 在多个基准逆问题上实现更优的采样准确性和不确定性量化。
  • 无需额外训练,可统一现有引导采样方法并提供误差修正机制。

扩散模型为贝叶斯逆问题提供了表达性强的数据驱动先验,但许多扩散后验采样器依赖启发式引导近似,在非线性算子和多模态后验下可能失效。本文提出一种稳定的路径空间框架用于扩散后验采样:从以先验分布为终态的基扩散过程出发,定义轨迹上的似然加权目标测度,并将后验采样转化为学习一个路径测度匹配该目标的受控随机过程。该形式将扩散后验采样与随机最优控制相联系,同时保持贝叶斯结构以支持不确定性量化。我们引入时间重参数化,通过消除未知初值函数引起的偏差使路径空间控制问题适定,且无需辅助训练。随后采用基于对数方差目标的信赖域路径空间优化方法学习控制策略。路径空间视角还统一了所提控制方法与现有基于引导的采样器,量化了近似控制带来的采样误差,并导出渐近精确后验期望的重要度采样修正。我们在一组具有解析表征或高质量参考后验的基准逆问题上评估该框架,实现对采样精度与不确定性量化的严格评估。实验揭示了扩散后验采样器的行为特性,并证明其在准确性与鲁棒性上优于主流方法。

原文摘要 · Abstract (English)

Diffusion models provide expressive data-driven priors for Bayesian inverse problems, but many diffusion posterior samplers rely on heuristic guidance approximations that can fail for nonlinear operators and multimodal posteriors. In this work, we develop a stabilized path-space framework for diffusion-based posterior sampling. Starting from a base diffusion process whose terminal marginal represents the prior, we define a likelihood-weighted target measure on trajectories and cast posterior sampling as learning a controlled stochastic process whose path measure matches this target. This formulation connects diffusion posterior sampling to stochastic optimal control while preserving the Bayesian structure needed for uncertainty quantification. We introduce a time reparameterization that makes the path-space control problem well posed by removing the bias induced by the unknown initial value function, without auxiliary training. We then learn the control via a trust-region path-space optimization method with log-variance objectives. The path-space perspective also unifies our learned control approach with existing guidance-based samplers, quantifies the sampling error induced by approximate controls, and yields importance sampling corrections for asymptotically exact posterior expectations. We evaluate the proposed framework on a suite of benchmark inverse problems with analytically characterized or high-quality reference posteriors, enabling principled assessment of sampling accuracy and uncertainty quantification. These experiments provide insight into the behavior of diffusion-based posterior samplers and demonstrate improved accuracy and robustness over leading approaches.

扩散模型贝叶斯推断逆问题采样优化

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