English

Deep Neural Networks for Survival Analysis Based on a Multi-Task Framework

Machine Learning 2018-01-18 v1 Machine Learning

Abstract

Survival analysis/time-to-event models are extremely useful as they can help companies predict when a customer will buy a product, churn or default on a loan, and therefore help them improve their ROI. In this paper, we introduce a new method to calculate survival functions using the Multi-Task Logistic Regression (MTLR) model as its base and a deep learning architecture as its core. Based on the Concordance index (C-index) and Brier score, this method outperforms the MTLR in all the experiments disclosed in this paper as well as the Cox Proportional Hazard (CoxPH) model when nonlinear dependencies are found.

Keywords

Cite

@article{arxiv.1801.05512,
  title  = {Deep Neural Networks for Survival Analysis Based on a Multi-Task Framework},
  author = {Stephane Fotso},
  journal= {arXiv preprint arXiv:1801.05512},
  year   = {2018}
}
R2 v1 2026-06-22T23:47:24.346Z