Explainable machine learning can outperform Cox regression predictions and provide insights in breast cancer survival

Arturo Moncada-Torres*, Marissa C. van Maaren, Mathijs P. Hendriks, Sabine Siesling, Gijs Geleijnse

*Corresponding author for this work

Research output: Contribution to journalArticleAcademicpeer-review

4 Citations (Scopus)
5 Downloads (Pure)

Abstract

Cox Proportional Hazards (CPH) analysis is the standard for survival analysis in oncology. Recently, several machine learning (ML) techniques have been adapted for this task. Although they have shown to yield results at least as good as classical methods, they are often disregarded because of their lack of transparency and little to no explainability, which are key for their adoption in clinical settings. In this paper, we used data from the Netherlands Cancer Registry of 36,658 non-metastatic breast cancer patients to compare the performance of CPH with ML techniques (Random Survival Forests, Survival Support Vector Machines, and Extreme Gradient Boosting [XGB]) in predicting survival using the c-index. We demonstrated that in our dataset, ML-based models can perform at least as good as the classical CPH regression (c-index ∼0.63), and in the case of XGB even better (c-index ∼ 0.73). Furthermore, we used Shapley Additive Explanation (SHAP) values to explain the models’ predictions. We concluded that the difference in performance can be attributed to XGB’s ability to model nonlinearities and complex interactions. We also investigated the impact of specific features on the models’ predictions as well as their corresponding insights. Lastly, we showed that explainable ML can generate explicit knowledge of how models make their predictions, which is crucial in increasing the trust and adoption of innovative ML techniques in oncology and healthcare overall.

Original languageEnglish
Article number6968
JournalScientific reports
Volume11
Issue number1
Early online date26 Mar 2021
DOIs
Publication statusPublished - Dec 2021

Fingerprint

Dive into the research topics of 'Explainable machine learning can outperform Cox regression predictions and provide insights in breast cancer survival'. Together they form a unique fingerprint.

Cite this