2000 character limit reached
ICTSurF: Implicit Continuous-Time Survival Functions with Neural Networks (2312.05818v2)
Published 10 Dec 2023 in cs.LG
Abstract: Survival analysis is a widely known method for predicting the likelihood of an event over time. The challenge of dealing with censored samples still remains. Traditional methods, such as the Cox Proportional Hazards (CPH) model, hinge on the limitations due to the strong assumptions of proportional hazards and the predetermined relationships between covariates. The rise of models based on deep neural networks (DNNs) has demonstrated enhanced effectiveness in survival analysis. This research introduces the Implicit Continuous-Time Survival Function (ICTSurF), built on a continuous-time survival model, and constructs survival distribution through implicit representation. As a result, our method is capable of accepting inputs in continuous-time space and producing survival probabilities in continuous-time space, independent of neural network architecture. Comparative assessments with existing methods underscore the high competitiveness of our proposed approach. Our implementation of ICTSurF is available at https://github.com/44REAM/ICTSurF.
- Abien Fred Agarap “Deep Learning using Rectified Linear Units (ReLU)” arXiv, 2019 arXiv: http://arxiv.org/abs/1803.08375
- Alaa Ahmed M. and van der Schaar Mihaela “Deep Multi-task Gaussian Processes for Survival Analysis with Competing Risks” In Proceedings of the 31st International Conference on Neural Information Processing Systems, NIPS’17 Long Beach, California, USA: Curran Associates Inc., 2017, pp. 2326–2334
- Laura Antolini, Patrizia Boracchi and Elia Biganzoli “A time-dependent discrimination index for survival data” In Statistics in Medicine 24.24, 2005, pp. 3927–3944 DOI: 10.1002/sim.2427
- Peter C. Austin “Generating survival times to simulate Cox proportional hazards models with time-varying covariates” In Statistics in Medicine 31.29, 2012, pp. 3946–3958 DOI: 10.1002/sim.5452
- Achraf Bennis, Sandrine Mouysset and Mathieu Serrurier “Estimation of Conditional Mixture Weibull Distribution with Right Censored Data Using Neural Network for Time-to-Event Analysis” Series Title: Lecture Notes in Computer Science In Advances in Knowledge Discovery and Data Mining 12084 Cham: Springer International Publishing, 2020, pp. 687–698 DOI: 10.1007/978-3-030-47426-3_53
- D.R. Cox “Regression Models and Life-Tables” In Journal of the Royal Statistical Society: Series B (Methodological) 34.2, 1972, pp. 187–202 DOI: 10.1111/j.2517-6161.1972.tb00899.x
- “The genomic and transcriptomic architecture of 2,000 breast tumours reveals novel subgroups” In Nature 486.7403, 2012, pp. 346–352 DOI: 10.1038/nature10983
- “A neural network model for survival data” In Statistics in Medicine 14.1, 1995, pp. 73–82 DOI: 10.1002/sim.4780140108
- Michael F. Gensheimer and Balasubramanian Narasimhan “A scalable discrete-time survival model for neural networks” In PeerJ 7, 2019, pp. e6257 DOI: 10.7717/peerj.6257
- “Assessment and comparison of prognostic classification schemes for survival data” In Statistics in Medicine 18.17, 1999, pp. 2529–2545 DOI: 10.1002/(sici)1097-0258(19990915/30)18:17/18<2529::aid-sim274>3.0.co;2-5
- Frank E. Harrell “Evaluating the Yield of Medical Tests” In JAMA: The Journal of the American Medical Association 247.18, 1982, pp. 2543 DOI: 10.1001/jama.1982.03320430047030
- “Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift” arXiv, 2015 arXiv: http://arxiv.org/abs/1502.03167
- “DeepSurv: personalized treatment recommender system using a Cox proportional hazards deep neural network” In BMC Medical Research Methodology 18.1, 2018, pp. 24 DOI: 10.1186/s12874-018-0482-1
- “Time2Vec: Learning a Vector Representation of Time” arXiv, 2019 arXiv: http://arxiv.org/abs/1907.05321
- Diederik P. Kingma and Jimmy Ba “Adam: A Method for Stochastic Optimization” arXiv, 2017 arXiv: http://arxiv.org/abs/1412.6980
- William A. Knaus “The SUPPORT Prognostic Model: Objective Estimates of Survival for Seriously Ill Hospitalized Adults” In Annals of Internal Medicine 122.3, 1995, pp. 191 DOI: 10.7326/0003-4819-122-3-199502010-00007
- “Continuous and discrete-time survival prediction with neural networks” In Lifetime Data Analysis 27.4, 2021, pp. 710–736 DOI: 10.1007/s10985-021-09532-6
- Håvard Kvamme, Ørnulf Borgan and Ida Scheel “Time-to-Event Prediction with Neural Networks and Cox Regression” In Journal of Machine Learning Research 20.129, 2019, pp. 1–30 URL: http://jmlr.org/papers/v20/18-424.html
- Changhee Lee, Jinsung Yoon and Mihaela van der Schaar “Dynamic-DeepHit: A Deep Learning Approach for Dynamic Survival Analysis With Competing Risks Based on Longitudinal Data” In IEEE Transactions on Biomedical Engineering 67.1, 2020, pp. 122–133 DOI: 10.1109/TBME.2019.2909027
- “DeepHit: A Deep Learning Approach to Survival Analysis With Competing Risks” In Proceedings of the AAAI Conference on Artificial Intelligence 32.1, 2018 DOI: 10.1609/aaai.v32i1.11842
- Yu Ling, Weimin Tan and Bo Yan “Learning Survival Distribution with Implicit Survival Function” arXiv, 2023 arXiv: http://arxiv.org/abs/2305.14655
- Chirag Nagpal, Xinyu Li and Artur Dubrawski “Deep Survival Machines: Fully Parametric Survival Regression and Representation Learning for Censored Data With Competing Risks” In IEEE Journal of Biomedical and Health Informatics 25.8, 2021, pp. 3163–3175 DOI: 10.1109/JBHI.2021.3052441
- Chirag Nagpal, Willa Potosnak and Artur Dubrawski “auton-survival: an Open-Source Package for Regression, Counterfactual Estimation, Evaluation and Phenotyping with Censored Time-to-Event Data” arXiv, 2022 arXiv: http://arxiv.org/abs/2204.07276
- “PyTorch: An Imperative Style, High-Performance Deep Learning Library” arXiv, 2019 arXiv: http://arxiv.org/abs/1912.01703
- “On the C-statistics for evaluating overall adequacy of risk prediction procedures with censored survival data” In Statistics in Medicine 30.10, 2011, pp. 1105–1117 DOI: 10.1002/sim.4154
- “Attention is All you Need” In Proceedings of the 31st International Conference on Neural Information Processing Systems, NIPS’17 Long Beach, California, USA: Curran Associates Inc., 2017, pp. 6000–6010
- “SurvTRACE: transformers for survival analysis with competing events” In Proceedings of the 13th ACM International Conference on Bioinformatics, Computational Biology and Health Informatics Northbrook Illinois: ACM, 2022, pp. 1–9 DOI: 10.1145/3535508.3545521
- “Deep Learning for Survival Analysis: A Review” arXiv, 2023 arXiv: http://arxiv.org/abs/2305.14961