用扩散模型先验提升低剂量CT重建的泛化能力,应对设备和人体差异带来的挑战。
Cross-Distribution Diffusion Priors-Driven Iterative Reconstruction for Sparse-View CT
- 融合跨分布扩散先验与迭代重建,通过条件无引导训练学习通用图像特征。
- 在不同扫描仪和患者数据上测试,对分布外样本的重建质量显著优于现有方法。
- 适合需要高鲁棒性低剂量CT重建的临床场景,尤其适用于设备或人群差异大的情况。
稀疏视角计算机断层成像(SVCT)可提高时间分辨率并降低辐射剂量,但因视角减少及扫描仪、成像协议或解剖差异导致的域偏移,常出现伪影,使分布外(OOD)场景下性能下降。本文提出交叉分布扩散先验驱动的迭代重建框架(CDPIR),将来自可扩展插值Transformer(SiT)的跨分布扩散先验与基于模型的迭代重建结合。具体地,训练一个扩展自扩散Transformer(DiT)架构的SiT骨干网络,构建统一的随机插值框架,利用分类器无引导(CFG)在多个数据集上进行训练。通过随机丢弃条件输入(以空嵌入替代),模型同时学习域特异性和域不变性先验,提升泛化能力。采样时,基于全局敏感的Transformer扩散模型在统一随机插值框架内利用跨分布先验,灵活稳定地控制多分布到噪声的插值路径,并实现解耦采样策略,从而增强对分布外重建的适应性。通过交替执行数据保真度更新与采样更新,模型在稀疏视角重建中达到当前最优性能,且细节保留更佳。大量实验表明,CDPIR显著优于现有方法,尤其在分布外条件下表现突出,凸显其在复杂成像场景中的鲁棒性与临床潜力。
原文摘要 · Abstract (English)
Sparse-View CT (SVCT) reconstruction enhances temporal resolution and reduces radiation dose, yet its clinical use is hindered by artifacts due to view reduction and domain shifts from scanner, protocol, or anatomical variations, leading to performance degradation in out-of-distribution (OOD) scenarios. In this work, we propose a Cross-Distribution Diffusion Priors-Driven Iterative Reconstruction (CDPIR) framework to tackle the OOD problem in SVCT. CDPIR integrates cross-distribution diffusion priors, derived from a Scalable Interpolant Transformer (SiT), with model-based iterative reconstruction methods. Specifically, we train a SiT backbone, an extension of the Diffusion Transformer (DiT) architecture, to establish a unified stochastic interpolant framework, leveraging Classifier-Free Guidance (CFG) across multiple datasets. By randomly dropping the conditioning with a null embedding during training, the model learns both domain-specific and domain-invariant priors, enhancing generalizability. During sampling, the globally sensitive transformer-based diffusion model exploits the cross-distribution prior within the unified stochastic interpolant framework, enabling flexible and stable control over multi-distribution-to-noise interpolation paths and decoupled sampling strategies, thereby improving adaptation to OOD reconstruction. By alternating between data fidelity and sampling updates, our model achieves state-of-the-art performance with superior detail preservation in SVCT reconstructions. Extensive experiments demonstrate that CDPIR significantly outperforms existing approaches, particularly under OOD conditions, highlighting its robustness and potential clinical value in challenging imaging scenarios.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。