ADID-UNET-a segmentation model for COVID-19 infection from lung CT scans

PeerJ Comput Sci. 2021 Jan 26:7:e349. doi: 10.7717/peerj-cs.349. eCollection 2021.

Abstract

Currently, the new coronavirus disease (COVID-19) is one of the biggest health crises threatening the world. Automatic detection from computed tomography (CT) scans is a classic method to detect lung infection, but it faces problems such as high variations in intensity, indistinct edges near lung infected region and noise due to data acquisition process. Therefore, this article proposes a new COVID-19 pulmonary infection segmentation depth network referred as the Attention Gate-Dense Network- Improved Dilation Convolution-UNET (ADID-UNET). The dense network replaces convolution and maximum pooling function to enhance feature propagation and solves gradient disappearance problem. An improved dilation convolution is used to increase the receptive field of the encoder output to further obtain more edge features from the small infected regions. The integration of attention gate into the model suppresses the background and improves prediction accuracy. The experimental results show that the ADID-UNET model can accurately segment COVID-19 lung infected areas, with performance measures greater than 80% for metrics like Accuracy, Specificity and Dice Coefficient (DC). Further when compared to other state-of-the-art architectures, the proposed model showed excellent segmentation effects with a high DC and F1 score of 0.8031 and 0.82 respectively.

Keywords: Attention gate; COVID-19 pulmonary infection; Dense network; Improved dilation convolution; Lung CT segmentation; UNET.

Associated data

  • figshare/10.6084/m9.figshare.13521488.v2

Grants and funding

This work is supported by The Scientific Research Grant of Shantou University, China (No: NTF17016), The National Natural Science Foundation of China (No. 82071992), The Key Project of Guangdong Province Science and Technology Plan (No. 2015B020233018) and Guangdong Province University Priority Field (Artificial Intelligence) Project (No. 2019KZDZX1013). The funders had no role in study design, data collection and analysis, decision to publish, or preparation of the manuscript.