用投影数据引导扩散模型,提升低剂量CT重建质量
Projection Embedded Diffusion Bridge for CT Reconstruction from Incomplete Data
- 将投影数据嵌入反向SDE的得分函数,实现数据一致性约束
- 在稀疏视图、限角、截断投影下均超越现有最优模型
- 可调节随机性水平,适配不同噪声与分布偏移场景
从不完整投影数据中重建CT图像仍具挑战性,因其问题本质病态。扩散桥模型近期在从滤波反投影(FBP)结果恢复清晰图像方面展现出潜力,但如何融入数据一致性仍研究不足。数据一致性可提升重建保真度,使图像与观测投影对齐,并通过整合投影中的结构信息增强细节恢复。本文提出投影嵌入扩散桥(PEDB),引入新型反向随机微分方程(SDE),从同时依赖FBP重建和不完整投影数据的清洁图像分布中采样。通过在采样时显式条件化于投影数据,PEDB天然实现数据一致性。我们将投影数据嵌入反向SDE的得分函数中,在特定假设下推导出后验得分的可计算表达式。此外,引入自由参数控制反向过程中的随机性水平,并设计了减少离散化误差的求解方案。大量实验表明,PEDB在三种不完整数据类型——稀疏视图、限角、截断投影——上均表现优异,且在标准、含噪及域偏移评估中均超越所比较的最先进扩散桥模型。
原文摘要 · Abstract (English)
Reconstructing CT images from incomplete projection data remains challenging due to the ill-posed nature of the problem. Diffusion bridge models have recently shown promise in restoring clean images from their corresponding Filtered Back Projection (FBP) reconstructions, but incorporating data consistency into these models remains largely underexplored. Incorporating data consistency can improve reconstruction fidelity by aligning the reconstructed image with the observed projection data, and can enhance detail recovery by integrating structural information contained in the projections. In this work, we propose the Projection Embedded Diffusion Bridge (PEDB). PEDB introduces a novel reverse stochastic differential equation (SDE) to sample from the distribution of clean images conditioned on both the FBP reconstruction and the incomplete projection data. By explicitly conditioning on the projection data in sampling the clean images, PEDB naturally incorporates data consistency. We embed the projection data into the score function of the reverse SDE. Under certain assumptions, we derive a tractable expression for the posterior score. In addition, we introduce a free parameter to control the level of stochasticity in the reverse process. We also design a discretization scheme for the reverse SDE to mitigate discretization error. Extensive experiments demonstrate that PEDB achieves strong performance in CT reconstruction from three types of incomplete data, including sparse-view, limited-angle, and truncated projections. For each of these types, PEDB outperforms evaluated state-of-the-art diffusion bridge models across standard, noisy, and domain-shift evaluations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。