Evaluating Random Forests for Survival Analysis using Prediction Error Curves

By U. Mogensen, Hemant Ishwaran, Thomas Alexander Gerds. Journal of Statistical Software, 2012

447 citations recorded by the source corpus.

Prediction error curves are increasingly used to assess and compare predictions in survival analysis. This article surveys the R package pec which provides a set of functions for efficient computation of prediction error curves. The software implements inverse probability of censoring weights to deal with right censored data and several variants of cross-validation to deal with the apparent error problem. In principle, all kinds of prediction models can be assessed, and the package readily supports most traditional regression modeling strategies, like Cox regression or additive hazard regression, as well as state of the art machine learning methods such as random forests, a nonparametric method which provides promising alternatives to traditional strategies in low and high-dimensional settings. We show how the functionality of pec can be extended to yet unsupported prediction models. As an example, we implement support for random forest prediction models based on the R-packages randomSurvivalForest and party. Using data of the Copenhagen Stroke Study we use pec to compare random forests to a Cox regression model derived from stepwise variable selection. Reproducible results on the user level are given for publicly available data from the German breast cancer study group.

Subfields: Artificial Intelligence, Statistics and Probability

View Paper: https://www.semanticscholar.org/paper/cf986ef388df2fa241b912d8fdbe8f61d8313b5d

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Lev V. Utkin
  2. Egor D. Satyukov
  3. Andrei V. Konstantinov

In Statistics and Probability

  1. Lev V. Utkin
  2. Egor D. Satyukov
  3. Andrei V. Konstantinov

Citations in the corpus