Deep Learning–Based Pneumonia Classification on Chest X-Ray Images

Mustakim Mustakim(1*), Danur Lestari(2), Hartono Hartono(3),

(1) Universiti Sultan Zainal Abidin, Terengganu
(2) Universitas Islam Negeri Sultan Syarif Kasim Riau
(3) Universitas Islam Negeri Sultan Syarif Kasim Riau
(*) Corresponding Author

Abstract


Pneumonia is a lung infection and one of the leading causes of mortality worldwide. Early and accurate diagnosis is essential to reduce death rates, with chest X-ray (CXR) imaging being the most commonly used diagnostic tool. However, CXR-based pneumonia identification remains challenging due to limited image quality and the shortage of experienced radiologists. To address this issue, this study proposes a hybrid deep learning framework that integrates Convolutional Neural Network (CNN), Recurrent Neural Network (RNN), and Generative Adversarial Network (GAN) to enhance the classification of bacterial and viral pneumonia from CXR images. The dataset comprises 7,927 CXR images, including 3,270 normal cases, 3,001 cases of bacterial pneumonia, and 1,656 cases of viral pneumonia. Four CNN architectures, Xception, InceptionV3, ResNet50V2, and DenseNet201, are evaluated using RMSprop and Stochastic Gradient Descent (SGD) optimizers. Model development and training are conducted using the TensorFlow framework. Experimental results demonstrate that ResNet50V2 with the RMSprop optimizer achieves the highest classification accuracy of 0.85, while also yielding the fastest training time of 2,215 seconds. These findings indicate that the proposed approach can support faster and more accurate pneumonia screening, particularly in healthcare facilities with limited diagnostic resources

Keywords


Convolutional Neural Network; Deep Learning; Generative Adversarial Network; Pneumonia; Recurrent Neural Network

References


A. U. Ibrahim, M. Ozsoz, S. Serte, F. Al-Turjman, and P. S. Yakoi, “Pneumonia classification using deep learning from chest X-ray images during COVID-19,” Cognit Comput, pp. 1–13, 2021.

R. Siddiqi and S. Javaid, “Deep Learning for Pneumonia Detection in Chest X-ray Images: A Comprehensive Survey,” Aug. 01, 2024, Multidisciplinary Digital Publishing Institute (MDPI). doi: 10.3390/jimaging10080176.

W. Khan, N. Zaki, and L. Ali, “Intelligent pneumonia identification from chest x-rays: A systematic literature review,” IEEE Access, vol. 9, pp. 51747–51771, 2021.

S. Sharma and K. Guleria, “A Deep Learning based model for the Detection of Pneumonia from Chest X-Ray Images using VGG-16 and Neural Networks,” in Procedia Computer Science, Elsevier B.V., 2022, pp. 357–366. doi: 10.1016/j.procs.2023.01.018.

D. Lestari, “Classification of Bacterial and Viral Pneumonia Using Chest X-Ray Images with Convolutional Neural Network,” in 2024 International Conference on Decision Aid Sciences and Applications (DASA), IEEE, 2024, pp. 1–6.

D. Chittora, B. R. Meena, S. Mittholiya, and K. Sharma, “Updates of COVID-19,” Research on Biomedical Engineering, vol. 37, no. 4, pp. 835–848, 2021.

H. Rossman et al., “Hospital load and increased COVID-19 related mortality in Israel,” Nat Commun, vol. 12, no. 1, p. 1904, 2021.

M. Yaseliani, A. Z. Hamadani, A. I. Maghsoodi, and A. Mosavi, “Pneumonia detection proposing a hybrid deep convolutional neural network based on two parallel visual geometry group architectures and machine learning classifiers,” IEEE access, vol. 10, pp. 62110–62128, 2022.

D. Schaudt et al., “A Critical Assessment of Generative Models for Synthetic Data Augmentation on Limited Pneumonia X-ray Data,” Bioengineering, vol. 10, no. 12, p. 1421, 2023.

X. Wang, Y. Peng, L. Lu, Z. Lu, M. Bagheri, and R. M. Summers, “Chestx-ray8: Hospital-scale chest x-ray database and benchmarks on weakly-supervised classification and localization of common thorax diseases,” in Proceedings of the IEEE conference on computer vision and pattern recognition, 2017, pp. 2097–2106.

M. Mardani et al., “Deep generative adversarial neural networks for compressive sensing MRI,” IEEE Trans Med Imaging, vol. 38, no. 1, pp. 167–179, 2018.

R. Siddiqi, “Efficient pediatric pneumonia diagnosis using depthwise separable convolutions,” SN Comput Sci, vol. 1, no. 6, p. 343, 2020.

E. Çallı, E. Sogancioglu, B. van Ginneken, K. G. van Leeuwen, and K. Murphy, “Deep learning for chest X-ray analysis: A survey,” Med Image Anal, vol. 72, p. 102125, 2021.

F. Al-Turjman, AI-powered IoT for COVID-19. CRC Press, 2020.

X. Zhou, P. Lu, Z. Zheng, D. Tolliver, and A. Keramati, “Accident prediction accuracy assessment for highway-rail grade crossings using random forest algorithm compared with decision tree,” Reliab Eng Syst Saf, vol. 200, p. 106931, 2020.

S. Kumar and S. Prakash, “Pneumonia Identification and Classification Using CNN Model Based on Chest X-Ray Image in Healthcare,” in International Conference on Advanced Computing and Intelligent Engineering, Springer, 2022, pp. 473–486.

D. Hastari, S. Winanda, A. R. Pratama, N. Nurhaliza, and E. S. Ginting, “Application of Convolutional Neural Network ResNet-50 V2 on Image Classification of Rice Plant Disease,” Public Research Journal of Engineering, Data Technology and Computer Science, vol. 1, no. 2, 2024.

R. Yamashita, M. Nishio, R. K. G. Do, and K. Togashi, “Convolutional neural networks: an overview and application in radiology,” Insights Imaging, vol. 9, pp. 611–629, 2018.

S. Motamed, P. Rogalla, and F. Khalvati, “Data augmentation using Generative Adversarial Networks (GANs) for GAN-based detection of Pneumonia and COVID-19 in chest X-ray images,” Inform Med Unlocked, vol. 27, p. 100779, 2021.

Y.-D. Zhang et al., “Image based fruit category classification by 13-layer deep convolutional neural network and data augmentation,” Multimed Tools Appl, vol. 78, pp. 3613–3632, 2019.

R. Hao, K. Namdar, L. Liu, M. A. Haider, and F. Khalvati, “A comprehensive study of data augmentation strategies for prostate cancer detection in diffusion-weighted MRI using convolutional neural networks,” J Digit Imaging, vol. 34, pp. 862–876, 2021.

I. Goodfellow et al., “Generative adversarial nets,” Adv Neural Inf Process Syst, vol. 27, 2014.

A. Makhlouf, M. Maayah, N. Abughanam, and C. Catal, “The use of generative adversarial networks in medical image augmentation,” Neural Comput Appl, vol. 35, no. 34, pp. 24055–24068, 2023.

D. Srivastav, A. Bajpai, and P. Srivastava, “Improved classification for pneumonia detection using transfer learning with GAN based synthetic image augmentation,” in 2021 11th international conference on cloud computing, data science & engineering (confluence), IEEE, 2021, pp. 433–437.

J. Zhang, Y. Zeng, and B. Starly, “Recurrent neural networks with long term temporal dependencies in machine tool wear diagnosis and prognosis,” SN Appl Sci, vol. 3, no. 4, p. 442, 2021.

I. Kanjanasurat, K. Tenghongsakul, B. Purahong, and A. Lasakul, “CNN–RNN network integration for the diagnosis of COVID-19 using chest X-ray and CT images,” Sensors, vol. 23, no. 3, p. 1356, 2023.

A. Makandar and M. N. Jadhav, “Disease Recognition in Medical Images Using CNN-LSTM-GRU Ensemble, a Hybrid Deep Learning,” in 2023 7th International Conference on Computation System and Information Technology for Sustainable Solutions (CSITSS), IEEE, 2023, pp. 1–9.

A. A. Wafa, R. M. Essa, A. A. Abohany, and H. E. Abdelkader, “Integrating deep learning for accurate gastrointestinal cancer classification: a comprehensive analysis of MSI and MSS patterns using histopathology data,” Neural Comput Appl, pp. 1–33, 2024.

S. Lafraxo, M. El Ansari, and L. Koutti, “A new hybrid approach for pneumonia detection using chest X-rays based on ACNN-LSTM and attention mechanism,” Multimed Tools Appl, pp. 1–23, 2024.

R. Karla and R. Yalavarthi, “A Hybrid RNN-based Deep Learning Model for Lung Cancer and COPD Detection,” Engineering, Technology and Applied Science Research, vol. 14, no. 5, pp. 16847–16853, Oct. 2024, doi: 10.48084/etasr.8181.

M. Kavitha, K. Vidhya, S. Sreeja, G. Roopashri, and P. Muhil, “Automated lung disease detection, classification and prediction using RNN framework,” in E3S Web of Conferences, EDP Sciences, Feb. 2024. doi: 10.1051/e3sconf/202449103015.

Y. Lu et al., “A hybrid CNN-RNN approach for survival analysis in a Lung Cancer Screening study,” Heliyon, vol. 9, no. 8, Aug. 2023, doi: 10.1016/j.heliyon.2023.e18695.

M. M. Islam, M. Z. Islam, A. Asraf, M. S. Al-Rakhami, W. Ding, and A. H. Sodhro, “Diagnosis of COVID-19 from X-rays using combined CNN-RNN architecture with transfer learning,” BenchCouncil Transactions on Benchmarks, Standards and Evaluations, vol. 2, no. 4, Oct. 2022, doi: 10.1016/j.tbench.2023.100088.

A. Mahadar, P. Mangukiya, and T. Baraskar, “Comparison and Evaluation of CNN Architectures for Classification of Covid-19 and Pneumonia,” in 2021 Sixth International Conference on Image Information Processing (ICIIP), IEEE, 2021, pp. 110–115.

B. Nugroho and A. Yuniarti, “Performance of root-mean-square propagation and adaptive gradient optimization algorithms on covid-19 pneumonia classification,” in 2022 IEEE 8th Information Technology International Seminar (ITIS), IEEE, 2022, pp. 333–338.

D. Kalra, V. Khare, and R. Kumar, “Class-Specific GANs to Improve Synthesis of Bacterial and Viral Pneumonia Chest X-Ray Images,” Journal Européen des Systèmes Automatisés, vol. 57, no. 6, p. 1817, 2024.

M. Rahimzadeh and A. 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,” Inform Med Unlocked, vol. 19, Jan. 2020, doi: 10.1016/j.imu.2020.100360.

Susanti, Mustakim, R. Novita, and I. Permana, “Application of Residual Network Architecture on Covid-19 Chest x-ray Classification,” in 2022 International Symposium on Information Technology and Digital Innovation (ISITDI), 2022, pp. 121–125. doi: 10.1109/ISITDI55734.2022.9944525.

S. T. Rizaldi, Mustakim, I. Permana, and M. Afdal, “Image Enhancement on Deep Learning Algorithm for COVID-19 Lung X-Ray Classification,” in 2022 International Symposium on Information Technology and Digital Innovation (ISITDI), 2022, pp. 11–15. doi: 10.1109/ISITDI55734.2022.9944501.

M. M. Hasan, M. M. J. Kabir, M. R. Haque, and M. Ahmed, “A combined approach using image processing and deep learning to detect pneumonia from chest X-ray image,” in 2019 3rd international conference on electrical, Computer & Telecommunication Engineering (ICECTE), IEEE, 2019, pp. 89–92.

T. Rahman et al., “Transfer learning with deep convolutional neural network (CNN) for pneumonia detection using chest X-ray,” Applied Sciences, vol. 10, no. 9, p. 3233, 2020.

U. Sait et al., “Curated dataset for covid-19 posterior-anterior chest radiography images (x-rays),” Mendeley Data, vol. 1, no. J, 2020.

K. He, X. Zhang, S. Ren, and J. Sun, “Identity mappings in deep residual networks,” in Computer Vision–ECCV 2016: 14th European Conference, Amsterdam, The Netherlands, October 11–14, 2016, Proceedings, Part IV 14, Springer, 2016, pp. 630–645.

P. D. Rinanda, D. N. Aini, T. A. Pertiwi, S. Suryani, and A. J. Prakash, “Implementation of Convolutional Neural Network (CNN) for Image Classification of Leaf Disease In Mango Plants Using Deep Learning Approach,” Public Research Journal of Engineering, Data Technology and Computer Science, vol. 1, no. 2, pp. 56–61, Feb. 2024, doi: 10.57152/predatecs.v1i2.872.

F. Zheng et al., “Accurately Discriminating COVID-19 from Viral and Bacterial Pneumonia According to CT Images Via Deep Learning,” Interdiscip Sci, vol. 13, no. 2, pp. 273–285, Jun. 2021, doi: 10.1007/s12539-021-00420-z.

G. Huang, Z. Liu, L. Van Der Maaten, and K. Q. Weinberger, “Densely connected convolutional networks,” in Proceedings of the IEEE conference on computer vision and pattern recognition, 2017, pp. 4700–4708.

S. Vashisht, S. Lamba, B. Sharma, and A. Sharma, “Pneumonia classification using CNN-GAN,” in 2023 International Conference on Sustainable Computing and Data Communication Systems (ICSCDS), IEEE, 2023, pp. 456–461.

D. Srivastav, A. Bajpai, and P. Srivastava, “Improved classification for pneumonia detection using transfer learning with gan based synthetic image augmentation,” in 2021 11th International Conference on Cloud Computing, Data Science & Engineering (Confluence), IEEE, 2021, pp. 433–437.

F. Chollet, “Xception: Deep learning with depthwise separable convolutions,” in Proceedings of the IEEE conference on computer vision and pattern recognition, 2017, pp. 1251–1258.

N. S. Shadin, S. Sanjana, and N. J. Lisa, “COVID-19 diagnosis from chest X-ray images using convolutional neural network (CNN) and InceptionV3,” in 2021 International conference on information technology (ICIT), IEEE, 2021, pp. 799–804.

N. Tajbakhsh et al., “Convolutional Neural Networks for Medical Image Analysis: Full Training or Fine Tuning?,” IEEE Trans Med Imaging, vol. 35, no. 5, pp. 1299–1312, May 2016, doi: 10.1109/TMI.2016.2535302.

M. A. Islam, S. I. Rashid, N. U. I. Hossain, R. Fleming, and A. Sokolov, “An integrated convolutional neural network and sorting algorithm for image classification for efficient flood disaster management,” Decision Analytics Journal, vol. 7, p. 100225, 2023.

D. Irfan, R. Rosnelly, M. Wahyuni, J. T. Samudra, and A. Rangga, “Perbandingan optimasi sgd, adadelta, dan adam dalam klasifikasi hydrangea menggunakan cnn,” Journal of Science and Social Research, vol. 5, no. 2, pp. 244–253, 2022.

K. He, G. Gkioxari, P. Dollár, and R. Girshick, “Mask r-cnn,” in Proceedings of the IEEE international conference on computer vision, 2017, pp. 2961–2969.

S. Singh and B. K. Tripathi, “Pneumonia classification using quaternion deep learning,” Multimed Tools Appl, vol. 81, no. 2, pp. 1743–1764, Jan. 2022, doi: 10.1007/s11042-021-11409-7.




DOI: http://dx.doi.org/10.61944/bids.v4i2.130

Refbacks

  • There are currently no refbacks.


Copyright (c) 2025 Mustakim Mustakim, Danur Lestari, Hartono Hartono

Creative Commons License
This work is licensed under a Creative Commons Attribution 4.0 International License.

Bulletin of Informatics and Data Science
Asosiasi Peneliti Data Science Indonesia
Email: pdsi.bids@gmail.com
This work is licensed under a Creative Commons Attribution 4.0 International License.