Lightweight convolutional neural network for chest X-ray images classification

Sci Rep. 2024 Nov 30;14(1):29759. doi: 10.1038/s41598-024-80826-z.

Abstract

In this study, we developed a lightweight and rapid convolutional neural network (CNN) architecture for chest X-ray images; it primarily consists of a redesigned feature extraction (FE) module and multiscale feature (MF) module and validated using publicly available COVID-19 datasets. Experiments were conducted on multiple updated versions of the COVID-19 Radiography Database, a publicly accessible dataset on Kaggle. The database contained images categorized into three classes: COVID-19 coronavirus, viral or bacterial pneumonia, and normal. The results revealed that the proposed method achieved a training accuracy of 99.85% and a validation accuracy of 96.28% when detecting the three classes. In the test set, the optimal results were 96.03% accuracy for COVID-19, 97.10% accuracy for viral or bacterial pneumonia, and 97.86% accuracy for normal individuals. By reducing the computational requirements and improving the speed of the model, the proposed method can achieve real-time, low-error performance to help medical professionals with rapid diagnosis of COVID-19.

Keywords: COVID-19; Chest x-ray imaging; Computer-aided diagnosis; Convolutional neural networks; Lightweight architecture.

MeSH terms

  • COVID-19* / diagnostic imaging
  • COVID-19* / virology
  • Databases, Factual
  • Humans
  • Image Processing, Computer-Assisted / methods
  • Neural Networks, Computer*
  • Pneumonia, Bacterial / diagnostic imaging
  • Pneumonia, Viral / diagnostic imaging
  • Pneumonia, Viral / virology
  • Radiography, Thoracic / methods
  • SARS-CoV-2* / isolation & purification