Predicting In-Hospital Mortality After Acute Myeloid Leukemia Therapy: Through Supervised Machine Learning Algorithms

Despite careful patient selection, induction chemotherapy for acute myeloid leukemia (AML) is associated with a considerable risk for treatment-related mortality (5%-20%). We evaluated machine learning (ML) algorithms trained using factors available at the time of admission for AML therapy to predict death during the hospitalization.

We included AML discharges with age > 17 years who received inpatient chemotherapy from State Inpatient Database from Arizona, Florida, New York, Maryland, Washington, and New Jersey for years 2008-2014. The primary objective was to predict inpatient mortality in patients undergoing chemotherapy using covariates present before initiation of chemotherapy. ML algorithms logistic regression (LR), decision tree, and random forest were compared.

29,613 hospitalizations for patients with AML were included in the analysis each with 4,177 features. The median age was 58.9 (18-101) years, 13,689 (53.7%) were male, and 20,203 (69%) were White. The mean time from admission to chemotherapy was 3 days (95% CI, 2.9 to 3.1), and 2,682 (9.1%) died during the hospitalization. Both LR and random forest models achieved an area under the curve (AUC) score of 0.78, whereas decision tree achieved an AUC of 0.70. The baseline LR model with age yielded an AUC of 0.62. To clinically balance and minimize false positives, we selected a decision threshold of 0.7 and at this threshold, 51 of our test set of 5,923 could have potentially averted treatment-related mortality.

Using readily accessible variables, inpatient mortality of patients on track for chemotherapy to treat AML can be predicted through ML algorithms. The model also predicted inpatient mortality when tested on different data representations and paves the way for future research.