Survival regression with accelerated failure time model in XGBoost

06/08/2020
by   Avinash Barnwal, et al.
0

Survival regression is used to estimate the relation between time-to-event and feature variables, and is important in application domains such as medicine, marketing, risk management and sales management. Nonlinear tree based machine learning algorithms as implemented in libraries such as XGBoost, scikit-learn, LightGBM, and CatBoost are often more accurate in practice than linear models. However, existing implementations of tree-based models have offered limited support for survival regression. In this work, we propose and implement loss functions for learning accelerated failure time (AFT) models in XGBoost, to increase the support for survival modeling for different kinds of label censoring. The AFT model assumes effects that directly accelerate or decelerate the survival time for different kinds of censored data sets. We demonstrate with real and simulated experiments the effectiveness of AFT in XGBoost with respect to a number of baselines, in two respects: generalization performance and training speed. Furthermore, we take advantage of the support for NVIDIA GPUs in XGBoost to achieve substantial speedup over multi-coreCPUs. To our knowledge, our work is the first implementation of AFT that utilizes the processing power of NVIDIA GPUs.

READ FULL TEXT

page 1

page 2

page 3

page 4

research
08/18/2020

mlr3proba: An R Package for Machine Learning in Survival Analysis

As machine learning has become increasingly popular over the last few de...
research
12/20/2019

Robust Estimation and Variable Selection for the Accelerated Failure Time Model

This paper considers robust modeling of the survival time for cancer pat...
research
05/04/2020

Survival Analysis of Organizational Networks – An Exploratory Study

Organizations interact with the environment and with other organizations...
research
06/13/2022

Deep Neural Network Based Accelerated Failure Time Models using Rank Loss

An accelerated failure time (AFT) model assumes a log-linear relationshi...
research
05/12/2020

Bayesian survival analysis with BUGS

Survival analysis is one of the most important fields of statistics in m...
research
09/24/2022

Concordance based Survival Cobra with regression type weak learners

In this paper, we predict conditional survival functions through a combi...

Please sign up or login with your details

Forgot password? Click here to reset