A Scalable Solution for Pneumonia Diagnosis: Transfer Learning for Chest X-ray Analysis
The high death rate and rapid spread of pneumonia make it a serious global health concern. The precision of traditional diagnostic techniques is often compromised, particularly in resource-limited settings. Existing studies have predominantly focused on either binary or multi-class classification, often relying on limited datasets, which affects the generalizability and robustness of models. Additionally, the interpretability and transparency of deep learning models remain challenging in clinical environments. Addressing these challenges is crucial for advancing diagnostic methodologies to combat pneumonia. This study explores the use of transfer learning to improve the reliability of pneumonia classification using chest X-ray images. An automated classification system was developed, utilizing a combination of image preprocessing techniques, feature extraction with GLRLM, and various machine learning and transfer learning algorithms. Two datasets were employed: the GWCMCx dataset with 5856 chest X-rays and another dataset comprising 9207 chest X-rays. The system was designed for both binary and multi-class classification scenarios. CapsuleNet emerged as the most effective model, achieving an accuracy of ${9 9. 0 2 \%}$ in binary classification and $90.43 \%$ in multi-class classification, significantly outperforming previous studies. Additionally, a user-friendly web application was developed to facilitate the integration of the classification system into clinical workflows, accelerating clinical decision-making. This study demonstrates the potential of transfer learning to automate and improve pneumonia diagnosis, offering a scalable solution that can alleviate radiologists’ workload and improve patient outcomes.
Authors
- Keshav Kaushik (ORCID: https://orcid.org/0000-0003-3777-765X)
- Chandan Prasad (ORCID: https://orcid.org/0009-0001-0185-4564)
- Rezaul Haque (ORCID: https://orcid.org/0000-0002-9922-8632)
- Abdullah Al Sakib
- Sadid A. Hasan (ORCID: https://orcid.org/0000-0002-5665-7752)
- S Z Sultana (ORCID: https://orcid.org/0009-0006-3345-5118)
- Neeru Malik
- Hasib Fardin
Institutions
- American University (US)
- American InterContinental University (US)
- Savitribai Phule Pune University (IN)
- East West University (BD)
Publication Details
- Published
- 2024-09-18
- DOI
- https://doi.org/10.1109/ic3i61595.2024.10829132
- Citations
- 8
- Primary Topic
- COVID-19 diagnosis using AI
- Type
- article
- Field-Weighted Citation Impact
- 3.23