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
相关论文
共 50 条
  • [21] Study of cardiovascular disease prediction model based on random forest in eastern China
    Li Yang
    Haibin Wu
    Xiaoqing Jin
    Pinpin Zheng
    Shiyun Hu
    Xiaoling Xu
    Wei Yu
    Jing Yan
    Scientific Reports, 10
  • [22] Experimental study and Random Forest prediction model of microbiome cell surface hydrophobicity
    Liu, Yong
    Tang, Shaoxun
    Fernandez-Lozano, Carlos
    Munteanu, Cristian R.
    Pazos, Alejandro
    Yu, Yi-zun
    Tan, Zhiliang
    Gonzalez-Diaz, Humberto
    EXPERT SYSTEMS WITH APPLICATIONS, 2017, 72 : 306 - 316
  • [23] Study on prediction model of liquid hold up based on random forest algorithm
    Liu, Jianyi
    Jiang, Lu
    Chen, Yizhao
    Liu, Zhibin
    Yuan, Hua
    Wen, Yimin
    CHEMICAL ENGINEERING SCIENCE, 2023, 268
  • [24] Study of cardiovascular disease prediction model based on random forest in eastern China
    Yang, Li
    Wu, Haibin
    Jin, Xiaoqing
    Zheng, Pinpin
    Hu, Shiyun
    Xu, Xiaoling
    Yu, Wei
    Yan, Jing
    SCIENTIFIC REPORTS, 2020, 10 (01)
  • [25] Which model is better in predicting the survival of laryngeal squamous cell carcinoma?: Comparison of the random survival forest based on machine learning algorithms to Cox regression: analyses based on SEER database
    Sun, Haili
    Wu, Shuangshuang
    Li, Shaoxiao
    Jiang, Xiaohua
    MEDICINE, 2023, 102 (10)
  • [26] Clinical pregnancy outcomes prediction in vitro fertilization women based on random forest prediction model: A nested case-control study
    Yang, Hongya
    Liu, Fang
    Ma, Yuan
    Di, Man
    MEDICINE, 2022, 101 (49) : E32232
  • [27] The development of a prediction model based on random survival forest for the prognosis of non- Hodgkin lymphoma: A prospective cohort study in China
    Li, Xiaosheng
    Yang, Zailin
    Li, Jieping
    Wang, Guixue
    Sun, Anlong
    Wang, Ying
    Zhang, Wei
    Liu, Yao
    Lei, Haike
    HELIYON, 2024, 10 (12)
  • [28] Random Survival Forest Machine Learning for the Prediction of Cardiovascular Events Among Patients With a Measured Lipoprotein(a) Level: A Model Development Study
    Lusk, Jay B.
    O'Brien, Emily C.
    Hammill, Bradley G.
    Li, Fan
    Mac Grory, Brian
    Patel, Manesh R.
    Pagidipati, Neha J.
    Shah, Nishant P.
    CIRCULATION-GENOMIC AND PRECISION MEDICINE, 2025, 18 (01):
  • [29] A prediction model based on random survival forest analysis of the overall survival of elderly female papillary thyroid carcinoma patients: a SEER-based study
    Lun, Yuqiang
    Yuan, Hao
    Ma, Pengwei
    Chen, Jiawei
    Lu, Peiheng
    Wang, Weilong
    Liang, Rui
    Zhang, Junjun
    Gao, Wei
    Ding, Xuerui
    Li, Siyu
    Wang, Zi
    Guo, Jianing
    Lu, Lianjun
    ENDOCRINE, 2024, 85 (02) : 598 - 600
  • [30] Penalized semiparametric Cox regression model on XGBoost and random survival forests
    Wang, Yating
    Su, Jinxia
    Zhao, Xuejing
    COMMUNICATIONS IN STATISTICS-SIMULATION AND COMPUTATION, 2023, 52 (07) : 3095 - 3103