On the use of Harrell’s C for clinical risk prediction via random survival forests

• Published in: Expert systems with applications Vol. 63; pp. 450 - 459
• Main Authors: Schmid, Matthias, Wright, Marvin N., Ziegler, Andreas
• Format: Journal Article
• Language: English
• Published: Elsevier Ltd 30-11-2016
• Subjects: Concordance index; Event history analysis; Log-rank statistic; Random survival forests; Risk prediction; Split rules; Event history analysis; Log-rank statistic; Risk prediction; Random survival forests; Concordance index; Split rules
• ISSN: 0957-4174; 1873-6793
• DOI: 10.1016/j.eswa.2016.07.018

•Harrell’s C is proposed as a split criterion in random survival forests.•Split points of continuous predictor variables differ substantially between Harrell’s C and log-rank splitting.•The log-rank statistic has a stronger end-cut preference than Harrell’s C.•Harrell’s C outperforms log-rank splitting in smaller scale studies.•Harrell’s C outperforms log-rank splitting if the censoring rate is high.
Random survival forests (RSF) are a powerful method for risk prediction of right-censored outcomes in biomedical research. RSF use the log-rank split criterion to form an ensemble of survival trees. The most common approach to evaluate the prediction accuracy of a RSF model is Harrell’s concordance index for survival data (‘C index’). Conceptually, this strategy implies that the split criterion in RSF is different from the evaluation criterion of interest. This discrepancy can be overcome by using Harrell’s C for both node splitting and evaluation. We compare the difference between the two split criteria analytically and in simulation studies with respect to the preference of more unbalanced splits, termed end-cut preference (ECP). Specifically, we show that the log-rank statistic has a stronger ECP compared to the C index. In simulation studies and with the help of two medical data sets we demonstrate that the accuracy of RSF predictions, as measured by Harrell’s C, can be improved if the log-rank statistic is replaced by the C index for node splitting. This is especially true in situations where the censoring rate or the fraction of informative continuous predictor variables is high. Conversely, log-rank splitting is preferable in noisy scenarios. Both C-based and log-rank splitting are implemented in the R package ranger. We recommend Harrell’s C as split criterion for use in smaller scale clinical studies and the log-rank split criterion for use in large-scale ‘omics’ studies.