Stacked ensemble learning based on deep convolutional neural networks for pediatric pneumonia diagnosis using chest

J Arun Prakash1, Vinayakumar Ravi2, V Sowmya1

  • 1Center for Computational Engineering and Networking (CEN), Amrita School of Engineering, Amrita Vishwa Vidyapeetham, Coimbatore, India.

Insights

A new deep learning model accurately detects pediatric pneumonia from chest X-rays, improving diagnosis speed and cost-efficiency. This AI tool aids physicians in identifying pneumonia in children, potentially saving lives.

Area of Science:

  • Medical Imaging
  • Artificial Intelligence
  • Pediatrics

Background:

  • Pneumonia is a leading cause of child mortality, particularly in developing nations.
  • Current diagnostic methods like chest X-rays are time-consuming and challenging in pediatric cases due to low radiation levels.
  • There is a critical need for rapid, accurate, and cost-effective diagnostic tools for pediatric pneumonia.

Purpose of the Study:

  • To develop and evaluate a novel computer-aided detection (CAD) model for pediatric pneumonia classification.
  • To leverage deep learning and ensemble methods for enhanced diagnostic accuracy and efficiency.
  • To provide a reliable tool for assisting radiologists and physicians in diagnosing pediatric pneumonia.

Main Methods:

  • A stacked ensemble learning approach was employed, utilizing deep learning features extracted from a fine-tuned Xception model.
  • Kernel Principal Component Analysis (KPCA) was used for dimensionality reduction of the extracted features.
  • A two-stage stacking classifier, including Random Forest, KNN, Logistic Regression, XGBoost, SVC, Nu-SVC, and MLP, was trained and validated using Stratified K-fold cross-validation.

Main Results:

  • The proposed model achieved high performance metrics on a public pediatric pneumonia dataset.
  • Key performance indicators included 98.3% accuracy, 99.29% precision, 98.36% recall, 98.83% F1-score, and 98.24% AUC score.
  • The results demonstrate the model's potential for reliable real-time deployment.

Conclusions:

  • The developed stacked ensemble deep learning model offers a highly accurate and efficient solution for pediatric pneumonia detection.
  • This AI-powered tool can significantly assist healthcare professionals in diagnosing pneumonia, potentially reducing mortality rates.
  • The model's performance suggests its suitability for integration into clinical workflows for improved pediatric care.