Development of an interpretable machine learning model to predict complete remission and first adverse event in

Ningshu Huang1, Weiwei Wang2, Rui Yang2

  • 1Department of Clinical Laboratory, Children's Hospital of Chongqing Medical University, National Clinical Research Center for Children and Adolescents' Health and Diseases, Ministry of Education Key Laboratory of Child Development and Disorders, Chongqing Key Laboratory of Pediatrics Metabolism and Inflammatory Diseases, Laboratory for clinical diagnostic and translational research of CHCQMU, Chongqing, China.

Frontiers in Oncology
|April 27, 2026
PubMed

Insights

Machine learning models accurately predict complete remission and adverse events in pediatric acute myeloid leukemia (pAML) using routine clinical data. This aids in personalized treatment strategies for pAML patients.

Area of Science:

  • Hematology
  • Oncology
  • Machine Learning

Background:

  • Pediatric acute myeloid leukemia (pAML) is a severe hematologic malignancy.
  • Predicting complete remission (CR) and adverse events (AE) is crucial for tailoring pAML treatment.
  • Interpretable machine learning (ML) models using routine clinical features are needed.

Purpose of the Study:

  • Develop and evaluate ML models for predicting CR and AE in pediatric AML patients.
  • Identify key clinical features that predict CR and AE.
  • Assess model interpretability for clinical application.

Main Methods:

  • Utilized data from 206 pediatric AML patients, excluding acute promyelocytic leukemia.
  • Developed seven supervised ML algorithms for CR and AE prediction.
  • Employed feature selection methods (random forest, stepwise regression, JMIM) and interpretability techniques (feature importance, ALE, SHAP).

Main Results:

  • Random forest models achieved high performance for both CR (test AUC 0.79) and AE prediction (test AUC 0.78).
  • Key predictors for CR included age, blasts, platelet count, and genetic markers.
  • Key predictors for AE included WBC, blasts, platelet count, hemoglobin, age, and genetic markers.
  • Higher platelet count predicted increased CR probability and reduced AE risk; elevated WBC and blasts predicted higher AE incidence.

Conclusions:

  • A random forest model using routine hematological parameters shows significant potential for predicting CR and AE in pAML.
  • The model facilitates early risk stratification and personalized treatment approaches for pediatric AML.
  • Interpretable ML models can effectively leverage routine clinical data for improved pAML management.
Abstract

Related Concept Videos