Learning Survival Distribution with Implicit Survival Function
Learning Survival Distribution with Implicit Survival Function
Yu Ling, Weimin Tan, Bo Yan
Proceedings of the Thirty-Second International Joint Conference on Artificial Intelligence
Main Track. Pages 3975-3983.
https://doi.org/10.24963/ijcai.2023/442
Survival analysis aims at modeling the relationship between covariates and event occurrence with some untracked (censored) samples. In implementation, existing methods model the survival distribution with strong assumptions or in a discrete time space for likelihood estimation with censorship, which leads to weak generalization. In this paper, we propose Implicit Survival Function (ISF) based on Implicit Neural Representation for survival distribution estimation without strong assumptions, and employ numerical integration to approximate the cumulative distribution function for prediction and optimization. Experimental results show that ISF outperforms the state-of-the-art methods in three public datasets and has robustness to the hyperparameter controlling estimation precision.
Keywords:
Machine Learning: ML: Other