Diffusion Probabilistic Priors for Zero-Shot Low-Dose CT Image Denoising

Xuan Liu,Yaoqin Xie,Jun Cheng,Songhui Diao,Shan Tan,Xiaokun Liang
2023-07-14
Abstract:Denoising low-dose computed tomography (CT) images is a critical task in medical image computing. Supervised deep learning-based approaches have made significant advancements in this area in recent years. However, these methods typically require pairs of low-dose and normal-dose CT images for training, which are challenging to obtain in clinical settings. Existing unsupervised deep learning-based methods often require training with a large number of low-dose CT images or rely on specially designed data acquisition processes to obtain training data. To address these limitations, we propose a novel unsupervised method that only utilizes normal-dose CT images during training, enabling zero-shot denoising of low-dose CT images. Our method leverages the diffusion model, a powerful generative model. We begin by training a cascaded unconditional diffusion model capable of generating high-quality normal-dose CT images from low-resolution to high-resolution. The cascaded architecture makes the training of high-resolution diffusion models more feasible. Subsequently, we introduce low-dose CT images into the reverse process of the diffusion model as likelihood, combined with the priors provided by the diffusion model and iteratively solve multiple maximum a posteriori (MAP) problems to achieve denoising. Additionally, we propose methods to adaptively adjust the coefficients that balance the likelihood and prior in MAP estimations, allowing for adaptation to different noise levels in low-dose CT images. We test our method on low-dose CT datasets of different regions with varying dose levels. The results demonstrate that our method outperforms the state-of-the-art unsupervised method and surpasses several supervised deep learning-based methods. Codes are available in <a class="link-external link-https" href="https://github.com/DeepXuan/Dn-Dp" rel="external noopener nofollow">this https URL</a>.
Image and Video Processing,Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The paper primarily focuses on addressing the problem of denoising low-dose computed tomography (CT) images. Specifically, the research team proposes a novel unsupervised method aimed at overcoming some limitations of existing deep learning approaches: 1. **No need for paired data**: Traditional supervised learning methods require paired data of low-dose and normal-dose CT images for training, but obtaining such datasets in clinical practice is very challenging. The method proposed in this paper only uses normal-dose CT images for training, without the need for low-dose CT images. 2. **Zero-sample denoising**: The proposed method can achieve zero-sample denoising of low-dose CT images, meaning it can effectively denoise such images without having seen any low-dose CT images. 3. **Utilizing diffusion models**: This method leverages a powerful generative model—the diffusion model. By training a cascaded unconditional diffusion model to generate high-quality normal-dose CT images, and further introducing low-dose CT images into the reverse process of the diffusion model, the denoising purpose is achieved. 4. **Adaptive weight adjustment**: To adapt to low-dose CT images with different noise levels, the research team also proposes an adaptive adjustment method to balance the weight coefficients of likelihood and prior information in the maximum a posteriori estimation. In summary, the main contribution of this paper is the proposal of a novel unsupervised learning method that can effectively denoise low-dose CT images using only normal-dose CT images, demonstrating high practical value and innovation. Additionally, the method is specifically optimized for images with different noise levels, enhancing the algorithm's generalization capability.