FastCPH: Efficient Survival Analysis for Neural NetworksDownload PDF

Published: 02 Dec 2022, Last Modified: 17 Mar 2024TS4H PosterReaders: Everyone
Keywords: survival analysis
TL;DR: We propose FastCPH, a new method that runs in linear time and supports both the standard Breslow and Efron methods for tied events.
Abstract: The Cox proportional hazards model is a canonical method in survival analysis for prediction of the life expectancy of a patient given clinical or genetic covariates -- it is a linear model in its original form. In recent years, several methods have been proposed to generalize the Cox model to neural networks, but none of these are both numerically correct and computationally efficient. We propose FastCPH, a new method that runs in linear time and supports both the standard Breslow and Efron methods for tied events. We also demonstrate the performance of FastCPH combined with LassoNet, a neural network that provides interpretability through feature sparsity, on survival datasets. The final procedure is efficient, selects useful covariates and outperforms existing CoxPH approaches.
Supplementary Material: zip
Archival Submission: Yes
Student Authorship: Yes
Community Implementations: [![CatalyzeX](/images/catalyzex_icon.svg) 1 code implementation](https://www.catalyzex.com/paper/arxiv:2208.09793/code)
0 Replies

Loading