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

Alex Noel Joseph Raj, Haipeng Zhu, Asiya Khan, Zhemin Zhuang, Zengbiao Yang, Vijayalakshmi G. V. Mahesh, Ganesan Karthik
2021 PeerJ Computer Science  
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-
more » ... proved 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.
doi:10.7717/peerj-cs.349 pmid:33816999 pmcid:PMC7924694 fatcat:2tmaagjsuzemdiactjy7jna4hu