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