A gradient boosting algorithm for survival analysis via direct optimization of concordance index.
Title
A gradient boosting algorithm for survival analysis via direct optimization of concordance index.
Creator
Chen Yifei; Jia Zhenyu; Mercola Dan; Xie Xiaohui
Publisher
Computational and mathematical methods in medicine
Date
2013
1905-07
Description
Survival analysis focuses on modeling and predicting the time to an event of interest. Many statistical models have been proposed for survival analysis. They often impose strong assumptions on hazard functions, which describe how the risk of an event changes over time depending on covariates associated with each individual. In particular, the prevalent proportional hazards model assumes that covariates are multiplicatively related to the hazard. Here we propose a nonparametric model for survival analysis that does not explicitly assume particular forms of hazard functions. Our nonparametric model utilizes an ensemble of regression trees to determine how the hazard function varies according to the associated covariates. The ensemble model is trained using a gradient boosting method to optimize a smoothed approximation of the concordance index, which is one of the most widely used metrics in survival model performance evaluation. We implemented our model in a software package called GBMCI (gradient boosting machine for concordance index) and benchmarked the performance of our model against other popular survival models with a large-scale breast cancer prognosis dataset. Our experiment shows that GBMCI consistently outperforms other methods based on a number of covariate settings. GBMCI is implemented in R and is freely available online.
Subject
*Survival Analysis; Algorithms; Artificial Intelligence; Breast Neoplasms/*epidemiology/*mortality; Clinical; Databases; Decision Support Systems; Factual; Female; Humans; Internet; Models; Prognosis; Proportional Hazards Models; Software; Theoretical
Identifier
Rights
Article information provided for research and reference use only. All rights are retained by the journal listed under publisher and/or the creator(s).
Citation
Chen Yifei; Jia Zhenyu; Mercola Dan; Xie Xiaohui, “A gradient boosting algorithm for survival analysis via direct optimization of concordance index.,” NEOMED Bibliography Database, accessed September 28, 2023, https://neomed.omeka.net/items/show/4694.