Machine learning applications for prediction of relapse in childhood acute lymphoblastic leukemia.

Scientific reports 2017 AI 8 Explanations View Original
Original Paper (PDF)

Unable to display PDF. Download it here or view on PMC.

Plain-English Explanations
Page 1
The Relapse Problem in Childhood Leukemia

Acute lymphoblastic leukemia (ALL) is the most common cancer in children. With modern treatment protocols, over 90% of children in developed countries achieve long-term remission. But about 20% of children relapse - their leukemia returns after initial treatment - and these children have a much worse prognosis. Relapsed ALL is the leading cause of cancer death in pediatric patients.

The central challenge is that current methods of sorting children into risk groups - standard, intermediate, and high risk - are imprecise. A child classified as standard-risk may still relapse, while a high-risk child may be cured with standard therapy. Over- or under-treatment carries serious consequences: excessive chemotherapy causes long-term side effects including organ damage and secondary cancers, while under-treatment leads to relapse and death. Better tools to predict who will relapse are urgently needed.

Clinicians already use many individual variables to assess prognosis - white blood cell count at diagnosis, age, bone marrow blast percentages after treatment begins, gene fusions like BCR-ABL, and measurements of minimal residual disease (MRD, tiny amounts of residual leukemia cells remaining after treatment). The problem is that no one knows the best way to combine these variables into a single reliable prediction. This study used machine learning to find that combination.

TL;DR: About 20% of children with ALL relapse after initial treatment, and current risk stratification methods are too imprecise to identify them reliably. Better predictive tools are urgently needed.
Pages 1, 6
Study Population: 570 Children, 121 Relapses

The researchers retrospectively collected data from 661 children (under age 16) diagnosed with ALL at Guangzhou Women and Children's Medical Center between January 2008 and December 2015. After excluding patients with insufficient follow-up, 570 children were included in the study. Median follow-up was 31.87 months. During this period, 121 children (21.2%) relapsed after achieving initial complete remission.

Most patients (486 children) were treated under the GD-ALL-2008 protocol - the Guangdong Acute Lymphoblastic Leukemia 2008 standard treatment regimen, which is aligned with the Chinese Children's Leukemia Group protocol. The remaining 84 patients received other treatment protocols. This separation was used strategically: the 486 GD-ALL-2008 patients were used for model development and cross-validation, while the 84 patients on other protocols served as a completely independent validation set - a stringent test of whether the model generalizes beyond the training conditions.

All variables were collected from electronic medical records (EMR) without additional clinical assessments. Starting with all available clinical, biochemical, immunological, and cytogenetic variables measured at diagnosis or at day 33 of treatment, the researchers compiled 103 candidate predictors after removing variables with more than 50% missing data (such as MRD measurements, which were only collected as an add-on study and were missing for over half of patients).

TL;DR: 570 ALL children were followed for a median of 31.87 months, of whom 121 (21.2%) relapsed. Data came entirely from routine electronic medical records, making the model practically deployable without additional testing.
Pages 6-7
Machine Learning Pipeline: Feature Selection and Model Building

Four machine learning algorithms were compared: Random Forest (RF), Decision Tree (DT), Logistic Regression (LR), and Support Vector Machine (SVM). To evaluate each algorithm fairly and avoid overfitting (where a model performs well on training data but poorly on new patients), the researchers used a rigorous nested cross-validation approach: 100-fold Monte Carlo cross-validation (randomly splitting data 100 times into 70% training and 30% test sets) nested within 10-fold cross-validation for model training.

A practical challenge was class imbalance: only 121 of 570 patients relapsed. Machine learning algorithms tend to ignore minority classes (relapses) and simply predict everyone as non-relapse, achieving high accuracy but missing the cases that matter most. The researchers addressed this using SMOTE (Synthetic Minority Over-sampling Technique), which generates synthetic relapse examples during training to balance the dataset without simply duplicating existing cases.

For feature selection, the researchers used a forward selection algorithm: starting with the single most predictive feature and adding features one at a time, measuring at each step whether accuracy, sensitivity, and AUC improved meaningfully. This process identified the minimum set of features that achieved peak performance - avoiding unnecessary complexity while preserving predictive power. Features were ranked by how often they were selected across 100 training sets, providing a data-driven importance ranking rather than relying on expert opinion alone.

TL;DR: A rigorous 100-fold Monte Carlo cross-validation framework compared four algorithms; class imbalance was addressed with SMOTE; and forward feature selection identified the minimal set of features needed for peak performance.
Pages 2-3
Random Forest Outperforms Other Algorithms

When tested on all 103 features, Random Forest substantially outperformed the other three algorithms. Tree-based algorithms (RF and Decision Tree) showed better balance between sensitivity and specificity than Logistic Regression and SVM, which had very high specificity but poor sensitivity - meaning they rarely predicted relapse even when it was present. For a clinical relapse-prediction tool, missing relapses (low sensitivity) is the most dangerous failure mode.

Random Forest specifically showed: accuracy 0.831, sensitivity 0.767, specificity 0.895, and AUC 0.902. AUC (area under the receiver operating characteristic curve) is a summary measure of overall discrimination - an AUC of 0.5 means the model is no better than chance, while 1.0 is perfect. An AUC of 0.902 represents excellent discriminative ability, meaning the model could correctly rank 90% of patient pairs (one who relapsed vs. one who did not) in the right order of risk.

The forward feature selection process found that the model reached peak performance with just 14 features, after which adding more variables did not improve and sometimes worsened performance. This is important for practical use: a model requiring only 14 routine clinical measurements is far easier to implement and interpret than one requiring dozens of variables, especially in settings where complex molecular testing may not be available.

TL;DR: Random Forest achieved AUC 0.902 with 14 features, clearly outperforming Logistic Regression and SVM which showed dangerously low sensitivity for relapse prediction.
Pages 3-4
The 14 Predictive Features and Independent Validation

The 14 features selected for the final Random Forest model were: Age, WBC count, Hemoglobin (HB), Platelet count (PLT), Peripheral Heterotypic Cells (PHC), Lactic Dehydrogenase (LDH), bone marrow blasts at diagnosis (D0-BM), bone marrow blasts at day 33 (D33-BM), birth weight (BW), ferroprotein (FER), bone marrow blasts at day 15 (D15-BM), hepatomegaly (liver enlargement), splenomegaly (spleen enlargement), and BCR-ABL fusion gene status. All but one feature (D33-BM) are measured at the time of first diagnosis, making the model applicable very early in the treatment course.

On the first independent test set (150 patients from Monte Carlo cross-validation splits), the model achieved accuracy 0.827, sensitivity 0.756, specificity 0.897, and AUC 0.902. On the entirely independent set of 84 patients treated under different protocols, the model held up with accuracy 0.798, sensitivity 0.750, specificity 0.813, and AUC 0.904. The consistency between these two test sets - particularly the near-identical AUC of 0.902 vs. 0.904 - is a strong signal that the model is not just memorizing patterns from the training data but has learned genuine relationships that generalize to new patients.

The researchers also tested whether adding MRD measurements improved performance. Adding MRD on days 15 and 33 increased accuracy from 0.821 to 0.837 and AUC from 0.884 to 0.903 in the subset of patients with MRD data - a modest improvement that did not justify the cost and complexity given that MRD was missing for more than half of patients. This suggests the 14-feature EMR-based model is nearly as informative as one that includes expensive flow cytometry MRD testing.

TL;DR: 14 routine clinical variables - all measurable from standard blood tests and bone marrow assessments - form a model that achieves AUC 0.90 and generalizes well to patients treated under different protocols.
Pages 4-5
Performance Across Risk Groups - and Where Current Stratification Fails

A key clinical insight from this study is that current standard risk stratification is imperfect. In the independent test set of 84 patients, more than half of the relapsed children (15 out of 28) came from the standard-risk and intermediate-risk groups - patients clinicians might expect to do well. This confirms that the standard/intermediate/high-risk classification system misses a substantial fraction of children who go on to relapse.

The Random Forest model performed across all three risk groups with accuracy ranging from 0.699 (intermediate-risk) to 0.829 (standard-risk). Most notably, in the high-risk group, it correctly predicted 11 of 13 relapses (sensitivity 0.855) - the group where sensitivity matters most because high-risk patients face the most intense treatment decisions. Performance was somewhat lower in the intermediate-risk group, which showed the most heterogeneous disease biology.

The practical implication is that the Random Forest model could be used as a supplementary tool to existing risk stratification, identifying children within a given risk group who are actually at higher or lower relapse risk than average. This would enable more personalized treatment intensity - escalating therapy for standard-risk patients flagged as high-relapse-risk, and potentially de-escalating for high-risk patients identified as likely to remain in remission, reducing long-term treatment toxicity.

TL;DR: More than half of relapses occurred in standard- and intermediate-risk patients - groups clinicians expect to do well. The RF model correctly identified many of these cases, demonstrating value beyond standard risk classification.
Page 5
A Practical Tool for Personalized Treatment Decisions

One of the most significant strengths of this model is its practical accessibility. All 14 features are routinely collected from electronic medical records at diagnosis and during early treatment - no additional biopsies, gene sequencing, or specialized assays are required. This means the model could be integrated directly into existing clinical workflows in any hospital that treats childhood ALL, including resource-limited settings where complex molecular testing is unavailable.

The model's potential clinical applications are threefold. First, high-relapse-risk patients identified by the model could receive more intensive therapy or experimental treatments earlier, potentially improving cure rates. Second, low-relapse-risk patients currently receiving intensive high-risk protocols could be de-escalated to regimens with fewer side effects, preserving quality of life. Third, the ranked feature list provides clinicians with specific variables to monitor closely when making real-time treatment adjustments.

The researchers also note that this is the first published study using machine learning to predict childhood ALL relapse using only EMR-based clinical data. Previous machine learning relapse models used gene expression data or combined MRD with genetic markers, which are harder to obtain consistently. By achieving comparable accuracy (AUC 0.902 vs. previous models' 0.735-0.790) using only routine clinical data, this study demonstrates a more practical path to machine learning-assisted treatment personalization.

TL;DR: All 14 model features come from routine clinical records available at any hospital - no special tests needed. The model outperforms previous gene-based relapse prediction approaches while remaining immediately deployable in real clinical settings.
Page 5
Toward Machine Learning-Assisted Leukemia Treatment

This study demonstrates for the first time that a machine learning model built entirely from routine electronic medical record data can accurately predict relapse in childhood ALL. The Random Forest model with 14 features achieves an AUC of 0.902, outperforms all three comparison algorithms, and generalizes to patients treated under different protocols - a critical test of real-world applicability.

The limitations are significant and acknowledged by the authors. The study is retrospective, single-center, and the training set (336 patients) is relatively small. The SMOTE over-sampling method used to address class imbalance introduces synthetic data that may not perfectly represent the true biology of relapsed ALL. MRD - now considered one of the strongest predictors of ALL relapse - was unavailable for most patients and excluded from the model.

Future work should validate the model in a prospective multicenter study, incorporate MRD and genomic data as they become more routinely available, and explore whether the model's predictions translate into measurable improvements in clinical outcomes when used to guide treatment decisions. The ultimate goal is a computer-aided decision support tool that helps clinicians make more precise, personalized treatment choices for each child with ALL - reducing both relapse rates and treatment-related harm.

TL;DR: A 14-feature Random Forest model achieves AUC 0.902 for predicting ALL relapse using only routine clinical data - the first such model based on electronic medical records - pointing toward practical computer-aided treatment personalization in pediatric leukemia.
Citation: Open Access, 2017. Available at: PMC5547099.