An XAI approach for COVID-19 detection using transfer learning with X-ray images

被引:15
作者
Sarp, Salih [1 ]
Catak, Ferhat Ozgur [2 ]
Kuzlu, Murat [3 ]
Cali, Umit [4 ]
Kusetogullari, Huseyin [5 ]
Zhao, Yanxiao [1 ]
Ates, Gungor [6 ]
Guler, Ozgur [7 ]
机构
[1] Virginia Commonwealth Univ, Elect & Comp Engn, Richmond, VA USA
[2] Univ Stavanger, Dept Elect Engn & Comp Sci, Rogaland, Norway
[3] Old Dominion Univ, Batten Coll Engn & Technol, Norfolk, VA USA
[4] Norwegian Univ Sci & Technol, Dept Elect Power Engn, Trondheim, Norway
[5] Blekinge Inst Technol, Dept Comp Sci, Karlskrona, Sweden
[6] Private Genesis Hosp, Dept Pulm Med, Diyarbakir, Turkiye
[7] eKare Inc, Fairfax, VA USA
关键词
COVID-19; Explainable artificial intelligence; Transfer learning; CXR;
D O I
10.1016/j.heliyon.2023.e15137
中图分类号
O [数理科学和化学]; P [天文学、地球科学]; Q [生物科学]; N [自然科学总论];
学科分类号
07 ; 0710 ; 09 ;
摘要
The coronavirus disease (COVID-19) has continued to cause severe challenges during this unprecedented time, affecting every part of daily life in terms of health, economics, and social development. There is an increasing demand for chest X-ray (CXR) scans, as pneumonia is the primary and vital complication of COVID-19. CXR is widely used as a screening tool for lung -related diseases due to its simple and relatively inexpensive application. However, these scans require expert radiologists to interpret the results for clinical decisions, i.e., diagnosis, treatment, and prognosis. The digitalization of various sectors, including healthcare, has accelerated during the pandemic, with the use and importance of Artificial Intelligence (AI) dramatically increasing. This paper proposes a model using an Explainable Artificial Intelligence (XAI) technique to detect and interpret COVID-19 positive CXR images. We further analyze the impact of COVID-19 positive CXR images using heatmaps. The proposed model leverages transfer learning and data augmentation techniques for faster and more adequate model training. Lung segmentation is applied to enhance the model performance further. We conducted a pre-trained network comparison with the highest classification performance (F1-Score: 98%) using the ResNet model.
引用
收藏
页数:12
相关论文
共 61 条
[1]  
Abid H., 2020, CURR MED RES PRACT
[2]   COVID-CAPS: A capsule network-based framework for identification of COVID-19 cases from X-ray images [J].
Afshar, Parnian ;
Heidarian, Shahin ;
Naderkhani, Farnoosh ;
Oikonomou, Anastasia ;
Plataniotis, Konstantinos N. ;
Mohammadi, Arash .
PATTERN RECOGNITION LETTERS, 2020, 138 :638-643
[3]   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
[4]   Adoption of COVID-19 triage strategies for low-income settings [J].
Ayebare, Rodgers R. ;
Flick, Robert ;
Okware, Solome ;
Bodo, Bongomin ;
Lamorde, Mohammed .
LANCET RESPIRATORY MEDICINE, 2020, 8 (04) :E22-E22
[5]   Classification of Brain Tumors from MRI Images Using a Convolutional Neural Network [J].
Badza, Milica M. ;
Barjaktarovic, Marko C. .
APPLIED SCIENCES-BASEL, 2020, 10 (06)
[6]  
Bandyopadhyay S. K., 2020, MEDRXIV
[7]   Deep learning with non-medical training used for chest pathology identification [J].
Bar, Yaniv ;
Diamant, Idit ;
Wolf, Lior ;
Greenspan, Hayit .
MEDICAL IMAGING 2015: COMPUTER-AIDED DIAGNOSIS, 2015, 9414
[8]  
Batista A. F. d. M., 2020, MEDRXIV
[9]  
Catak F.O., 2022, COMPUTATIONAL INTELL, P71, DOI 10.1007/978-981-16-3783-4_4
[10]  
Cohen JP, 2020, COVID-19 image data col- lection