arXiv:2410.09945stat.MLcs.LG2024-10ICLR被引 34

提出新采样方法,提升扩散模型在贝叶斯反问题中的推理效率。

Variational Diffusion Posterior Sampling with Midpoint Guidance

  • 基于中点引导的变分扩散后验采样,分解转移过程降低复杂度。
  • 在非线性反问题与潜在扩散模型上验证,重构精度显著提升。
  • 适用于多模态数据,已用于心电图不完整数据修复,具临床潜力。

扩散模型作为先验在解决贝叶斯反问题中展现出巨大潜力,但其去噪后验分布的采样仍面临不可计算项的挑战。现有方法将问题转化为从目标后验的代理扩散模型采样,并将得分分解为先验得分与不可计算的引导项。前者由预训练扩散模型的得分替代,而引导项需估计。本文提出一种新方法,通过新的转移过程分解,实现引导项复杂度与先验转移复杂度之间的权衡。在各类线性与非线性反问题上进行广泛实验,包括使用潜在扩散模型作为先验的复杂情况。进一步展示了该方法在多模态数据上的适用性及其在公共健康领域的前景:通过重建不完整心电图,助力心血管疾病诊断。代码已公开于 exttt{https://github.com/yazidjanati/mgps}。

原文摘要 · Abstract (English)

Diffusion models have recently shown considerable potential in solving Bayesian inverse problems when used as priors. However, sampling from the resulting denoising posterior distributions remains a challenge as it involves intractable terms. To tackle this issue, state-of-the-art approaches formulate the problem as that of sampling from a surrogate diffusion model targeting the posterior and decompose its scores into two terms: the prior score and an intractable guidance term. While the former is replaced by the pre-trained score of the considered diffusion model, the guidance term has to be estimated. In this paper, we propose a novel approach that utilises a decomposition of the transitions which, in contrast to previous methods, allows a trade-off between the complexity of the intractable guidance term and that of the prior transitions. We validate the proposed approach through extensive experiments on linear and nonlinear inverse problems, including challenging cases with latent diffusion models as priors. We then demonstrate its applicability to various modalities and its promising impact on public health by tackling cardiovascular disease diagnosis through the reconstruction of incomplete electrocardiograms. The code is publicly available at \url{https://github.com/yazidjanati/mgps}.

扩散模型贝叶斯推断反问题医疗影像

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