Comparison of Random Survival Forest and Cox Model for Prediction Performance: A Case Study

被引:0
作者
Oliveira, Tiago A. [1 ]
Silva, Pedro Augusto F. [2 ]
Martins, Hiago Jose A. A. [2 ]
Pereira, Lucas C. [2 ]
Brito, Alisson L. [3 ]
Mendonca, Edndrio B. [1 ]
机构
[1] UEPB, Dept Estat, Campina Grande, PB, Brazil
[2] UEPB Univ Estadual Paraiba, Campina Grande, PB, Brazil
[3] Univ Fed Lavras, Programa Pos Grad Estat & Expt Agr, Lavras, MG, Brazil
来源
SIGMAE | 2019年 / 8卷 / 02期
关键词
Survival Analysis; Proportional risks; Machine Learning;
D O I
暂无
中图分类号
O1 [数学];
学科分类号
0701 ; 070101 ;
摘要
Survival analysis is currently one of the fastest growing statistical tools in academia. In survival analysis there is a robust regression model theory that can be used to model data with incomplete observations called censoring, most of these models are parametric, and there is also the Cox proportional hazards model. Machine Learning in conjunction with Random Forest in Survival Analysis (RSF) are an increasing alternative for use in prediction. Four different configurations of coefficients were adjusted in the RSF, starting from a saturated model with presence of interaction to a parsimonious model based on the criteria of the Machine Learning area to choose variables. The models were compared against the Cox model using the C-index and Brier Score Index (IBS) criteria. The best model adjusted for prediction was the complete model with all covariates under Random Survival Forest modeling.
引用
收藏
页码:490 / 508
页数:19
相关论文
共 11 条
[1]  
[Anonymous], COMPUTATIONAL MATH M
[2]   A review of survival trees [J].
Bou-Hamad, Imad ;
Larocque, Denis ;
Ben-Ameur, Hatem .
STATISTICS SURVEYS, 2011, 5 :44-71
[3]   Random forests [J].
Breiman, L .
MACHINE LEARNING, 2001, 45 (01) :5-32
[4]  
Carvalho M., 2005, Analise de Sobrevida Teoria e Aplicacoes em Saude
[5]  
COLOSSIMO E. A, 2006, ANALISE SOBREVIVENCI
[6]  
EHRLINGER J., 2016, ANN APPL STAT, V1, P1
[7]  
Ishwaran H., 2007, R. R News, V7, P25
[8]   RANDOM SURVIVAL FORESTS [J].
Ishwaran, Hemant ;
Kogalur, Udaya B. ;
Blackstone, Eugene H. ;
Lauer, Michael S. .
ANNALS OF APPLIED STATISTICS, 2008, 2 (03) :841-860
[9]  
Mogensen UB, 2012, J STAT SOFTW, V50, P1
[10]  
Pournelle G. H., 1953, Journal of Mammalogy, V34, P133