用扩散模型提升光学成像反演精度,还能量化不确定性。
Score-based diffusion models for diffuse optical tomography with uncertainty quantification
- 构建混合得分函数,融合数据驱动与物理模型优势
- 在模拟和实验数据上均实现低方差、高精度的后验采样
- 适合需要可靠反演结果的生物医学成像研究者
基于得分的扩散模型是一种新兴的贝叶斯反问题后验采样框架,通过从实测数据中学习强大先验分布,在严重不适定问题上表现优异。尽管在机器学习领域引发广泛关注,但在存在建模误差和真实物理测量数据情况下的实际反问题研究仍不充分。本文评估了无条件表示条件得分函数(UCoS)框架在线性化差分光学断层成像(DOT)中的应用。DOT利用近红外光边界测量来估计生物组织中吸收与散射参数的空间分布,该问题高度不适定,对噪声和建模误差敏感。我们提出一种新正则化方法,通过组合学习得分与模型基得分,防止得分函数过拟合。在模拟和实验测量数据上的验证表明,数据驱动先验可生成方差低、围绕真实值的后验样本,即使在高度不适定且存在建模误差的情况下仍表现良好。
原文摘要 · Abstract (English)
Score-based diffusion models are a recently developed framework for posterior sampling in Bayesian inverse problems with a state-of-the-art performance for severely ill-posed problems by leveraging a powerful prior distribution learned from empirical data. Despite generating significant interest especially in the machine-learning community, a thorough study of realistic inverse problems in the presence of modelling error and utilization of physical measurement data is still outstanding. In this work, the framework of unconditional representation for the conditional score function (UCoS) is evaluated for linearized difference imaging in diffuse optical tomography (DOT). DOT uses boundary measurements of near-infrared light to estimate the spatial distribution of absorption and scattering parameters in biological tissues. The problem is highly ill-posed and thus sensitive to noise and modelling errors. We introduce a novel regularization approach that prevents overfitting of the score function by constructing a mixed score composed of a learned and a model-based component. Validation of this approach is done using both simulated and experimental measurement data. The experiments demonstrate that a data-driven prior distribution results in posterior samples with low variance, compared to classical model-based estimation, and centred around the ground truth, even in the context of a highly ill-posed problem and in the presence of modelling errors.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。