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