Capsule Neural Networks with Bayesian Optimization for Pediatric Pneumonia Detection from Chest X-Ray Images

Area of Science:

• Artificial Intelligence
• Medical Imaging
• Pediatric Pneumonia Diagnosis

Background:

• Pneumonia in children is a significant health threat requiring prompt detection.
• Artificial intelligence (AI) offers potential solutions for improving diagnostic accuracy.
• Early identification of pediatric pneumonia is crucial for effective treatment and improved outcomes.

Purpose of the Study:

• To develop and evaluate an AI model for predicting childhood pneumonia using chest X-ray images.
• To leverage capsule networks and Bayesian optimization for enhanced diagnostic performance.
• To assess the clinical relevance and explainability of the AI model's predictions.

Main Methods:

• Chest X-ray images were preprocessed and divided into training, validation, and testing sets.
• A capsule neural network model was designed and optimized using Bayesian optimization.
• Explainability analysis was performed using Grad-CAM to interpret model predictions.

Main Results:

• The AI model achieved high performance metrics, including 95.1% accuracy, 98.9% sensitivity, and 85.4% specificity.
• The model demonstrated strong predictive values with a positive predictive value (PPV) of 94.8% and negative predictive value (NPV) of 96.2%.
• Grad-CAM analysis confirmed that the model's predictions were based on clinically relevant pulmonary regions.

Conclusions:

• The proposed AI model exhibits high accuracy and reliability for pediatric pneumonia detection from chest X-rays.
• The model shows significant potential for integration into clinical practice to aid in early diagnosis.
• The methodology can be extended to other medical image analysis tasks, highlighting its versatility.