针对低剂量CT重建泛化难题,提出新型噪声引导扩散模型。
Noise-Inspired Diffusion Model for Generalizable Low-Dose CT Reconstruction
- 设计双阶段扩散模型,分别处理投影数据与图像重建中的非高斯噪声。
- 仅需正常剂量数据训练,即可在未知低剂量水平上实现优异重建效果。
- 适合需要跨剂量泛化的医学影像重建研究者使用。
基于深度学习的低剂量计算机断层扫描(CT)重建模型在未见剂量下的泛化能力至关重要但依然困难。以往方法依赖配对数据通过多样化数据重训练或少量测试数据微调来提升性能。尽管扩散模型在低剂量CT重建中表现良好且具有泛化潜力,但其可能因CT图像噪声偏离高斯分布及噪声图像引导信息不准确而产生不真实结构。本文提出一种噪声引导的扩散模型NEED,针对各数据域的噪声特性进行定制。首先,提出一种新的移位泊松扩散模型以去噪预对数低剂量投影数据,使扩散过程与预对数低剂量投影中的噪声模型对齐。其次,设计双重引导扩散模型以优化重建图像,利用低剂量图像和初始重建结果更准确地定位先验信息,提升重建保真度。通过级联两个扩散模型实现双域重建,NEED仅需正常剂量数据训练,并可通过时间步匹配策略有效扩展至多种未见剂量水平。在两个数据集上的定性、定量及分割评估均表明,该方法在重建质量和泛化性能上优于现有最先进方法。源代码已公开于https://github.com/qgao21/NEED。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。