A Convolutional Neural Network (CNN) Based Framework for Enhanced Diagnosis and Classification of COVID-19 Pneumonia
DOI:
https://doi.org/10.21015/vtcs.v12i2.1999Abstract
COVID-19 pneumonia is a persistent worldwide health problem that usually affects the most vulnerable groups in society: the newborn and aged populations. Most of the current endeavors toward handling diagnosis and classification of pneumonia have used numerous techniques for machine learning and deep learning, with a particular focus on COVID-19 pneumonia. However, most of these techniques have raised concerns with regard to diagnostic precision as a result of the limited application of advanced algorithms, datasets whose validation is mostly inadequate and predictive capability. To address these limitations, our research introduces a deep learning-based approach by Convolutional Neural Networks (CNNs), which enhances the performance in classifying COVID-19 pneumonia. Salient features of the proposed method include a four-step process: first, data acquisition from a comprehensive chest X-ray dataset on GitHub; second, processing and analyzing the data through visual means like histograms and scatter plots; third, using CNNs supplemented with techniques for data augmentation in supervised learning; lastly, performance evaluation to benchmark against existing models. The present study uses features from X-ray images with the help of CNN's advanced pattern recognition capabilities in pursuit of achieving better generalization and precision in classification. The model achieved an accuracy of 85.70\% and precision of 88.6%, which surpasses the existing techniques and thereby built the promise of improving the accuracy of the diagnostic process, hence, leading to improved care for the patients, and more so forms the foundation on which future AI-powered healthcare diagnostics are expected to take off.
References
F. Shi et al., "Large-Scale Screening to Distinguish between COVID-19 and Community-Acquired Pneumonia Using Infection Size-Aware Classification," Phys. Med. Biol., vol. 66, no. 6, p. 065031, 2021.
K. Wang and Y. Shi, "Discussion on the Syndrome Differentiation Treatment of Pneumonia and Cough in Children," MEDS Chinese Med., vol. 5, no. 8, pp. 121–126, 2023.
T. M. File Jr and J. A. Ramirez, "Community-Acquired Pneumonia," N. Engl. J. Med., vol. 389, no. 7, pp. 632–641, 2023.
X. Yu et al., "Nomogram Using CT Radiomics Features for Differentiation of Pneumonia-Type Invasive Mucinous Adenocarcinoma and Pneumonia: Multicenter Development and External Validation Study," Am. J. Roentgenol., vol. 220, no. 2, pp. 224–234, 2023.
M. K. Alshammari et al., "Prevalence and Etiology of Community- and Hospital-Acquired Pneumonia in Saudi Arabia and Their Antimicrobial Susceptibility Patterns: A Systematic Review," Medicina, vol. 59, no. 4, p. 760, 2023.
M. Suliman, F. Malik, M. Q. Khan, I. Ullah, and A. U. Rub, "Integrating Data Augmentation with AdaBoost for Effective COVID-19 Pneumonia Classification," J. Comput. Biomed. Inform., vol. 7, no. 1, pp. 590–605, 2024.
H.-R. Ma et al., "Lung Ultrasound to Diagnose Infectious Pneumonia of Newborns: A Prospective Multicenter Study," Pediatr. Pulmonol., vol. 58, no. 1, pp. 122–129, 2023.
A. Kakde, N. Arora, D. Sharma, and S. C. Sharma, "Multispectral Classification and Recognition of Breast Cancer and Pneumonia," Pol. J. Med. Phys. Eng., vol. 26, no. 1, pp. 1–9, 2020.
K.-F. Chen et al., "Diagnostic Accuracy of Clinical Signs and Symptoms of COVID-19: A Systematic Review and Meta-Analysis to Investigate the Different Estimates in a Different Stage of the Pandemic Outbreak," J. Glob. Health, vol. 13, 2023.
M. E. H. Chowdhury et al., "Can AI Help in Screening Viral and COVID-19 Pneumonia?" IEEE Access, vol. 8, pp. 132665–132676, 2020.
N. E. M. Khalifa, F. Smarandache, and M. Loey, "COVID-19 Chest X-Ray Images Diagnosis: A Neutrosophic and Deep Transfer Learning Approach," 2020.
X. Xu et al., "A Deep Learning System to Screen Novel Coronavirus Disease 2019 Pneumonia," Eng., vol. 6, no. 10, pp. 1122–1129, 2020.
C. G. Moorthy and G. U. Sankar, "Analysis on Electromagnetic Waves of CT Scanners and MRI Scanners for Applications," World Sci. News, vol. 188, pp. 1–14, 2024.
A. Hussain, A. Khan, and H. Yar, "Efficient Deep Learning Approach for Classification of Pneumonia Using Resources Constraint Devices in Healthcare," in Proc. 5th Int. Conf. Next Gener. Comput., Bidholi Via-Prem Nagar, India, 2019, pp. 20–21.
R. M. Pereira et al., "COVID-19 Identification in Chest X-Ray Images on Flat and Hierarchical Classification Scenarios," Comput. Methods Programs Biomed., vol. 194, p. 105532, 2020.
M. Loey, F. Smarandache, and N. E. M. Khalifa, "Within the lack of COVID-19 benchmark dataset: A novel GAN with deep transfer learning for coronavirus detection in chest X-ray images," Symmetry, vol. 12, no. 4, pp. 1–19, 2020.
J. D. Janizek, G. Erion, A. J. DeGrave, and S.-I. Lee, "An adversarial approach for the robust classification of pneumonia from chest radiographs," in Proc. ACM Conf. Health, Inference, Learn., 2020, pp. 69–79.
T. Rahman, M. E. H. Chowdhury, A. Khandakar, K. R. Islam, K. F. Islam, Z. B. Mahbub, M. A. Kadir, and S. Kashem, "Transfer learning with deep convolutional neural network (CNN) for pneumonia detection using chest X-ray," Appl. Sci., vol. 10, no. 9, p. 3233, 2020.
M. Zhou et al., "Deep learning for differentiating novel coronavirus pneumonia and influenza pneumonia," Ann. Transl. Med., vol. 9, no. 2, 2021.
A. Rehman, S. Naz, A. Khan, A. Zaib, and I. Razzak, "Improving coronavirus (COVID-19) diagnosis using deep transfer learning," medRxiv, 2020. [Online]. Available: https://doi.org/10.1101/2020.04.10.035717
Z. Neili, M. Fezari, and A. Redjati, "ELM and K-nn machine learning in classification of breath sounds signals," Int. J. Electr. Comput. Eng., vol. 10, no. 4, pp. 3528–3536, 2020.
V. Chouhan et al., "A novel transfer learning-based approach for pneumonia detection in chest X-ray images," Appl. Sci., vol. 10, no. 2, p. 559, 2020.
D. Varshni, K. Thakral, L. Agarwal, R. Nijhawan, and A. Mittal, "Pneumonia detection using CNN-based feature extraction," in 2019 IEEE Int. Conf. Electr., Comput., Commun. Technol. (ICECCT), 2019, pp. 1–7.
P. Zhao, J.-W. Shan, Y.-J. Zhang, and Z.-H. Zhou, "Exploratory machine learning with unknown unknowns," Artif. Intell., vol. 327, p. 104059, 2024.
K. El Asnaoui, Y. Chawki, and A. Idri, "Automated methods for detection and classification of pneumonia based on X-ray images using deep learning," in Artif. Intell. Blockchain Future Cybersecurity Appl., Springer, 2021, pp. 257–284.
N. E. M. Khalifa, M. H. N. Taha, A. E. Hassanien, and S. Elghamrawy, "Detection of coronavirus (COVID-19) associated pneumonia based on generative adversarial networks and a fine-tuned deep transfer learning model using chest X-ray dataset," in Int. Conf. Adv. Intell. Syst. Informatics, Springer, 2022, pp. 234–247.
L. O. Hall, R. Paul, D. B. Goldgof, and G. M. Goldgof, "Finding COVID-19 from chest X-rays using deep learning on a small dataset," arXiv Preprint arXiv:2004.02060, 2020.
M. Ilyas, H. Rehman, and A. Naït-Ali, "Detection of COVID-19 from chest X-ray images using artificial intelligence: An early review," arXiv Preprint arXiv:2004.05436, 2020.
E. D. Evans et al., "Predicting human health from biofluid-based metabolomics using machine learning," Sci. Rep., vol. 10, no. 1, p. 17635, 2020.
R. Nagamounika, C. N. S. V. Sri, A. Harshitha, K. L. Tejaswi, and P. R. S. M. Lakshmi, "Prediction of pneumonia disease by using deep convolutional neural networks," J. Eng. Sci. Criterion, vol. 17, pp. 18, 2020.
M. Liebenlito, M. Y. Irene, and A. Hamid, "Classification of Tuberculosis and Pneumonia in Human Lungs Based on Chest X-ray Images Using Convolutional Neural Network," InPrime: Indonesian Journal of Pure and Applied Mathematics, vol. 2, no. 1, pp. 24–32, 2020.
S. Yao, Y. Chen, X. Tian, R. Jiang, and S. Ma, "An Improved Algorithm for Detecting Pneumonia Based on YOLOv3," Applied Sciences, vol. 10, no. 5, p. 1818, 2020.
N. Ansari, A. Faizabadi, S. Motakabber, and M. Ibrahimy, "Effective Pneumonia Detection Using ResNet-Based Transfer Learning," Test Engineering and Management, vol. 82, pp. 15146–15153, 2020.
G. Verma and S. Prakash, "Pneumonia Classification Using Deep Learning in Healthcare," International Journal of Innovative Technology and Exploring Engineering (IJITEE), vol. 9, no. 4, pp. 1715–1723, 2020.
R. C. Hidayatullah and S. Violina, "Convolutional Neural Network Architecture and Data Augmentation for Pneumonia Classification from Chest X-rays Images," International Journal of Innovative Science and Research Technology, vol. 5, pp. 158–164, 2020.
R. Chadha, "A Novel Approach for Detecting Pneumonia in Machine Learning," International Journal of Trend in Innovative Research (IJTIIR), 2019.
M. Chumbita et al., "Can Artificial Intelligence Improve the Management of Pneumonia?" Journal of Clinical Medicine, vol. 9, no. 1, p. 248, 2020.
L. Ebner et al., "Meta-analysis of the Radiological and Clinical Features of Usual Interstitial Pneumonia (UIP) and Nonspecific Interstitial Pneumonia (NSIP)," PLOS ONE, vol. 15, no. 1, p. e0226084, 2020.
K. M. Shuaib, S. A. Shahid, A. A. Javed, and M. Zaid, "Pneumonia Detection through X-ray Using Deep Learning," IOSR Journal of Computer Engineering (IOSR-JCE), vol. 22, no. 1, pp. 08–11, 2020.
R. Reddy, T. G. S. Teja, T. D. S., and P. V. Babu, "Prediction of Pneumonia Using Deep Learning," International Journal of Advance Research, Ideas and Innovations in Technology, vol. 2020.
L. Li et al., "Artificial Intelligence Distinguishes COVID-19 from Community Acquired Pneumonia on Chest CT," Radiology, 2020.
Y.-H. Liao et al., "Machine Learning Methods Applied to Predict Ventilator-Associated Pneumonia with Pseudomonas Aeruginosa Infection via Sensor Array of Electronic Nose in Intensive Care Unit," Sensors, vol. 19, no. 8, p. 1866, 2019.
W. Setiawan and F. Damayanti, "Layers Modification of Convolutional Neural Network for Pneumonia Detection," Journal of Physics: Conference Series, vol. 1477, no. 5, p. 052055, 2020.
C. Feng et al., "A Novel Triage Tool of Artificial Intelligence-Assisted Diagnosis Aid System for Suspected COVID-19 Pneumonia in Fever Clinics," MedRxiv, 2020.
K. T. Islam, S. N. Wijewickrema, A. Collins, and S. J. O'Leary, "A Deep Transfer Learning Framework for Pneumonia Detection from Chest X-ray Images," in VISIGRAPP (5: VISAPP), 2020, pp. 286–293.
P. S. Mann, S. D. Panchal, S. Singh, G. S. Saggu, and K. Gupta, "A Hybrid Deep Convolutional Neural Network Model for Improved Diagnosis of Pneumonia," Neural Computing and Applications, vol. 36, no. 4, pp. 1791–1804, 2024.
J. Yang et al., "Highly Diverse Sputum Microbiota Correlates with the Disease Severity in Patients with Community-Acquired Pneumonia: A Longitudinal Cohort Study," Respiratory Research, vol. 25, no. 1, p. 223, 2024.
A. Panwar and S. Gupta, "Deep Learning Models for Early Detection of Pneumonia Using Chest X-ray Images," in Proceedings of the International Conference on Paradigms of Communication, Computing and Data Sciences: PCCDS 2021, Springer Singapore, 2022, pp. 701–709.
Y. H. Bhosale and K. S. Patnaik, "PulDi-COVID: Chronic Obstructive Pulmonary (Lung) Diseases with COVID-19 Classification Using Ensemble Deep Convolutional Neural Network from Chest X-ray Images to Minimize Severity and Mortality Rates," Biomedical Signal Processing and Control, vol. 81, p. 104445, 2023.
C. Ieracitano et al., "A Fuzzy-Enhanced Deep Learning Approach for Early Detection of COVID-19 Pneumonia from Portable Chest X-ray Images," Neurocomputing, vol. 481, pp. 202–215, 2022.
H. Bhatt and M. Shah, "A Convolutional Neural Network Ensemble Model for Pneumonia Detection Using Chest X-ray Images," Healthcare Analytics, vol. 3, p. 100176, 2023.
F. Garcea, A. Serra, F. Lamberti, and L. Morra, "Data Augmentation for Medical Imaging: A Systematic Literature Review," Computers in Biology and Medicine, vol. 152, p. 106391, 2023.
A. K. Tyagi, Ed., Automated Secure Computing for Next-Generation Systems, John Wiley & Sons, 2023.
A. Shah et al., "A Comprehensive Study on Skin Cancer Detection Using Artificial Neural Network (ANN) and Convolutional Neural Network (CNN)," Clinical eHealth, 2023.
Downloads
Published
How to Cite
Issue
Section
License
Authors who publish with this journal agree to the following terms:
- Authors retain copyright and grant the journal right of first publication with the work simultaneously licensed under a Creative Commons Attribution License (CC-By) that allows others to share the work with an acknowledgment of the work's authorship and initial publication in this journal.
- Authors are able to enter into separate, additional contractual arrangements for the non-exclusive distribution of the journal's published version of the work (e.g., post it to an institutional repository or publish it in a book), with an acknowledgement of its initial publication in this journal.
- Authors are permitted and encouraged to post their work online (e.g., in institutional repositories or on their website) prior to and during the submission process, as it can lead to productive exchanges, as well as earlier and greater citation of published work (See The Effect of Open Access).
This work is licensed under a Creative Commons Attribution License CC BY
