Machine Learning to Predict 5-Year Survival among Pediatric Acute Myeloid Leukemia Patients and Development of OSPAM-C Online Survival Prediction Tool

Authors: Das AK*, Mishra S, Mishra DK and Gopalan SS

DOI: 10.23880/jqhe-16000187

Volume 3, Issue 6

Abstract

Introduction: Acute myeloid leukemia (AML) accounts for a fifth of childhood leukemia. Although survival rates for AML have greatly improved over the past few decades, they vary depending on demographic and AML type factors. We predict the fiveyear survival among pediatric AML patients using machine learning algorithms and deploy the best performing algorithm as an online survival prediction tool. Methods: Pediatric patients (0 to 14 years) with microscopically confirmed AML were extracted from the Surveillance Epidemiology and End Results (SEER) database (2000-2011) and randomly split into training and test datasets (80/20 ratio). Four machine learning algorithms (logistic regression, support vector machine, gradient boosting, and K nearest neighbor) were trained on features to predict five-year survival. Performances of the algorithms were compared and the best performing algorithm was deployed as an online prediction tool. Results: A total of 1,477 patients met our inclusion criteria. The gradient boosting algorithm was the best performer in terms of discrimination and predictive ability. It was deployed as the online survival prediction tool named OSPAM-C (https://ashisdas. shinyapps.io/ospam/). Conclusions: Our study provides a framework for the development and deployment of an online survival prediction tool for pediatric patients with AML. While external validation is needed, our survival prediction tool presents an opportunity to reach informed clinical decision-making for AML patients.

Keywords: Myeloid Leukemia; Cancer; Surveillance Epidemiology; Patients

View PDF