Abstract
We propose recursively imputed survival tree (RIST) regression for right-censored data. This new nonparametric regression procedure uses a novel recursive imputation approach combined with extremely randomized trees that allows significantly better use of censored data than previous tree-based methods, yielding improved model fit and reduced prediction error. The proposed method can also be viewed as a type of Monte Carlo EM algorithm, which generates extra diversity in the tree-based fitting process. Simulation studies and data analyses demonstrate the superior performance of RIST compared with previous methods.
Original language | English (US) |
---|---|
Pages (from-to) | 331-340 |
Number of pages | 10 |
Journal | Journal of the American Statistical Association |
Volume | 107 |
Issue number | 497 |
DOIs | |
State | Published - 2012 |
Externally published | Yes |
Keywords
- Censored data
- Ensemble
- Imputation
- Random forests
- Survival analysis
- Trees
ASJC Scopus subject areas
- Statistics and Probability
- Statistics, Probability and Uncertainty