A Transfer Learning Scheme for COVID-19 Diagnosis from Chest X-Ray Images Using Gradient-Weighted Class Activation Mapping

Ricardo Araguillin, Diego Maldonado, Felipe Grijalva, Diego S. Benítez, Noel Pérez-Pérez

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

This paper proposes a new method for detecting COVID-19 in chest X-ray images by comparing it with existing models. Our proposed deep learning model is customized for an accurate diagnosis of COVID-19. To improve model performance and prevent overfitting, we first apply data augmentation techniques. Unlike traditional image segmentation methods, we use gradient-weighted class activation mapping (Grad-CAM) to highlight regions critical to identifying COVID-19. We then used transfer learning of Xception convolutional neural networks to extract the X-ray image data into a compact feature set. Finally, we design, parameterize, and train the neural classification network. The network showed impressive results, achieving an astonishing 97% accuracy in identifying healthy patients. At the same time, its detection rate in COVID-19-infected patients was 92%, making it a worthy competitor compared to other detection models.

Original languageEnglish
Title of host publicationApplications of Computational Intelligence - 6th IEEE Colombian Conference, ColCACI 2023, Revised Selected Papers
EditorsAlvaro David Orjuela-Cañón, Jesus A Lopez, Julián David Arias-Londoño
PublisherSpringer Science and Business Media Deutschland GmbH
Pages3-18
Number of pages16
ISBN (Print)9783031484148
DOIs
StatePublished - 2024
Event6th IEEE Colombian Conference on Applications of Computational Intelligence, ColCACI 2023 - Bogota, Colombia
Duration: 26 Jul 202328 Jul 2023

Publication series

NameCommunications in Computer and Information Science
Volume1865 CCIS
ISSN (Print)1865-0929
ISSN (Electronic)1865-0937

Conference

Conference6th IEEE Colombian Conference on Applications of Computational Intelligence, ColCACI 2023
Country/TerritoryColombia
CityBogota
Period26/07/2328/07/23

Keywords

  • COVID-19
  • MobilNetV2
  • NNs
  • Xception
  • computer X-ray diagnostic tool
  • deep learning
  • transfer learning

Fingerprint

Dive into the research topics of 'A Transfer Learning Scheme for COVID-19 Diagnosis from Chest X-Ray Images Using Gradient-Weighted Class Activation Mapping'. Together they form a unique fingerprint.

Cite this