Integrated ensemble CNN and explainable AI for COVID-19 diagnosis from CT scan and X-ray images

被引:3
作者
Rajpoot, Reenu [1 ]
Gour, Mahesh [1 ]
Jain, Sweta [1 ]
Semwal, Vijay Bhaskar [1 ]
机构
[1] Maulana Azad Natl Inst Technol, Dept Comp Sci & Engn, Bhopal 462003, India
关键词
Deep learning; Ensemble; Convolutional neural networks (CNN); Explainable AI; Chest X-ray images; CT scan images; COVID-19;
D O I
10.1038/s41598-024-75915-y
中图分类号
O [数理科学和化学]; P [天文学、地球科学]; Q [生物科学]; N [自然科学总论];
学科分类号
07 ; 0710 ; 09 ;
摘要
In light of the ongoing battle against COVID-19, while the pandemic may eventually subside, sporadic cases may still emerge, underscoring the need for accurate detection from radiological images. However, the limited explainability of current deep learning models restricts clinician acceptance. To address this issue, our research integrates multiple CNN models with explainable AI techniques, ensuring model interpretability before ensemble construction. Our approach enhances both accuracy and interpretability by evaluating advanced CNN models on the largest publicly available X-ray dataset, COVIDx CXR-3, which includes 29,986 images, and the CT scan dataset for SARS-CoV-2 from Kaggle, which includes a total of 2,482 images. We also employed additional public datasets for cross-dataset evaluation, ensuring a thorough assessment of model performance across various imaging conditions. By leveraging methods including LIME, SHAP, Grad-CAM, and Grad-CAM++, we provide transparent insights into model decisions. Our ensemble model, which includes DenseNet169, ResNet50, and VGG16, demonstrates strong performance. For the X-ray image dataset, sensitivity, specificity, accuracy, F1-score, and AUC are recorded at 99.00%, 99.00%, 99.00%, 0.99, and 0.99, respectively. For the CT image dataset, these metrics are 96.18%, 96.18%, 96.18%, 0.9618, and 0.96, respectively. Our methodology bridges the gap between precision and interpretability in clinical settings by combining model diversity with explainability, promising enhanced disease diagnosis and greater clinician acceptance.
引用
收藏
页数:21
相关论文
共 51 条
[1]   Generalizable disease detection using model ensemble on chest X-ray images [J].
Abad, Maider ;
Casas-Roma, Jordi ;
Prados, Ferran .
SCIENTIFIC REPORTS, 2024, 14 (01)
[2]   Hybrid Whale Optimization and Canonical Correlation based COVID-19 Classification Approach [J].
Abdellatef, Essam ;
Allah, M. I. Fath .
MULTIMEDIA TOOLS AND APPLICATIONS, 2024, 83 (26) :67421-67442
[3]   A hybridized feature extraction for COVID-19 multi-class classification on computed tomography images [J].
Abubakar, Hassana ;
Al-Turjman, Fadi ;
Ameen, Zubaida S. ;
Mubarak, Auwalu S. ;
Altrjman, Chadi .
HELIYON, 2024, 10 (05)
[4]   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
[5]  
Angelov P, 2020, medRxiv, DOI [10.1101/2020.04.24.20078584, 10.1101/2020.04.24.20078584, DOI 10.1101/2020.04.24.20078584]
[6]   A Decision Support System for Diagnosis of COVID-19 from Non-COVID-19 Influenza-like Illness Using Explainable Artificial Intelligence [J].
Chadaga, Krishnaraj ;
Prabhu, Srikanth ;
Bhat, Vivekananda ;
Sampathila, Niranjana ;
Umakanth, Shashikiran ;
Chadaga, Rajagopala .
BIOENGINEERING-BASEL, 2023, 10 (04)
[7]   Grad-CAM plus plus : Generalized Gradient-based Visual Explanations for Deep Convolutional Networks [J].
Chattopadhay, Aditya ;
Sarkar, Anirban ;
Howlader, Prantik ;
Balasubramanian, Vineeth N. .
2018 IEEE WINTER CONFERENCE ON APPLICATIONS OF COMPUTER VISION (WACV 2018), 2018, :839-847
[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]   Improving retinal OCT image classification accuracy using medical pre-training and sample replication methods [J].
Dai, Hao ;
Yang, Yaliang ;
Yue, Xian ;
Chen, Shen .
BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2024, 91
[10]   CovidConvLSTM: A fuzzy ensemble model for COVID-19 detection from chest X-rays [J].
Dey, Subhrajit ;
Bhattacharya, Rajdeep ;
Malakar, Samir ;
Schwenker, Friedhelm ;
Sarkar, Ram .
EXPERT SYSTEMS WITH APPLICATIONS, 2022, 206