arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2506.22012eess.IVcs.CV

面向泛化性低剂量CT重建的噪声启发扩散模型

Noise-Inspired Diffusion Model for Generalizable Low-Dose CT Reconstruction

Institute of Science and Technology for Brain-inspired Intelligence(脑启发智能科学与技术研究院);Fudan University(复旦大学);School of Biomedical Engineering(生物医学工程学院);Southern Medical University(南方医科大学);School of Computer Science(计算机科学学院);Xi’an Jiaotong University(西安交通大学);School of Life Science and Technology(生命科学与技术学院);MOE Frontiers Center for Brain Science(教育部脑科学前沿中心);Key Laboratory of Computational Neuroscience and Brain-Inspired Intelligence (Ministry of Education)(教育部计算神经科学与脑启发智能重点实验室);State Key Laboratory of Brain Function and Disorders(脑功能与疾病国家重点实验室)

Qi Gao, Zhihao Chen, Dong Zeng, Junping Zhang, Jianhua Ma, Hongming Shan

AI总结:

提出一种针对各域噪声特征定制的扩散模型NEED,通过级联移位泊松扩散模型和双重引导扩散模型进行双域重建,仅需正常剂量数据训练即可在各种未见剂量水平上实现优异的低剂量CT重建与泛化性能。

摘要:

基于深度学习的低剂量计算机断层扫描(CT)重建模型在训练数据中未见过的剂量水平上的泛化能力非常重要,且仍然具有挑战性。以往的努力严重依赖配对数据,通过收集多样化的CT数据进行重新训练或少量测试数据进行微调,来提升泛化性能和鲁棒性。最近,扩散模型在低剂量CT(LDCT)重建中展现出了有前景且泛化的性能,然而,由于CT图像噪声偏离高斯分布以及来自含噪LDCT图像引导的不精确先验信息,它们可能会产生不真实的结构。在本文中,我们提出了一种用于泛化LDCT重建的噪声启发扩散模型,称为NEED,该模型针对每个域的噪声特征定制扩散模型。首先,我们提出了一种新颖的移位泊松扩散模型对投影数据进行去噪,该模型将扩散过程与对数预处理前LDCT投影中的噪声模型对齐。其次,我们设计了一种双重引导扩散模型来精炼重建图像,该模型利用LDCT图像和初始重建更准确地定位先验信息并增强重建保真度。通过级联这两个扩散模型进行双域重建,我们的NEED仅需正常剂量数据进行训练,并可通过时间步匹配策略在测试期间有效扩展到各种未见过的剂量水平。在两个数据集上的广泛定性、定量和基于分割的评估表明,我们的NEED在重建和泛化性能上始终优于现有最先进的方法。源代码可在 https://github.com/qgao21/NEED 获取。

英文摘要:

The generalization of deep learning-based low-dose computed tomography (CT) reconstruction models to doses unseen in the training data is important and remains challenging. Previous efforts heavily rely on paired data to improve the generalization performance and robustness through collecting either diverse CT data for re-training or a few test data for fine-tuning. Recently, diffusion models have shown promising and generalizable performance in low-dose CT (LDCT) reconstruction, however, they may produce unrealistic structures due to the CT image noise deviating from Gaussian distribution and imprecise prior information from the guidance of noisy LDCT images. In this paper, we propose a noise-inspired diffusion model for generalizable LDCT reconstruction, termed NEED, which tailors diffusion models for noise characteristics of each domain. First, we propose a novel shifted Poisson diffusion model to denoise projection data, which aligns the diffusion process with the noise model in pre-log LDCT projections. Second, we devise a doubly guided diffusion model to refine reconstructed images, which leverages LDCT images and initial reconstructions to more accurately locate prior information and enhance reconstruction fidelity. By cascading these two diffusion models for dual-domain reconstruction, our NEED requires only normal-dose data for training and can be effectively extended to various unseen dose levels during testing via a time step matching strategy. Extensive qualitative, quantitative, and segmentation-based evaluations on two datasets demonstrate that our NEED consistently outperforms state-of-the-art methods in reconstruction and generalization performance. Source code is made available at https://github.com/qgao21/NEED.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑