Unbiased split variable selection for random survival forests using maximally selected rank statistics

被引:117
作者
Wright, Marvin N. [1 ]
Dankowski, Theresa [1 ]
Ziegler, Andreas [1 ,2 ,3 ,4 ]
机构
[1] Univ Lubeck, Inst Med Biometrie & Stat, Univ Klinikum Schleswig Holstein, Campus Lubeck,Ratzeburger Allee 160, D-23562 Lubeck, Germany
[2] Univ Lubeck, Zentrum Klin Studien, Lubeck, Germany
[3] Univ KwaZulu Natal, Sch Math Stat & Comp Sci, Pietermaritzburg, South Africa
[4] Deutsch Zentrum Herz Kreislauf Forsch, Lubeck, Germany
关键词
maximally selected statistics; random forests; rank statistics; survival analysis; trees; CHI-SQUARE STATISTICS; CLASSIFICATION;
D O I
10.1002/sim.7212
中图分类号
Q [生物科学];
学科分类号
07 ; 0710 ; 09 ;
摘要
The most popular approach for analyzing survival data is the Cox regression model. The Cox model may, however, be misspecified, and its proportionality assumption may not always be fulfilled. An alternative approach for survival prediction is random forests for survival outcomes. The standard split criterion for random survival forests is the log-rank test statistic, which favors splitting variables with many possible split points. Conditional inference forests avoid this split variable selection bias. However, linear rank statistics are utilized by default in conditional inference forests to select the optimal splitting variable, which cannot detect non-linear effects in the independent variables. An alternative is to use maximally selected rank statistics for the split point selection. As in conditional inference forests, splitting variables are compared on the p-value scale. However, instead of the conditional Monte-Carlo approach used in conditional inference forests, p-value approximations are employed. We describe several p-value approximations and the implementation of the proposed random forest approach. A simulation study demonstrates that unbiased split variable selection is possible. However, there is a trade-off between unbiased split variable selection and runtime. In benchmark studies of prediction performance on simulated and real datasets, the newmethod performs better than random survival forests if informative dichotomous variables are combined with uninformative variables with more categories and better than conditional inference forests if non-linear covariate effects are included. In a runtime comparison, the method proves to be computationally faster than both alternatives, if a simple p-value approximation is used. Copyright (C) 2017 JohnWiley & Sons, Ltd.
引用
收藏
页码:1272 / 1284
页数:13
相关论文
共 56 条
[1]  
[Anonymous], THESIS
[2]  
[Anonymous], 2014, INT STAT REV, DOI DOI 10.1111/insr.12016
[3]  
[Anonymous], STAT SOFTW NEWSL
[4]  
[Anonymous], J STAT SOFT IN PRESS
[5]  
[Anonymous], 2001, J. Clin. Microbiol
[6]  
[Anonymous], 2014, ARXIV14050352
[7]  
[Anonymous], ANN STAT
[8]   CONTROLLING THE FALSE DISCOVERY RATE - A PRACTICAL AND POWERFUL APPROACH TO MULTIPLE TESTING [J].
BENJAMINI, Y ;
HOCHBERG, Y .
JOURNAL OF THE ROYAL STATISTICAL SOCIETY SERIES B-STATISTICAL METHODOLOGY, 1995, 57 (01) :289-300
[9]   Maximally selected χ2 statistics for kx2 tables [J].
Betensky, RA ;
Rabinowitz, D .
BIOMETRICS, 1999, 55 (01) :317-320
[10]   A random forest guided tour [J].
Biau, Gerard ;
Scornet, Erwan .
TEST, 2016, 25 (02) :197-227