Stacked Generalization-Based Deep Learning Approach for Pneumonia Detection

Year : 2025 | Volume : 12 | Issue : 03 | Page : 20 31
    By

    Jyoti Dabass,

  • Bhupender Singh Dabass,

  1. Postdoctoral Research Associate, Department of Electronics & Electrical Communication, Yardi School of Artificial Intelligence, Indian Institute of Technology Delhi, Hauz Khas, New Delhi, India
  2. Student, Department of Law, Institute of Law and Research, Faridabad, Haryana, India

Abstract

The proposed work focuses on a stacked generalization-based approach for diagnosing pneumonia from chest X-ray images. It utilizes regularization, early stopping, and data augmentation to deal with overfitting. It uses safe level SMOTE to deal with class imbalance and attention-based feature fusion to adaptively weigh features based on their importance. It uses two publicly available datasets (RSNA and Kermany) with ground truth provided by expert radiologists. The proposed work used ChexNet, SqueezeNet, and EfficientNet-B0, as ChexNet has already been fine-tuned on chest X-ray images, SqueezeNet uses its parameters efficiently, and EfficientNet-B0 balances accuracy and efficiency with limited parameters. We experimented with different regularization methods and optimizers to identify the most effective combinations of hyperparameters for our model training. Ensembling these basic classifiers adds diversity and makes it more generalizable for adaptation to unseen data with limited training parameters. Experimental results show that the proposed work is better compared to other state-of-the-art techniques, providing 99.4% accuracy on the Kermany dataset and 98.5% accuracy on the RSNA dataset. It shall be useful to the radiologists in improving the efficiency of pneumonia detection and thus will be beneficial in saving lives.

Keywords: Pneumonia detection, stacked generalization-based approach, chest X-ray images, RSNA and Kermany, SMOTE

[This article belongs to Journal of Image Processing & Pattern Recognition Progress ]

How to cite this article:
Jyoti Dabass, Bhupender Singh Dabass. Stacked Generalization-Based Deep Learning Approach for Pneumonia Detection. Journal of Image Processing & Pattern Recognition Progress. 2025; 12(03):20-31.
How to cite this URL:
Jyoti Dabass, Bhupender Singh Dabass. Stacked Generalization-Based Deep Learning Approach for Pneumonia Detection. Journal of Image Processing & Pattern Recognition Progress. 2025; 12(03):20-31. Available from: https://journals.stmjournals.com/joipprp/article=2025/view=227995


References

  1. Rehman MU, Shafique A, Khan KH, Khalid S, Alotaibi AA, Althobaiti T, Ramzan N, Ahmad J, Shah SA, Abbasi QH. Novel privacy preserving non-invasive sensing-based diagnoses of pneumonia disease leveraging deep network model. Sensors. 2022 Jan 8;22(2):461.
  2. Kareem A, Liu H, Sant P. Review on pneumonia image detection: A machine learning approach. Hum-Centric Intell Syst. 2022; 2(1): 31–43.
  3. Harsono IW, Liawatimena S, Cenggoro TW. Lung nodule detection and classification from Thorax CT-scan using RetinaNet with transfer learning. J King Saud Univ-Comput Inf Sci. 2022; 34(3): 567–577.
  4. Aydogdu M, Ozyilmaz E, Aksoy H, Gursel G, Ekim N. Mortality prediction in community-acquired pneumonia requiring mechanical ventilation; values of pneumonia and intensive care unit severity scores. Tuberk Toraks. 2010 Jan 1; 58(1): 25–34.
  5. QH. Novel privacy-preserving non-invasive sensing-based diagnoses of pneumonia disease leveraging a deep network model. Sensors. 2022; 22(2): 461.
  6. Colin J, Surantha N. Interpretable Deep Learning for Pneumonia Detection Using Chest X-Ray Images. Information. 2025; 16(1): 53.
  7. Asnake NW, Salau AO, Ayalew AM. X-ray image-based pneumonia detection and classification using deep learning. Multimed Tools Appl. 2024 Jun; 83(21): 60789–807.
  8. Das AK, Ghosh S, Thunder S, Dutta R, Agarwal S, Chakrabarti A. Automatic COVID-19 detection from X-ray images using ensemble learning with a convolutional neural network. Pattern Anal Appl. 2021; 24: 1111–1124.
  9. Monshi MMA, Poon J, Chung V, Monshi FM. CovidXrayNet: Optimizing data augmentation and CNN hyperparameters for improved COVID-19 detection from CXR. Comput Biol Med. 2021; 133: 104375.
  10. Usman C, Rehman SU, Ali A, Khan AM, Ahmad B. Pneumonia Disease Detection Using Chest X-Rays and Machine Learning. Algorithms. 2025; 18(2): 82.
  11. Aljawarneh SA, Al-Quraan R. Pneumonia detection using enhanced convolutional neural network model on chest x-ray images. Big Data. 2025 Feb 1; 13(1): 16–29.
  12. Kaya M. Feature fusion-based ensemble CNN learning optimization for automated detection of pediatric pneumonia. Biomed Signal Process Control. 2024; 87: 105472.
  13. Farhan AM, Yang S, Al-Malahi AQ, Al-antari MA. MCLSG: Multi-modal classification of lung disease and severity grading framework using consolidated feature engineering mechanisms. Biomed Signal Process Control. 2023; 85: 104916.
  14. Kundu R, Das R, Geem ZW, Han GT, Sarkar R. Pneumonia detection in chest X-ray images using an ensemble of deep learning models. PloS one. 2021; 16(9): e0256630.
  15. Bhatt H, Shah M. A Convolutional Neural Network ensemble model for Pneumonia Detection using chest X-ray images. Healthcare Analytics. 2023 Nov 1; 3: 100176.
  16. Liang G, Zheng L. A transfer learning method with the deep residual network for pediatric pneumonia diagnosis. Comput Methods Programs Biomed. 2020; 187: 104964.
  17. Soares LP, Soares CP. Automatic detection of COVID-19 cases on X-ray images using convolutional neural networks. arXiv preprint arXiv:2007.05494. 2020.
  18. Zhang D, Ren F, Li Y, Na L, Ma Y. Pneumonia detection from chest X-ray images based on a convolutional neural network. Electronics. 2021; 10(13): 1512.
  19. Dey N, Zhang YD, Rajinikanth V, Pugalenthi R, Raja NSM. Customized VGG19 architecture for pneumonia detection in chest X-rays. Pattern Recognit Lett. 2021; 143: 67–74.
  20. Sharma S, Guleria K. A deep learning-based model for the detection of pneumonia from chest X-ray images using VGG-16 and neural networks. Procedia Comput Sci. 2023; 218: 357–366.
  21. An Q, Chen W, Shao W. A deep convolutional neural network for pneumonia detection in X-ray images with attention ensemble. Diagnostics. 2024 Feb 11; 14(4): 390.
  22. Nettur SB, Karpurapu S, Nettur U, Gajja LS, Myneni S, Dusi A, Posham L. Lightweight Weighted Average Ensemble Model for Pneumonia Detection in Chest X-Ray Images. arXiv preprint arXiv:2501.16249. 2025.
  23. Hemdan EED, Shouman MA, Karar ME. Covidx-net: A framework of deep learning classifiers to diagnose COVID- 19 in X-ray images. arXiv preprint arXiv:2003.11055. 2020.
  24. Rana N, Marwaha H. Pneumonia detection from X-ray images using federated learning–An unsupervised learning approach. Meas: Sens. 2025; 37: 101410.
  25. Singh S, Kumar M, Kumar A, Verma BK, Abhishek K, Selvarajan S. Efficient pneumonia detection using Vision Transformers on chest X-rays. Sci Rep. 2024; 14(1): 2487.
  26. Xu J, Wang Y. FMT: A Multimodal Pneumonia Detection Model Based on Stacking MOE Framework. arXiv preprint arXiv:2503.05626. 2025.
  27. Mustapha B, Zhou Y, Shan C, Xiao Z. Enhanced Pneumonia Detection in Chest X-Rays Using Hybrid Convolutional and Vision Transformer Networks. Curr Med Imaging. 2025; 21: e15734056326685.
  28. Shih G, Wu CC, Halabi SS, Kohli MD, Prevedello LM, Cook TS, Sharma A, Amorosa JK, Arteaga V, Galperin-Aizenberg M, Gill RR. Augmenting the National Institutes of Health chest radiograph dataset with expert annotations of possible pneumonia. Radiol: Artif Intell. 2019; 1(1): e180041.
  29. Stein A, Wu C, Carr C, Shih G, Dulkowski J, Kalpathy-Cramer J. RSNA pneumonia detection challenge. Mountain View: Kaggle. 2018.
  30. Kermany D. Labeled optical coherence tomography (OCT) and chest x-ray images for classification. Mendeley data. 2018.
  31. Gosain A, Sardana S. Handling class imbalance problem using oversampling techniques: A review. In: 2017 International Conference on Advances in Computing, Communications and Informatics (ICACCI). 2017; 79–85.
  32. Rajpurkar P, Irvin J, Zhu K, Yang B, Mehta H, Duan T, Ding D, Bagul A, Langlotz C, Shpanskaya K, Lungren MP. Chexnet: Radiologist-level pneumonia detection on chest x-rays with deep learning. arXiv preprint arXiv:1711.05225. 2017.
  33. Bianco S, Cadene R, Celona L, Napoletano P. Benchmark Analysis of Representative Deep Neural Network Architectures. IEEE Access. 2018; 6: 64270–64277.
  34. Tan M, Le Q. EfficientNet: Rethinking model scaling for convolutional neural networks. In International Conference on machine learning; PMLR. 2019 May; 6105–6114.
  35. Polikar R. Ensemble-based systems in decision making. IEEE Circuits Syst Mag. 2006; 6(3): 21–45.
  36. Weerts HJ, Mueller AC, Vanschoren J. Importance of tuning hyperparameters of machine learning algorithms. arXiv preprint arXiv:2007.07588. 2020.
  37. Shati A, Hassan GM, Datta A. A comprehensive fusion model for improved pneumonia prediction based on KNN- wavelet-GLCM and a residual network. Intell Syst Appl. 2025; 26: 200492.
  38. Chiwariro R, Wosowei JB. Comparative analysis of deep learning convolutional neural networks based on transfer learning for pneumonia detection. Int J Res Appl Sci Eng Technol. 2023; 11(1): 1161–1170.

Regular Issue Subscription Review Article
Volume 12
Issue 03
Received 25/06/2025
Accepted 16/07/2025
Published 19/09/2025
Publication Time 86 Days


Login


My IP

PlumX Metrics