Title: "DeepHit: A Deep Learning Approach to Survival Analysis with Competing Risks"
Authors: Changhee Lee, William R. Zame, Jinsung Yoon, Mihaela van der Schaar
- Reference: C. Lee, W. R. Zame, J. Yoon, M. van der Schaar, "DeepHit: A Deep Learning Approach to Survival Analysis with Competing Risks," AAAI Conference on Artificial Intelligence (AAAI), 2018
- Paper: http://medianetlab.ee.ucla.edu/papers/AAAI_2018_DeepHit
- Supplementary: http://medianetlab.ee.ucla.edu/papers/AAAI_2018_DeepHit_Appendix
This code shows the modified implementation of DeepHit on Metabric (single risk) and Synthetic (competing risks) datasets.
The detailed modifications are as follows:
- Hyper-parameter opimization using random search is implemented
- Residual connections are removed
- The definition of the time-dependent C-index is changed; please refer to T.A. Gerds et al, "Estimating a Time-Dependent Concordance Index for Survival Prediction Models with Covariate Dependent Censoring," Stat Med., 2013
- Set "EVAL_TIMES" to a list of evaluation times of interest for optimizating the network with respect these evaluation times.