CT-SDM: A Sampling Diffusion Model for Sparse-View CT Reconstruction across All Sampling Rates

Liutao Yang,Jiahao Huang,Guang Yang,Daoqiang Zhang
2024-09-03
Abstract:Sparse views X-ray computed tomography has emerged as a contemporary technique to mitigate radiation dose. Because of the reduced number of projection views, traditional reconstruction methods can lead to severe artifacts. Recently, research studies utilizing deep learning methods has made promising progress in removing artifacts for Sparse-View Computed Tomography (SVCT). However, given the limitations on the generalization capability of deep learning models, current methods usually train models on fixed sampling rates, affecting the usability and flexibility of model deployment in real clinical settings. To address this issue, our study proposes a adaptive reconstruction method to achieve high-performance SVCT reconstruction at any sampling rate. Specifically, we design a novel imaging degradation operator in the proposed sampling diffusion model for SVCT (CT-SDM) to simulate the projection process in the sinogram domain. Thus, the CT-SDM can gradually add projection views to highly undersampled measurements to generalize the full-view sinograms. By choosing an appropriate starting point in diffusion inference, the proposed model can recover the full-view sinograms from any sampling rate with only one trained model. Experiments on several datasets have verified the effectiveness and robustness of our approach, demonstrating its superiority in reconstructing high-quality images from sparse-view CT scans across various sampling rates.
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
This paper attempts to address the issues of generality and flexibility in sparse-view CT (SVCT) reconstruction under different sampling rates. Specifically, existing deep learning-based methods are typically trained and tested at fixed sampling rates, which leads to performance degradation when these methods face different sampling rates in practical clinical applications. The paper proposes a sampling rate adaptive method based on a diffusion model (CT-SDM), aiming to achieve high-quality SVCT image reconstruction at any sampling rate with a single model. ### Main Issues 1. **Limitations of Existing Methods**: - Existing deep learning-based SVCT reconstruction methods are usually trained at specific sampling rates, resulting in poor generalization ability at different sampling rates. - This limitation makes it difficult for these methods to flexibly respond to different scanning needs in practical clinical applications, as the sampling rate may vary due to patient conditions, equipment limitations, or diagnostic requirements. 2. **Specific Problems to be Solved**: - How to design an SVCT reconstruction method that can adapt to any sampling rate to improve the model's generalization ability and practicality. - How to maintain high-quality image reconstruction at different sampling rates, reducing artifacts and noise. ### Solution 1. **Proposed Method**: - **CT-SDM (Sampling Diffusion Model)**: By replacing the traditional image degradation process with projection view sampling, a new imaging degradation operator is designed. The model gradually adds projection views in the forward process, thereby gradually recovering the full-view sinogram in the reverse process. - **Grouped Random Sampling Strategy**: Dynamically adjust the sampling rate and select projection views during training to ensure diversity and uniformity of sampling, prevent bias, and provide data augmentation. 2. **Technical Details**: - **Forward Process**: Determine the undersampled CT measurements (i.e., sinogram) at each sampling step t, with each sampling step corresponding to a specific sampling rate αt. - **Reverse Process**: Starting from a certain initial sampling rate αt, gradually recover the full-view sinogram y0. By iterating the reverse process, gradually improve the reconstruction quality. - **Network Training**: Includes two CNNs for reconstruction tasks in the sinogram domain and image domain, respectively. By jointly training these two networks, optimize the loss function and improve overall performance. ### Experimental Validation 1. **Datasets**: - Used low-dose CT image and projection datasets (LDCT Datasets) and TCGA-KIRC dataset for experimental validation. 2. **Evaluation Metrics**: - Used Peak Signal-to-Noise Ratio (PSNR), Structural Similarity Index (SSIM), and Learned Perceptual Image Patch Similarity (LPIPS) to evaluate reconstruction quality. 3. **Experimental Results**: - The proposed CT-SDM method demonstrated excellent performance and stability at different sampling rates, significantly outperforming other methods. Especially at non-training sampling rates, CT-SDM was still able to maintain high-quality image reconstruction, showcasing its strong generalization ability. ### Conclusion The paper proposes a sampling rate adaptive method based on a diffusion model (CT-SDM), achieving high-quality SVCT image reconstruction at any sampling rate with a single model. This method not only improves the model's generalization ability but also greatly expands its applicability and practicality in real clinical applications.