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

被引:30
作者
Raj, Alex Noel Joseph [1 ]
Zhu, Haipeng [1 ]
Khan, Asiya [2 ]
Zhuang, Zhemin [1 ]
Yang, Zengbiao [1 ]
Mahesh, Vijayalakshmi G. V. [3 ]
Karthik, Ganesan [4 ]
机构
[1] Shantou Univ, Coll Engn, Dept Elect Engn, Shantou, Peoples R China
[2] Univ Plymouth, Sch Engn Comp & Math, Plymouth, Devon, England
[3] BMS Inst Technol & Management, Dept Elect & Commun, Bangalore, Karnataka, India
[4] Madras Med Coll & Govt Gen Hosp, COVID CARE Inst Orthoped & Traumatol, Chennai, Tamil Nadu, India
基金
中国国家自然科学基金;
关键词
COVID-19 pulmonary infection; Dense network; Attention gate; Improved dilation convolution; UNET; Lung CT segmentation; CLASSIFICATION; PNEUMONIA; NOISE;
D O I
10.7717/peerj-cs.349
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
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.
引用
收藏
页数:34
相关论文
共 61 条
[1]  
Agarwal V, 2020, MED IMAGING 2020 COM, P356
[2]   Correlation of Chest CT and RT-PCR Testing for Coronavirus Disease 2019 (COVID-19) in China: A Report of 1014 Cases [J].
Ai, Tao ;
Yang, Zhenlu ;
Hou, Hongyan ;
Zhan, Chenao ;
Chen, Chong ;
Lv, Wenzhi ;
Tao, Qian ;
Sun, Ziyong ;
Xia, Liming .
RADIOLOGY, 2020, 296 (02) :E32-E40
[3]  
Almajalid R, 2019, 2018 17 IEEE INT C M
[4]   Recurrent residual U-Net for medical image segmentation [J].
Alom, Md Zahangir ;
Yakopcic, Chris ;
Hasan, Mahmudul ;
Taha, Tarek M. ;
Asari, Vijayan K. .
JOURNAL OF MEDICAL IMAGING, 2019, 6 (01)
[5]   Lung Pattern Classification for Interstitial Lung Diseases Using a Deep Convolutional Neural Network [J].
Anthimopoulos, Marios ;
Christodoulidis, Stergios ;
Ebner, Lukas ;
Christe, Andreas ;
Mougiakakou, Stavroula .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2016, 35 (05) :1207-1216
[6]   Semantic object classes in video: A high-definition ground truth database [J].
Brostow, Gabriel J. ;
Fauqueur, Julien ;
Cipolla, Roberto .
PATTERN RECOGNITION LETTERS, 2009, 30 (02) :88-97
[7]   An Integrated Micro- and Macroarchitectural Analysis of the Drosophila Brain by Computer-Assisted Serial Section Electron Microscopy [J].
Cardona, Albert ;
Saalfeld, Stephan ;
Preibisch, Stephan ;
Schmid, Benjamin ;
Cheng, Anchi ;
Pulokas, Jim ;
Tomancak, Pavel ;
Hartenstein, Volker .
PLOS BIOLOGY, 2010, 8 (10)
[8]  
Carlson, 2020, ARXIV200406673
[9]  
Chaganti S, 2020, ARXIV200401289V5
[10]  
Chen Xinlei, 2020, AUTOPHAGY, DOI DOI 10.1080/15548627.2020.1810918