NeuralSurv: Deep Survival Analysis with Bayesian Uncertainty Quantification

Neural Information Processing Systems 

We introduce, the first deep survival model to incorporate Bayesian uncertainty quantification. Our non parametric, architecture agnostic framework flexibly captures time varying covariate-risk relationships in continuous time via a novel two stage data augmentation scheme, for which we establish theoretical guarantees. For efficient posterior inference, we introduce a mean field variational algorithm with coordinate ascent updates that scale linearly in model size. By locally linearizing the Bayesian neural network, we obtain full conjugacy and derive all coordinate updates in closed form. In experiments, delivers superior calibration compared to state-of-the-art deep survival models, while matching or exceeding their discriminative performance across both synthetic benchmarks and real-world datasets. Our results demonstrate the value of Bayesian principles in data scarce regimes by enhancing model calibration and providing robust, well calibrated uncertainty estimates for the survival function.