Exploring machine learning algorithms to predict acute respiratory tract infection and identify its determinants

Tirualem Zeleke Yehuala1, Bezawit Melak Fente2, Sisay Maru Wubante1

  • 1Department Health Informatics, Institute of Public Health, College of Medicine and Health Sciences, University of Gondar, Gondar, Ethiopia.

Frontiers in Pediatrics
|December 5, 2024
PubMed

Insights

Predicting acute respiratory infections (ARI) in children under five using machine learning can save lives. Vaccination, breastfeeding, and facility birth are key protective factors against ARI in Sub-Saharan Africa.

Area of Science:

  • Pediatrics
  • Public Health
  • Machine Learning

Background:

  • Acute respiratory infections (ARI) are a leading cause of mortality in children under five globally.
  • Early prediction and identification of ARI determinants are crucial for effective intervention strategies.
  • Machine learning offers advanced tools for analyzing complex health datasets to predict and prevent childhood diseases.

Purpose of the Study:

  • To predict acute respiratory infections (ARI) in children under five using machine learning models.
  • To identify the key determinants and risk factors associated with ARI in Sub-Saharan Africa.
  • To leverage advanced AI for reducing child mortality due to respiratory illnesses.

Main Methods:

  • Utilized Demographic and Health Survey (DHS) data from 36 Sub-Saharan African countries (2005-2022).
  • Employed five machine learning algorithms: Random Forest, Decision Tree, XGBoost, Logistic Regression, and Naive Bayes.
  • Evaluated model performance using accuracy, precision, recall, and AUC curve metrics.

Main Results:

  • Random Forest achieved the highest performance with 96.40% accuracy, 87.9% precision, and 94% ROC.
  • Key protective factors against ARI include breastfeeding, vaccination, media exposure, absence of diarrhea, and facility birth.
  • Naive Bayes showed the lowest performance among the tested models.

Conclusions:

  • Machine learning, particularly Random Forest, demonstrates high predictive power for ARI in children.
  • Vaccination status is a significant factor in preventing ARI, highlighting the importance of immunization programs.
  • Findings support policy development to reduce infant mortality by addressing identified ARI risk factors.
Abstract