Treatment Stratification of Patients with Metastatic Castration-Resistant Prostate Cancer by Machine Learning

iScience 2020 Machine Learning 6 Explanations View Original
Original Paper (PDF)

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

Plain-English Explanations
Pages 1-2
The Challenge of Predicting Docetaxel Response in Advanced Prostate Cancer

Metastatic castration-resistant prostate cancer (mCRPC) is an advanced stage of prostate cancer that has spread to other organs and no longer responds to hormone deprivation therapy. It remains one of the most difficult cancers to treat, with limited median survival after the disease becomes castration-resistant.

Docetaxel is a standard chemotherapy agent used in mCRPC, but its benefit is not uniform across patients. Some men experience meaningful survival extensions, while others suffer significant side effects without clinical benefit. Identifying which patients will respond to docetaxel before starting treatment would allow clinicians to prioritize it for likely responders and spare non-responders from unnecessary toxicity.

Clinical data collected during large drug trials contains rich information about patient characteristics, laboratory values, and medical history that may predict treatment outcomes. The DREAM Prostate Cancer Challenge was a community-wide data science competition that provided researchers with access to this clinical trial data to develop and compare predictive models for docetaxel response.

TL;DR: Predicting which mCRPC patients will respond to docetaxel chemotherapy could personalize treatment and spare non-responders from unnecessary toxicity.
Pages 3-5
Random Forest Models Trained on Three Clinical Trial Cohorts

The winning team trained a random forest classifier, an ensemble machine learning method that builds many decision trees on random subsets of the data and combines their predictions, to stratify patients by their likelihood of benefiting from docetaxel treatment.

Training data came from three large clinical trial cohorts totaling approximately 1,600 patients. Using multiple cohorts was important because each trial enrolled patients under slightly different eligibility criteria and collected data in subtly different ways. A model trained and validated only within a single trial might not generalize to clinical practice.

The outcome variable being predicted was overall survival, specifically whether patients survived longer than expected on docetaxel therapy. The model was designed not just to predict who lives longer overall, but to identify a treatment-effect subgroup: patients whose survival is specifically extended by the drug rather than by favorable baseline characteristics alone.

Feature engineering involved preparing over 100 baseline clinical variables including laboratory values, vital signs, disease history, and demographics. The random forest's built-in feature importance scores were then used to identify which variables contributed most to the model's predictions.

TL;DR: A random forest trained across 1,600 patients from three clinical trial cohorts predicted which mCRPC patients would benefit from docetaxel chemotherapy.
Pages 6-8
Top Predictive Features: Albumin, Sodium, and Metabolic Markers

The ten most important features identified by the random forest model were: albumin (ALB), sodium (NA), total protein (TPRO), magnesium (MG), testosterone (TESTO), white blood cell count (WBC), phosphate (PHOS), medical history of vascular disease (MHVASC), neutrophil count (NEU), and medical history of social issues (MHSOCIAL).

Several of these features reflect general health and nutritional status rather than cancer-specific biology. Low albumin and low sodium are well-established markers of overall frailty and systemic inflammation, conditions that independently predict poor outcomes in cancer patients. Their prominence suggests the model is partly capturing whether patients are robust enough to both tolerate and benefit from aggressive chemotherapy.

The inclusion of testosterone is biologically interesting in the context of castration-resistant disease, where serum testosterone is expected to be suppressed. Residual testosterone levels or variability in androgen signaling may still influence chemotherapy responsiveness even after castration, a hypothesis consistent with emerging research on intratumoral androgen synthesis.

TL;DR: Albumin, sodium, and total protein topped the list of predictive features, suggesting overall health status is as important as cancer-specific markers for predicting docetaxel benefit.
Pages 8-10
Model Performance and the DREAM Challenge Evaluation Framework

The primary performance metric for the DREAM Challenge was the Area Under the Precision-Recall Curve (AUPRC), also called average precision. Unlike the more familiar AUC of the ROC curve, AUPRC is specifically suited for imbalanced datasets where the positive class (patients who benefit from treatment) is a minority, making it a more stringent and clinically relevant measure.

The winning model achieved an AUPRC of 0.190 compared to a baseline of 0.104 for a simple reference model, representing a substantial relative improvement. While an AUPRC of 0.190 may appear low in absolute terms, this reflects the inherent difficulty of the prediction task: identifying a treatment-responsive subgroup in a heterogeneous advanced cancer population is genuinely hard.

The DREAM Challenge framework included independent test sets not seen by any competing team during model development, ensuring that reported performance reflected true generalization rather than overfitting. Multiple competing teams submitted models, and the winning solution was selected based on performance across all test cohorts.

TL;DR: The winning model achieved AUPRC of 0.190 versus a baseline of 0.104, evaluated on independent test sets through the rigorous DREAM Challenge framework.
Pages 11-13
Translating a Predictive Model Into Clinical Decision Support

A model that stratifies patients by predicted docetaxel benefit could be integrated into clinical decision support tools that help oncologists and patients make more informed treatment choices. Rather than replacing physician judgment, such a model would provide a quantitative estimate of expected benefit to supplement clinical assessment.

Patients predicted to derive low benefit from docetaxel could be considered for alternative regimens such as cabazitaxel, abiraterone, enzalutamide, or clinical trial enrollment. For patients with limited performance status who are at high risk of chemotherapy toxicity, a model showing low predicted benefit could support a shared decision to defer or avoid docetaxel altogether.

Implementing such a model in practice requires that the input variables be reliably collected at the point of care, which is largely the case here since albumin, sodium, blood counts, and other laboratory values are routinely obtained before starting chemotherapy. The practical barrier is less about data availability and more about clinical trust, validation in prospective trials, and regulatory approval.

TL;DR: The predictive model could guide treatment selection in mCRPC by identifying patients unlikely to benefit from docetaxel, enabling earlier consideration of alternatives.
Pages 21-22
Machine Learning as a Tool for Precision Oncology in mCRPC

This study demonstrates that random forest machine learning applied to routinely collected clinical trial data can identify features that predict differential benefit from docetaxel in metastatic castration-resistant prostate cancer, outperforming standard clinical predictors.

The DREAM Challenge model provides a validated starting point for developing clinically deployable treatment stratification tools. However, further validation in independent prospective cohorts, particularly from community oncology settings rather than clinical trials, will be necessary before the model can be adopted as a standard decision support tool.

The success of this approach reinforces the broader promise of applying data-driven methods to existing clinical trial datasets. Large-scale trials that collect detailed baseline variables represent an underutilized resource for generating actionable predictive tools in oncology, and the DREAM Challenge model illustrates a reproducible framework for doing so.

TL;DR: Random forest machine learning can meaningfully stratify mCRPC patients by predicted docetaxel benefit using routine clinical variables, supporting precision treatment selection.
Citation: Open Access, . Available at: PMC6976944.