Explainable COVID-19 Detection Based on Chest X-rays Using an End-to-End RegNet Architecture

被引:10
作者
Chetoui, Mohamed [1 ]
Akhloufi, Moulay A. [1 ]
Bouattane, El Mostafa [2 ]
Abdulnour, Joseph [2 ]
Roux, Stephane [2 ]
Bernard, Chantal D'Aoust [2 ]
机构
[1] Univ Moncton, Dept Comp Sci, Percept Robot & Intelligent Machines PRIME, Moncton, NB E1A 3E9, Canada
[2] Montfort Acad Hosp, Inst Savoir Montfort, Ottawa, ON 61350, Canada
来源
VIRUSES-BASEL | 2023年 / 15卷 / 06期
基金
加拿大自然科学与工程研究理事会;
关键词
RegNet; convolutional neural networks; COVID-19; deep learning;
D O I
10.3390/v15061327
中图分类号
Q93 [微生物学];
学科分类号
071005 ; 100705 ;
摘要
COVID-19,which is caused by the severe acute respiratory syndrome coronavirus 2 (SARS-CoV-2), is one of the worst pandemics in recent history. The identification of patients suspected to be infected with COVID-19 is becoming crucial to reduce its spread. We aimed to validate and test a deep learning model to detect COVID-19 based on chest X-rays. The recent deep convolutional neural network (CNN) RegNetX032 was adapted for detecting COVID-19 from chest X-ray (CXR) images using polymerase chain reaction (RT-PCR) as a reference. The model was customized and trained on five datasets containing more than 15,000 CXR images (including 4148COVID-19-positive cases) and then tested on 321 images (150 COVID-19-positive) from Montfort Hospital. Twenty percent of the data from the five datasets were used as validation data for hyperparameter optimization. Each CXR image was processed by the model to detect COVID-19. Multi-binary classifications were proposed, such as: COVID-19 vs. normal, COVID-19 + pneumonia vs. normal, and pneumonia vs. normal. The performance results were based on the area under the curve (AUC), sensitivity, and specificity. In addition, an explainability model was developed that demonstrated the high performance and high generalization degree of the proposed model in detecting and highlighting the signs of the disease. The fine-tuned RegNetX032 model achieved an overall accuracy score of 96.0%, with an AUC score of 99.1%. The model showed a superior sensitivity of 98.0% in detecting signs from CXR images of COVID-19 patients, and a specificity of 93.0% in detecting healthy CXR images. A second scenario compared COVID-19 + pneumonia vs. normal (healthy X-ray) patients. The model achieved an overall score of 99.1% (AUC) with a sensitivity of 96.0% and specificity of 93.0% on the Montfort dataset. For the validation set, the model achieved an average accuracy of 98.6%, an AUC score of 98.0%, a sensitivity of 98.0%, and a specificity of 96.0% for detection (COVID-19 patients vs. healthy patients). The second scenario compared COVID-19 + pneumonia vs. normal patients. The model achieved an overall score of 98.8% (AUC) with a sensitivity of 97.0% and a specificity of 96.0%. This robust deep learning model demonstrated excellent performance in detecting COVID-19 from chest X-rays. This model could be used to automate the detection of COVID-19 and improve decision making for patient triage and isolation in hospital settings. This could also be used as a complementary aid for radiologists or clinicians when differentiating to make smart decisions.
引用
收藏
页数:15
相关论文
共 51 条
[1]   COVID-19 diagnosis using deep learning neural networks applied to CT images [J].
Akinyelu, Andronicus A. ;
Blignaut, Pieter .
FRONTIERS IN ARTIFICIAL INTELLIGENCE, 2022, 5
[2]   Detection and classification of COVID-19 disease from X-ray images using convolutional neural networks and histogram of oriented gradients [J].
Ayalew, Aleka Melese ;
Salau, Ayodeji Olalekan ;
Abeje, Bekalu Tadele ;
Enyew, Belay .
BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2022, 74
[3]  
BIMCV Medical Imaging Databank of the Valencia Region, 2020, OSF, DOI 10.17605/OSF.IO/NH7G8
[4]   Analysis and Forecasting Incidence, Intensive Care Unit Admissions, and Projected Mortality Attributable to COVID-19 in Portugal, the UK, Germany, Italy, and France: Predictions for 4 Weeks Ahead [J].
Carvalho, Kathleen ;
Vicente, Joao Paulo ;
Jakovljevic, Mihajlo ;
Teixeira, Joao Paulo Ramos .
BIOENGINEERING-BASEL, 2021, 8 (06)
[5]   Explainable Vision Transformers and Radiomics for COVID-19 Detection in Chest X-rays [J].
Chetoui, Mohamed ;
Akhloufi, Moulay A. .
JOURNAL OF CLINICAL MEDICINE, 2022, 11 (11)
[6]   Deep Efficient Neural Networks for Explainable COVID-19 Detection on CXR Images [J].
Chetoui, Mohamed ;
Akhloufi, Moulay A. .
ADVANCES AND TRENDS IN ARTIFICIAL INTELLIGENCE. ARTIFICIAL INTELLIGENCE PRACTICES, IEA/AIE 2021, PT I, 2021, 12798 :329-340
[7]  
Chollet F., 2015, Keras
[8]   Xception: Deep Learning with Depthwise Separable Convolutions [J].
Chollet, Francois .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :1800-1807
[9]   COVID-19 detection in CT and CXR images using deep learning models [J].
Chouat, Ines ;
Echtioui, Amira ;
Khemakhem, Rafik ;
Zouch, Wassim ;
Ghorbel, Mohamed ;
Ben Hamida, Ahmed .
BIOGERONTOLOGY, 2022, 23 (01) :65-84
[10]   Can AI Help in Screening Viral and COVID-19 Pneumonia? [J].
Chowdhury, Muhammad E. H. ;
Rahman, Tawsifur ;
Khandakar, Amith ;
Mazhar, Rashid ;
Kadir, Muhammad Abdul ;
Bin Mahbub, Zaid ;
Islam, Khandakar Reajul ;
Khan, Muhammad Salman ;
Iqbal, Atif ;
Al Emadi, Nasser ;
Reaz, Mamun Bin Ibne ;
Islam, Mohammad Tariqul .
IEEE ACCESS, 2020, 8 :132665-132676