Interpretable machine learning for allergic rhinitis prediction among preschool children in Urumqi, China

Jinyang Wang1, Ye Yang2, Xueli Gong3

  • 1Department of Clinical Medicine, Xinjiang Medical University, Urumqi, 830017, China.

Scientific Reports
|September 27, 2024
PubMed

Insights

Machine learning models, particularly Random Forest, show superior performance in predicting allergic rhinitis (AR) risk in children aged 2-8 compared to logistic regression. Key predictors include parental allergy history and early life environmental factors.

Area of Science:

  • Pediatric Allergy and Immunology
  • Computational Medicine
  • Data Science in Healthcare

Background:

  • Allergic rhinitis (AR) is a common condition in children, impacting quality of life and potentially leading to other allergic diseases.
  • Accurate prediction of AR risk in early childhood is crucial for timely intervention and management.
  • Traditional risk assessment methods may not fully capture the complex interplay of genetic and environmental factors contributing to AR.

Purpose of the Study:

  • To compare the predictive performance of machine learning (ML) models against logistic regression (LR) for childhood allergic rhinitis (AR).
  • To identify key predictors of AR in children aged 2-8 using advanced feature selection techniques.
  • To develop and validate a robust ML model for early AR risk prediction in pediatric populations.

Main Methods:

  • Analysis of questionnaire data from 7131 children aged 2-8, with random data splitting for training, validation, and testing (100 repetitions).
  • Implementation and comparison of LR, Support Vector Machine (SVM), Random Forest (RF), and Extreme Gradient Boosting Tree (XGBoost) models.
  • Feature engineering using chi-square and Boruta algorithms; model evaluation via Area Under the Receiver Operating Characteristic Curve (AUROC) and SHAP analysis for feature importance.

Main Results:

  • The Random Forest (RF) model demonstrated superior performance (AUROC = 0.747 ± 0.015) compared to Logistic Regression (LR) (AUROC = 0.715 ± 0.023), with statistically significant improvement (p < 0.001).
  • Top predictors for AR included parental history of AR, presence of older siblings, history of food allergy, and father's educational level.
  • Model performance remained stable across gender, birth mode, and age subgroups, indicating robustness.

Conclusions:

  • Machine learning models, especially Random Forest, offer significant advantages over traditional logistic regression for predicting allergic rhinitis risk in young children.
  • Parental allergy history and specific early-life environmental factors are critical determinants of AR development.
  • This study provides a validated ML framework for early AR risk identification in children, supporting personalized pediatric allergy care.

Related Concept Videos