Machine learning models developed and internally validated for predicting chronicity in pediatric immune thrombocytopenia

Abstract

Background

Primary immune thrombocytopenia (ITP) in children is typically self-limiting; however, 20–30% of patients may experience prolonged thrombocytopenia lasting over a year. The challenge is predicting chronicity to ensure personalized treatment approaches.

Objective

To address this issue, we developed and internally validated four machine learning (ML) models using demographic and immunological characteristics to predict the likelihood of chronicity.

Methods

The present study was conducted at Beijing Children's Hospital from June 2018 to December 2021, aiming to develop predictive models for determining the chronicity of pediatric ITP. Four ML models, based on logistic regression classifier, random forest classifier, eXtreme Gradient Boosting (XGBoost), and support vector machine, were employed. These models utilized a set of 16 variables including 14 immunological and 2 demographic predictors. The performance evaluation criteria included prediction accuracy, precision, recall, F1 score, and area under the ROC curve (AUC).

Results

Data were collected from 662 patients who were randomly assigned to either a training dataset or a testing dataset using a random number generator. Among them, 26.5% had chronic disease. All models performed well with AUC values ranging from 0.81 to 0.84, and XGBoost was selected for its highest AUC score and interpretability in constructing the predictive model. Age, Th17, Th17/Treg, TH1, and DNT were identified as significant predictors by the XGBoost algorithm.

Conclusion

We developed a precise predictive model for chronicity in pediatric ITP using ML during the initial phase. The XGBoost model achieved high predictive accuracy by utilizing individual patient clinical parameters and demonstrated commendable interpretability.