DESDGAN

This repo provides the source code of our paper: A Dual-Encoder-Single-Decoder Based Low-Dose CT Denoising Network https://ieeexplore.ieee.org/stamp/stamp.jsp?tp=&arnumber=9726904

@article{han2022dual, title={A Dual-Encoder-Single-Decoder Based Low-dose CT Denoising Network}, author={Han, Zefang and Shangguan, Hong and Zhang, Xiong and Zhang, Pengcheng and Cui, Xueying and Ren, Huiying}, journal={IEEE Journal of Biomedical and Health Informatics}, year={2022}, publisher={IEEE} }

Abstract

Generative adversarial networks (GAN) have shown great potential for image quality improvement in low-dose CT (LDCT). In general, the shallow features of generator include more shallow visual information such as edges and texture, while the deep features of generator contain more deep semantic information such as organization structure. To improve the network’s ability to categorically deal with different kinds of information, this paper proposes a new type of GAN with dual-encoder single-decoder structure. In the structure of the generator, firstly, a pyramid non-local attention module in the main encoder channel is designed to improve the feature extraction effectiveness by enhancing the features with self-similarity; Secondly, another encoder with shallow feature processing module and deep feature processing module is proposed to improve the encoding capabilities of the generator; Finally, the final denoised CT image is generated by fusing main encoder’s features, shallow visual features, and deep semantic features. The quality of the generated images is improved due to the use of feature complementation in the generator. In order to improve the adversarial training ability of discriminator, a hierarchical-split ResNet structure is proposed, which improves the feature’s richness and reduces the feature’s redundancy in discriminator. The experimental results show that compared with the traditional single-encoder- single-decoder based GAN, the proposed method performs better in both image quality and medical diagnostic acceptability. Code is available in https://github.com/hanzefang/DESDGAN.

0. Dependencies

Create a conda environment with python version = 3.6 pytorch

1. dataset

Mayo dataset：Mayo dataset piglet：http://homepage.usask.ca/protect%20$elax%20sim%20$xiy525/ download mayo dataset from the link //www.aapm.org/GrandChallenge/LowDoseCT/ convert .dcm to .png

2. Training

python train.py --dataset facades --cuda ###--dataset need to be modified.

3. Evaluation

python test.py --dataset facades --cuda ###--dataset need to be modified.

4. Reference