Leveraging deep transfer learning and explainable AI for accurate COVID-19 diagnosis: Insights from a multi-national chest CT scan study

被引:0
|
作者
Pham, Nhat Truong [1 ]
Ko, Jinsol [2 ,3 ]
Shah, Masaud [2 ]
Rakkiyappan, Rajan [4 ]
Woo, Hyun Goo [2 ,3 ,5 ]
Manavalan, Balachandran [1 ]
机构
[1] Department of Integrative Biotechnology, College of Biotechnology and Bioengineering, Sungkyunkwan University, Gyeonggi-do, Suwon
[2] Department of Physiology, Ajou University School of Medicine, Suwon
[3] Department of Biomedical Science, Graduate School, Ajou University, Suwon
[4] Department of Mathematics, Bharathiar University, Tamil Nadu, Coimbatore
[5] Ajou Translational Omics Center (ATOC), Ajou University Medical Center
基金
新加坡国家研究基金会; 美国国家卫生研究院;
关键词
Chest computed tomography scan; Convolutional neural networks; COVID-19; detection; Deep transfer learning; Explainable artificial intelligence; Hyperparameter optimization;
D O I
10.1016/j.compbiomed.2024.109461
中图分类号
学科分类号
摘要
The COVID-19 pandemic has emerged as a global health crisis, impacting millions worldwide. Although chest computed tomography (CT) scan images are pivotal in diagnosing COVID-19, their manual interpretation by radiologists is time-consuming and potentially subjective. Automated computer-aided diagnostic (CAD) frameworks offer efficient and objective solutions. However, machine or deep learning methods often face challenges in their reproducibility due to underlying biases and methodological flaws. To address these issues, we propose XCT-COVID, an explainable, transferable, and reproducible CAD framework based on deep transfer learning to predict COVID-19 infection from CT scan images accurately. This is the first study to develop three distinct models within a unified framework by leveraging a previously unexplored large dataset and two widely used smaller datasets. We employed five known convolutional neural network architectures, both with and without pretrained weights, on the larger dataset. We optimized hyperparameters through extensive grid search and 5-fold cross-validation (CV), significantly enhancing the model performance. Experimental results from the larger dataset showed that the VGG16 architecture (XCT-COVID-L) with pretrained weights consistently outperformed other architectures, achieving the best performance, on both 5-fold CV and independent test. When evaluated with the external datasets, XCT-COVID-L performed well with data with similar distributions, demonstrating its transferability. However, its performance significantly decreased on smaller datasets with lower-quality images. To address this, we developed other models, XCT-COVID-S1 and XCT-COVID-S2, specifically for the smaller datasets, outperforming existing methods. Moreover, eXplainable Artificial Intelligence (XAI) analyses were employed to interpret the models’ functionalities. For prediction and reproducibility purposes, the implementation of XCT-COVID is publicly accessible at https://github.com/cbbl-skku-org/XCT-COVID/. © 2024 Elsevier Ltd
引用
收藏
相关论文
共 50 条
  • [21] Artificial Intelligence and COVID-19 Using Chest CT Scan and Chest X-ray Images: Machine Learning and Deep Learning Approaches for Diagnosis and Treatment
    Fusco, Roberta
    Grassi, Roberta
    Granata, Vincenza
    Setola, Sergio Venanzio
    Grassi, Francesca
    Cozzi, Diletta
    Pecori, Biagio
    Izzo, Francesco
    Petrillo, Antonella
    JOURNAL OF PERSONALIZED MEDICINE, 2021, 11 (10):
  • [22] DETECTING COVID-19 AND COMMUNITY ACQUIRED PNEUMONIA USING CHEST CT SCAN IMAGES WITH DEEP LEARNING
    Chaudhary, Shubham
    Sadbhawna
    Jakhetiya, Vinit
    Subudhi, Badri N.
    Baid, Ujjwal
    Guntuku, Sharath Chandra
    2021 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH AND SIGNAL PROCESSING (ICASSP 2021), 2021, : 8583 - 8587
  • [24] COVIDScreen: explainable deep learning framework for differential diagnosis of COVID-19 using chest X-rays
    Singh, Rajeev Kumar
    Pandey, Rohan
    Babu, Rishie Nandhan
    NEURAL COMPUTING & APPLICATIONS, 2021, 33 (14) : 8871 - 8892
  • [25] COVIDScreen: explainable deep learning framework for differential diagnosis of COVID-19 using chest X-rays
    Rajeev Kumar Singh
    Rohan Pandey
    Rishie Nandhan Babu
    Neural Computing and Applications, 2021, 33 : 8871 - 8892
  • [26] AI aiding in diagnosing, tracking recovery of COVID-19 using deep learning on Chest CT scans
    Maheshwar Kuchana
    Amritesh Srivastava
    Ronald Das
    Justin Mathew
    Atul Mishra
    Kiran Khatter
    Multimedia Tools and Applications, 2021, 80 : 9161 - 9175
  • [27] AI aiding in diagnosing, tracking recovery of COVID-19 using deep learning on Chest CT scans
    Kuchana, Maheshwar
    Srivastava, Amritesh
    Das, Ronald
    Mathew, Justin
    Mishra, Atul
    Khatter, Kiran
    MULTIMEDIA TOOLS AND APPLICATIONS, 2021, 80 (06) : 9161 - 9175
  • [28] Deep transfer learning-based automated detection of COVID-19 from lung CT scan slices
    Ahuja, Sakshi
    Panigrahi, Bijaya Ketan
    Dey, Nilanjan
    Rajinikanth, Venkatesan
    Gandhi, Tapan Kumar
    APPLIED INTELLIGENCE, 2021, 51 (01) : 571 - 585
  • [29] Deep transfer learning-based automated detection of COVID-19 from lung CT scan slices
    Sakshi Ahuja
    Bijaya Ketan Panigrahi
    Nilanjan Dey
    Venkatesan Rajinikanth
    Tapan Kumar Gandhi
    Applied Intelligence, 2021, 51 : 571 - 585
  • [30] Detection of COVID-19 from chest X-rays using deep transfer learning
    Vo, Tri-Nhan
    Le, Ngoc-Bich
    Phan, Quoc-Hung
    Le, Thanh-Hai
    Pham, Thi-Thu-Hien
    HEALTH INFORMATICS JOURNAL, 2024, 30 (04)