Detection of COVID-19 Patients from CT Scan and Chest X-ray Data Using Modified MobileNetV2 and LIME

被引:62
作者
Ahsan, Md Manjurul [1 ]
Nazim, Redwan [2 ]
Siddique, Zahed [3 ]
Huebner, Pedro [1 ]
机构
[1] Univ Oklahoma, Ind & Syst Engn, Norman, OK 73019 USA
[2] Univ Oklahoma, Chem Biol & Mat Engn, Norman, OK 73019 USA
[3] Univ Oklahoma, Sch Aerosp & Mech Engn, Norman, OK 73019 USA
关键词
chest X-ray; CT scan; coronavirus; COVID-19; deep learning; imbalanced data; mixed-data; SARS-CoV-2; small data; explainable AI;
D O I
10.3390/healthcare9091099
中图分类号
R19 [保健组织与事业(卫生事业管理)];
学科分类号
摘要
The COVID-19 global pandemic caused by the widespread transmission of the novel coronavirus (SARS-CoV-2) has become one of modern history's most challenging issues from a healthcare perspective. At its dawn, still without a vaccine, contagion containment strategies remained most effective in preventing the disease's spread. Patient isolation has been primarily driven by the results of polymerase chain reaction (PCR) testing, but its initial reach was challenged by low availability and high cost, especially in developing countries. As a means of taking advantage of a preexisting infrastructure for respiratory disease diagnosis, researchers have proposed COVID-19 patient screening based on the results of Chest Computerized Tomography (CT) and Chest Radiographs (X-ray). When paired with artificial-intelligence- and deep-learning-based approaches for analysis, early studies have achieved a comparatively high accuracy in diagnosing the disease. Considering the opportunity to further explore these methods, we implement six different Deep Convolutional Neural Networks (Deep CNN) models-VGG16, MobileNetV2, InceptionResNetV2, ResNet50, ResNet101, and VGG19-and use a mixed dataset of CT and X-ray images to classify COVID-19 patients. Preliminary results showed that a modified MobileNetV2 model performs best with an accuracy of 95 +/- 1.12% (AUC = 0.816). Notably, a high performance was also observed for the VGG16 model, outperforming several previously proposed models with an accuracy of 98.5 +/- 1.19% on the X-ray dataset. Our findings are supported by recent works in the academic literature, which also uphold the higher performance of MobileNetV2 when X-ray, CT, and their mixed datasets are considered. Lastly, we further explain the process of feature extraction using Local Interpretable Model-Agnostic Explanations (LIME), which contributes to a better understanding of what features in CT/X-ray images characterize the onset of COVID-19.
引用
收藏
页数:12
相关论文
共 39 条
[1]  
Ahsan M.M., 2018, REAL TIME FACE RECOG
[2]   Detecting SARS-CoV-2 From Chest X-Ray Using Artificial Intelligence [J].
Ahsan, Md Manjurul ;
Ahad, Md Tanvir ;
Soma, Farzana Akter ;
Paul, Shuva ;
Chowdhury, Ananna ;
Luna, Shahana Akter ;
Yazdan, Munshi Md. Shafwat ;
Rahman, Akhlaqur ;
Siddique, Zahed ;
Huebner, Pedro .
IEEE ACCESS, 2021, 9 :35501-35513
[3]  
[Anonymous], 2014, Comput. Sci.
[4]  
[Anonymous], 2017, ARXIV171104325
[5]   Covid-19: automatic detection from X-ray images utilizing transfer learning with convolutional neural networks [J].
Apostolopoulos, Ioannis D. ;
Mpesiana, Tzani A. .
PHYSICAL AND ENGINEERING SCIENCES IN MEDICINE, 2020, 43 (02) :635-640
[6]   Application of deep learning technique to manage COVID-19 in routine clinical practice using CT images: Results of 10 convolutional neural networks [J].
Ardakani, Ali Abbasian ;
Kanafi, Alireza Rajabzadeh ;
Acharya, U. Rajendra ;
Khadem, Nazanin ;
Mohammadi, Afshin .
COMPUTERS IN BIOLOGY AND MEDICINE, 2020, 121
[7]  
Bergstra J, 2012, J MACH LEARN RES, V13, P281
[8]   Explainable Deep Learning for Pulmonary Disease and Coronavirus COVID-19 Detection from X-rays [J].
Brunese, Luca ;
Mercaldo, Francesco ;
Reginelli, Alfonso ;
Santone, Antonella .
COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE, 2020, 196
[9]   Deep learning-based model for detecting 2019 novel coronavirus pneumonia on high-resolution computed tomography [J].
Chen, Jun ;
Wu, Lianlian ;
Zhang, Jun ;
Zhang, Liang ;
Gong, Dexin ;
Zhao, Yilin ;
Chen, Qiuxiang ;
Huang, Shulan ;
Yang, Ming ;
Yang, Xiao ;
Hu, Shan ;
Wang, Yonggui ;
Hu, Xiao ;
Zheng, Biqing ;
Zhang, Kuo ;
Wu, Huiling ;
Dong, Zehua ;
Xu, Youming ;
Zhu, Yijie ;
Chen, Xi ;
Zhang, Mengjiao ;
Yu, Lilei ;
Cheng, Fan ;
Yu, Honggang .
SCIENTIFIC REPORTS, 2020, 10 (01)
[10]  
Chollet F., 2017, Deep Learning With Python