2000 character limit reached
Neural Network-Based Piecewise Survival Models (2403.18664v1)
Published 27 Mar 2024 in stat.ML, cs.LG, cs.SY, and eess.SY
Abstract: In this paper, a family of neural network-based survival models is presented. The models are specified based on piecewise definitions of the hazard function and the density function on a partitioning of the time; both constant and linear piecewise definitions are presented, resulting in a family of four models. The models can be seen as an extension of the commonly used discrete-time and piecewise exponential models and thereby add flexibility to this set of standard models. Using a simulated dataset the models are shown to perform well compared to the highly expressive, state-of-the-art energy-based model, while only requiring a fraction of the computation time.
- Estimation of conditional mixture weibull distribution with right censored data using neural network for time-to-event analysis. In Advances in Knowledge Discovery and Data Mining: 24th Pacific-Asia Conference, PAKDD 2020, Singapore, May 11–14, 2020, Proceedings, Part I 24, pages 687–698. Springer, 2020.
- On the use of artificial neural networks for the analysis of survival data. IEEE transactions on neural networks, 8(5):1071–1077, 1997.
- Weibull recurrent neural networks for failure prognosis using histogram data. Neural Computing and Applications, 35(4):3011–3024, 2023.
- A neural network model for survival data. Statistics in medicine, 14(1):73–82, 1995.
- Robert Flynn. Survival analysis. Journal of Clinical Nursing, 21(19pt20):2789–2797, 2012.
- Energy-based survival models for predictive maintenance. IFAC-PapersOnLine, 56(2):10862–10867, 2023.
- Deepcompete: A deep learning approach to competing risks in continuous time domain. In AMIA annual symposium proceedings, volume 2020, page 177. American Medical Informatics Association, 2020.
- Deepsurv: personalized treatment recommender system using a cox proportional hazards deep neural network. BMC medical research methodology, 18:1–12, 2018.
- Continuous and discrete-time survival prediction with neural networks. Lifetime data analysis, 27:710–736, 2021.
- Deephit: A deep learning approach to survival analysis with competing risks. In Proceedings of the AAAI conference on artificial intelligence, volume 32, 2018.
- Attention-based deep survival model for time series data. Reliability Engineering & System Safety, 217:108033, 2022.
- Soden: A scalable continuous-time survival model through ordinary differential equation networks. Journal of Machine Learning Research, 23(34):1–29, 2022.
- Heavy-duty truck battery failure prognostics using random survival forests. IFAC-PapersOnLine, 49(11):562–569, 2016.
- Deep learning for survival analysis: A review. arXiv preprint arXiv:2305.14961, 2023.