Use of CNN with Transfer Learning to Improve the Accuracy of Pneumonia Diagnosis

Use of CNN with Transfer Learning to Improve the Accuracy of Pneumonia Diagnosis
  IJETT-book-cover           
  
© 2025 by IJETT Journal
Volume-73 Issue-6
Year of Publication : 2025
Author : Orlando Iparraguirre-Villanueva, Wilmer Robles-Espiritu, María Suxe-Ramírez, Rosalynn Ornella Flores-Castañeda
DOI : 10.14445/22315381/IJETT-V73I6P131

How to Cite?
Orlando Iparraguirre-Villanueva, Wilmer Robles-Espiritu, María Suxe-Ramírez, Rosalynn Ornella Flores-Castañeda, "Use of CNN with Transfer Learning to Improve the Accuracy of Pneumonia Diagnosis," International Journal of Engineering Trends and Technology, vol. 73, no. 6, pp.382-395, 2025. Crossref, https://doi.org/10.14445/22315381/IJETT-V73I6P131

Abstract
Pneumonia is a respiratory disease affecting both lungs, causing symptoms such as cough with phlegm or pus, fever, chills and shortness of breath. Different microorganisms, such as bacteria, viruses and fungi, can cause pneumonia. Today, combating pneumonia is challenging for physicians, especially in vulnerable communities and during cold or sudden weather changes. In this work, we explored how Artificial Intelligence can contribute to improving the diagnosis of pneumonia. For this purpose, two Convolutional Neural Network (CNN) models, VGG16 and ResNet50-v2, were evaluated using the transfer learning technique to identify between healthy lungs and those affected by the disease. We worked with a dataset extracted from the Kaggle platform, which included 5216 images for training, 624 for testing and 16 for validation. The results showed that the ResNet50-v2 model obtained better results, reaching an accuracy rate of 90.87% in the test set, standing out for its ability to identify pneumonia cases. These results reinforce the potential of artificial intelligence as a diagnostic support tool. Finally, it can be stated that this work represents a significant advance in using neural networks in medicine. Integrating this technology into public health may not only improve the identification of pneumonia but may also contribute to the development of new efficient healthcare systems.

Keywords
Deep Learning, Pneumonia, CNN, Transfer Learning, Diagnosis.

References
[1] A. Pranaya et al., “Pneumonia Detection Using Deep Learning,” E3S Web of Conferences, vol. 391, 2023.
[CrossRef] [Google Scholar] [Publisher Link]
[2] M. Vinodhini et al., “Detection of Post COVID-Pneumonia Using Histogram Equalization, CLAHE Deep Learning Techniques,” Inteligencia Artificial, vol. 26, no. 72, pp. 137-145, 2023.
[CrossRef] [Google Scholar] [Publisher Link]
[3] Harsh Bhatt, and Manan Shah, “A Convolutional Neural Network Ensemble Model for Pneumonia Detection Using Chest X-Ray Images,” Healthcare Analytics, vol. 3, 2023.
[CrossRef] [Google Scholar] [Publisher Link]
[4] Amer Kareem, Haiming Liu, and Vladan Velisavljevic, “A Federated Learning Framework for Pneumonia Image Detection Using Distributed Data,” Healthcare Analytics, vol. 4, 2023.
[CrossRef] [Google Scholar] [Publisher Link]
[5] Shengchao Chen et al., “Interpretable CNN-Multilevel Attention Transformer for Rapid Recognition of Pneumonia from Chest X-Ray Images,” IEEE Journal of Biomedical and Health Informatics, vol. 28, no. 2, pp. 753-764, 2024.
[CrossRef] [Google Scholar] [Publisher Link]
[6] Niranjan C. Kundur et al., “Pneumonia Detection in Chest X-Rays Using Transfer Learning and TPUs,” Engineering, Technology and Applied Science Research, vol. 13, no. 5, pp. 11878-11883, 2023.
[CrossRef] [Google Scholar] [Publisher Link]
[7] Abdullah Irfan et al., “Classifying Pneumonia among Chest X-Rays Using Transfer Learning,” 2020 42nd Annual International Conference of the IEEE Engineering in Medicine & Biology Society (EMBC), Montreal, QC, Canada, pp. 2186-2189, 2020.
[CrossRef] [Google Scholar] [Publisher Link]
[8] Zarah Campos-Lopez et al., “Identification of Pneumonia with X-ray Images Using Deep Transfer Learning,” XLVI Mexican Conference on Biomedical Engineering, Malaysia, pp. 32-40, 2024.
[CrossRef] [Google Scholar] [Publisher Link]
[9] Rong Yi et al., “Identification and Classification of Pneumonia Disease Using A Deep Learning-Based Intelligent Computational Framework,” Neural Computing and Applications, vol. 35, no. 20, pp. 14473-14486, 2023.
[CrossRef] [Google Scholar] [Publisher Link]
[10] Cosimo Ieracitan 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.
[CrossRef] [Google Scholar] [Publisher Link]
[11] Mouna Afif et al., “Deep Learning-Based Technique for Lesions Segmentation in CT Scan Images for COVID-19 Prediction,” Multimedia Tools and Applications, vol. 82, no. 17, pp. 26885-26899, 2023.
[CrossRef] [Google Scholar] [Publisher Link]
[12] Pranab Sahoo et al., “A Multi-Stage Framework for COVID-19 Detection and Severity Assessment from Chest Radiography Images Using Advanced Fuzzy Ensemble Technique,” Expert Systems with Applications, vol. 238, part B, 2024.
[CrossRef] [Google Scholar] [Publisher Link]
[13] Kamila Kassylkassova et al., “Automated Pneumonia Diagnosis Using a 2D Deep Convolutional Neural Network with Chest X-Ray Images,” International Journal of Advanced Computer Science and Applications, vol. 14, no. 2, pp. 699-708, 2023.
[CrossRef] [Google Scholar] [Publisher Link]
[14] Tianmu Wang et al., “PneuNet: Deep Learning for COVID-19 Pneumonia Diagnosis on Chest X-Ray Image Analysis Using Vision Transformer,” Medical & Biological Engineering & Computing, vol. 61, no. 6, pp. 1395-1408, 2023.
[CrossRef] [Google Scholar] [Publisher Link]
[15] Muhab Hariri, and Ercan Avsar, “COVID-19 and Pneumonia Diagnosis from Chest X-Ray Images Using Convolutional Neural Networks,” Network Modeling Analysis in Health Informatics and Bioinformatics, vol. 12, no. 1, 2023.
[CrossRef] [Google Scholar] [Publisher Link]
[16] Mana Saleh Al Reshan et al., “Detection of Pneumonia from Chest X-Ray Images Utilizing MobileNet Model,” Healthcare, vol. 11, no. 11, pp. 1-18, 2023.
[CrossRef] [Google Scholar] [Publisher Link]
[17] Yeongbong Jin, Woojin Chang, and Bonggyun Ko, “Generating Chest X-Ray Progression of Pneumonia Using Conditional Cycle Generative Adversarial Networks,” IEEE Access, vol. 11, pp. 88152-88160, 2023.
[CrossRef] [Google Scholar] [Publisher Link]
[18] Titipong Kaewlek et al., “Classification of Pneumonia, Tuberculosis, and COVID-19 on Computed Tomography Images using Deep Learning,” Trends in Sciences, vol. 20, no. 11, pp. 1-15, 2023.
[CrossRef] [Google Scholar] [Publisher Link]
[19] Amr E. Eldin Rashed, and Waleed M. Bahgat, “Conditional Cascaded Network (CCN) Approach for Diagnosis of COVID-19 in Chest X-Ray and CT Images Using Transfer Learning,” Biomed Signal Process Control, vol. 87, part B, 2024.
[CrossRef] [Google Scholar] [Publisher Link]
[20] Simeon Yuda Prasetyo et al., “Pneumonia Detection on X-Ray Imaging Using Softmax Output in Multilevel Meta Ensemble Algorithm of Deep Convolutional Neural Network Transfer Learning Models,” International Journal of Advances in Intelligent Informatics, vol. 9, no. 2, pp. 319-330, 2023.
[CrossRef] [Google Scholar] [Publisher Link]
[21] Pascal Riedel et al., “ResNetFed: Federated Deep Learning Architecture for Privacy-Preserving Pneumonia Detection from COVID-19 Chest Radiographs,” Journal of Healthcare Informatics Research, vol. 7, no. 2, pp. 203-224, 2023.
[CrossRef] [Google Scholar] [Publisher Link]
[22] Asifuzzaman Lasker et al., “LWSNe - A Novel Deep-Learning Architecture to Segregate Covid-19 and Pneumonia From X-Ray Imagery,” Multimedia Tools and Applications, vol. 82, no. 14, pp. 21801-21823, 2023.
[CrossRef] [Google Scholar] [Publisher Link]
[23] S.K. Mydhili et al., “CNN Based Deep Learning for Pneumonia Detection,” 2023 12th IEEE International Conference on Advanced Computing (ICoAC), Chennai, India, pp. 1-7, 2023.
[CrossRef] [Google Scholar] [Publisher Link]
[24] Sonal Agrawal, and Pragati Agrawal, “Detection of Pneumonia Cases from X-Ray Chest Images Using Deep Learning Based on Transfer Learning CNN and Hyperparameter Optimization,” 2023 Third International Conference on Secure Cyber Computing and Communication (ICSCCC), Jalandhar, India, pp. 37-42, 2023.
[CrossRef] [Google Scholar] [Publisher Link]
[25] Liu mao, Tan Yumeng, and Chen Lina, “Pneumonia Detection in Chest X-Rays: A Deep Learning Approach Based on Ensemble RetinaNet and Mask R-CNN,” 2020 Eighth International Conference on Advanced Cloud and Big Data (CBD), Taiyuan, China, pp. 213-218, 2020.
[CrossRef] [Google Scholar] [Publisher Link]
[26] Ruek-Kai Sheu et al., “Interpretable Classification of Pneumonia Infection Using Explainable AI (XAI-ICP),” IEEE Access, vol. 11, pp. 28896-28919, 2023.
[CrossRef] [Google Scholar] [Publisher Link]
[27] José Raniery Ferreira et al., “Multi-View Ensemble Convolutional Neural Network to Improve Classification of Pneumonia in Low Contrast Chest X-Ray Images,” 2020 42nd Annual International Conference of the IEEE Engineering in Medicine & Biology Society (EMBC), Montreal, QC, Canada, pp. 1238-1241, 2020.
[CrossRef] [Google Scholar] [Publisher Link]
[28] El-Sayed M. El-Kenawy et al., “Advanced Meta-Heuristics, Convolutional Neural Networks, and Feature Selectors for Efficient COVID-19 X-Ray Chest Image Classification,” IEEE Access, vol. 9, pp. 36019-36037, 2021.
[CrossRef] [Google Scholar] [Publisher Link]
[29] Md Manjurul Ahsan et al., “Detecting SARS-CoV-2 from Chest X-Ray Using Artificial Intelligence,” IEEE Access, vol. 9, pp. 35501-35513, 2021.
[CrossRef] [Google Scholar] [Publisher Link]
[30] A. Madhavi et al., “Automated Diagnosis of Pneumonia Using CNN and Transfer Learning Approaches,” E3S Web of Conferences, vol. 430, 2023.
[CrossRef] [Google Scholar] [Publisher Link]
[31] Ovai Sarkar et al., “Multi-Scale CNN: An Explainable AI-Integrated Unique Deep Learning Framework for Lung-Affected Disease Classification,” Technologies, vol. 11, no. 5, pp. 1-40, 2023.
[CrossRef] [Google Scholar] [Publisher Link]
[32] Dário Passos, and Puneet Mishra, “Deep Tutti Frutti: Exploring CNN Architectures for Dry Matter Prediction in Fruit From Multi-Fruit Near-Infrared Spectra,” Chemometrics and Intelligent Laboratory Systems, vol. 243, 2023.
[CrossRef] [Google Scholar] [Publisher Link]
[33] Iosu Rodriguez-Martinez et al., “A Study on the Suitability of Different Pooling Operators for Convolutional Neural Networks in the Prediction of COVID-19 through Chest X-Ray Image Analysis,” Expert Systems with Applications, vol. 235, 2024.
[CrossRef] [Google Scholar] [Publisher Link]
[34] Sen Zhang et al., “Image Target Recognition Model of Multi- Channel Structure Convolutional Neural Network Training Automatic Encoder,” IEEE Access, vol. 8, pp. 113090-113103, 2020.
[CrossRef] [Google Scholar] [Publisher Link]
[35] Maria Vakalopoulou et al., Deep Learning: Basics and Convolutional Neural Networks (CNNs), Machine Learning for Brain Disorders, Neuromethods, vol. 197, pp. 77-115, 2023.
[CrossRef] [Google Scholar] [Publisher Link]
[36] Zhenjie Zheng, Zhiqian Zhang, and Wei Pan, “Virtual Prototyping- and Transfer Learning-Enabled Module Detection for Modular Integrated Construction,” Automation in Construction, vol. 120, 2020.
[CrossRef] [Google Scholar] [Publisher Link]
[37] Stevo Bozinovski, “Reminder of the First Paper on Transfer Learning in Neural Networks, 1976,” An International Journal of Computing and Informatics, vol. 44, no. 3, pp. 291-302, 2020.
[Google Scholar] [Publisher Link]
[38] Qiaoning Yang et al., “Deep Convolution Neural Network-Based Transfer Learning Method for Civil Infrastructure Crack Detection,” Automation in Construction, vol. 116, 2020.
[CrossRef] [Google Scholar] [Publisher Link]
[39] Ömer Faruk Akmeşe et al., “Prediction of Gender and Age Period from Periorbital Region with VGG16,” Chaos Theory and Applications, vol. 5, no. 2, pp. 105-110, 2023.
[CrossRef] [Google Scholar] [Publisher Link]
[40] Jun Sun et al., “Nondestructive Identification of Soybean Protein in Minced Chicken Meat Based on Hyperspectral Imaging and VGG16-SVM,” Journal of Food Composition and Analysis, vol. 125, 2024.
[CrossRef] [Google Scholar] [Publisher Link]
[41] Mohamad Aqib Haqmi Abas et al., “VGG16 for Plant Image Classification with Transfer Learning and Data Augmentation,” International Journal of Engineering and Technology, vol. 7, no. 4.11, pp. 90-94, 2018.
[CrossRef] [Google Scholar] [Publisher Link]
[42] Hajar Indah Fitriasari, and Mia Rizkinia, “Improvement of Xception-ResNet50V2 Concatenation for COVID-19 Detection on Chest X-Ray Images,” 2021 3rd East Indonesia Conference on Computer and Information Technology (EIConCIT), Surabaya, Indonesia, pp. 343-347, 2021.
[CrossRef] [Google Scholar] [Publisher Link]
[43] Mohammad Rahimzadeh, and Abolfazl Attar, “A Modified Deep Convolutional Neural Network for Detecting COVID-19 and Pneumonia from Chest X-Ray Images Based on the Concatenation of Xception and ResNet50V2,” Informatics in Medicine Unlocked, vol. 19, 2020.
[CrossRef] [Google Scholar] [Publisher Link]
[44] Soumya Ranjan Nayak et al., “An Automated Lightweight Deep Neural Network for Diagnosis of COVID-19 from Chest X-Ray Images,” Arabian Journal for Science and Engineering, vol. 48, no. 8, pp. 11085-11102, 2023.
[CrossRef] [Google Scholar] [Publisher Link]