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 条
  • [41] AI support for accurate and fast radiological diagnosis of COVID-19: an international multicenter, multivendor CT study
    Fanyang Meng
    Jonathan Kottlors
    Rahil Shahzad
    Haifeng Liu
    Philipp Fervers
    Yinhua Jin
    Miriam Rinneburger
    Dou Le
    Mathilda Weisthoff
    Wenyun Liu
    Mengzhe Ni
    Ye Sun
    Liying An
    Xiaochen Huai
    Dorottya Móré
    Athanasios Giannakis
    Isabel Kaltenborn
    Andreas Bucher
    David Maintz
    Lei Zhang
    Frank Thiele
    Mingyang Li
    Michael Perkuhn
    Huimao Zhang
    Thorsten Persigehl
    European Radiology, 2023, 33 : 4280 - 4291
  • [42] Quantifying prognosis severity of COVID-19 patients from deep learning based analysis of CT chest images
    Rana, Ashish
    Singh, Harpreet
    Mavuduru, Ravimohan
    Pattanaik, Smita
    Rana, Prashant Singh
    MULTIMEDIA TOOLS AND APPLICATIONS, 2022, 81 (13) : 18129 - 18153
  • [43] Prediction of COVID-19 from Chest CT Images Using an Ensemble of Deep Learning Models
    Biswas, Shreya
    Chatterjee, Somnath
    Majee, Arindam
    Sen, Shibaprasad
    Schwenker, Friedhelm
    Sarkar, Ram
    APPLIED SCIENCES-BASEL, 2021, 11 (15):
  • [44] DEEP ACTIVE LEARNING FOR FIBROSIS SEGMENTATION OF CHEST CT SCANS FROM COVID-19 PATIENTS
    Liu, Xiaohong
    Wang, Kai
    Chen, Ting
    2021 IEEE INTERNATIONAL CONFERENCE ON IMAGE PROCESSING (ICIP), 2021, : 175 - 179
  • [45] A Deep Learning Ensemble Approach for Automated COVID-19 Detection from Chest CT Images
    Zazzaro, Gaetano
    Martone, Francesco
    Romano, Gianpaolo
    Pavone, Luigi
    JOURNAL OF CLINICAL MEDICINE, 2021, 10 (24)
  • [46] An ensemble learning method based on ordinal regression for COVID-19 diagnosis from chest CT
    Guo, Xiaodong
    Lei, Yiming
    He, Peng
    Zeng, Wenbing
    Yang, Ran
    Ma, Yinjin
    Feng, Peng
    Lyu, Qing
    Wang, Ge
    Shan, Hongming
    PHYSICS IN MEDICINE AND BIOLOGY, 2021, 66 (24)
  • [47] Online diagnosis of COVID-19 from chest radiography images by using deep learning algorithms
    Cafer Budak
    Vasfiye Mençik
    Osman Varışlı
    Neural Computing and Applications, 2023, 35 : 20717 - 20734
  • [48] VGG-FusionNet: A Feature Fusion Framework from CT scan and Chest X-ray Images based Deep Learning for COVID-19 Detection
    Zhou, Yihan
    Liu, Jiali
    Yang, Zekun
    Liu, Tianyi
    Meng, Xinyue
    Zhou, Zhuocheng
    Anaissi, Ali
    Braytee, Ali
    2022 IEEE INTERNATIONAL CONFERENCE ON DATA MINING WORKSHOPS, ICDMW, 2022, : 349 - 357
  • [49] Online diagnosis of COVID-19 from chest radiography images by using deep learning algorithms
    Budak, Cafer
    Mencik, Vasfiye
    Varisli, Osman
    NEURAL COMPUTING & APPLICATIONS, 2023, 35 (28) : 20717 - 20734
  • [50] Automatic diagnosis of COVID-19 from CT images using CycleGAN and transfer learning
    Ghassemi, Navid
    Shoeibi, Afshin
    Khodatars, Marjane
    Heras, Jonathan
    Rahimi, Alireza
    Zare, Assef
    Zhang, Yu-Dong
    Pachori, Ram Bilas
    Gorriz, Manuel
    APPLIED SOFT COMPUTING, 2023, 144