FastCPH: Efficient Survival Analysis for Neural NetworksDownload PDF

Published: 20 Oct 2022, Last Modified: 22 Oct 2023HITY Workshop NeurIPS 2022Readers: 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.
Community Implementations: [![CatalyzeX](/images/catalyzex_icon.svg) 1 code implementation](https://www.catalyzex.com/paper/arxiv:2208.09793/code)
3 Replies

Loading