Comparative Analysis of Deep Learning Networks for COVID-19 and Pneumonia Identification: Grad-CAM Visualization of Chest X-Ray Images
This study evaluates deep learning networks for diagnosing COVID-19 and pneumonia from chest X-rays (CXR). 22 convolutional neural network (CNN) models were trained on a dataset of 10,530 CXR images from Kaggle, categorized as normal, pneumonia, or COVID-19. There were 9 CNN models removed from our investigation as they had poor performance in disease classification. The top 10 performing models were tested on a separate dataset. DenseNet201 achieved the highest F1-score of 97.57% and balanced accuracy of 97.98%, but required highest computational time. In contrast, ResNet18 demonstrated the fastest computational time while maintaining 89.54% F1-score and 91.52 % balanced accuracy. Gradient-weighted Class Activation Mapping (Grad-CAM) visualization was employed to enhance model interpretability and validate predictions. The study highlights the trade-off between model performance and computational efficiency in AI-driven diagnostic tools for respi-ratory diseases. While these models show promise for clinical applications like disease monitoring and treatment planning, limitations such as potential overdiagnosis were noted. Future research should focus on validating model performance in real-world clinical settings and assessing impact on patient outcomes.
Authors
- Pichit Boonkrong (ORCID: https://orcid.org/0000-0001-5105-0460)
- Suejit Pechprasarn (ORCID: https://orcid.org/0000-0001-9105-8627)
- Sasithorn Tengjongdee
- Chanapa Chaitan
- Sasipatcha Hanmanop
- Tatpol Jongsiri
- Manatsanan Khongtan
Institutions
- Rangsit University (TH)
Publication Details
- Published
- 2024-11-21
- DOI
- https://doi.org/10.1109/bmeicon64021.2024.10896306
- Citations
- 2
- Primary Topic
- COVID-19 diagnosis using AI
- Type
- article
- Field-Weighted Citation Impact
- 0.81