Search papers, labs, and topics across Lattice.
The paper introduces Survival Diffusion Probabilistic Model (SDPM), a generative approach leveraging denoising diffusion models for continuous-time survival analysis, estimating time-to-event distributions from censored data without imposing structural assumptions or discretizing the time axis. SDPM models the conditional distribution of observed time and censoring indicators, and uses the Kaplan-Meier estimator to transform generated samples into survival function estimates. Experiments on ten real-world datasets demonstrate SDPM's competitive performance against tree-based, boosting-based, and neural survival models, particularly in recovering continuous survival distributions.
Diffusion models can now predict when you're likely to kick the bucket, offering a flexible, continuous-time alternative to traditional survival analysis methods.
Survival analysis aims to estimate a time-to-event distribution from data with censored observations. Many existing methods either impose structural assumptions on the hazard function or discretize the time axis, which may limit flexibility and introduce approximation errors. We propose the Survival Diffusion Probabilistic Model (SDPM), a generative approach to continuous-time survival analysis. SDPM models the conditional distribution of the survival outcome, represented by the pair of observed time and censoring indicator, $\mathbb{P}(T,未\mid \mathbf{x})$, using a denoising diffusion model. Under the assumption of conditionally independent censoring, conditional samples generated by the model can be transformed into survival function estimates using the Kaplan-Meier estimator. This formulation avoids parametric assumptions on the event-time distribution and does not require a discretization of the output time space. The model operates in a transformed target space, using standardized log-times and a continuous Gaussian-mixture representation of the censoring indicator. We evaluate SDPM on ten real survival datasets and compare it with five strong baselines, including tree-based, boosting-based, and neural survival models. Results show that SDPM achieves competitive predictive performance across C-index, integrated time-dependent AUC, and integrated Brier score. A study on synthetic Cox-Weibull data demonstrates that SDPM can recover the shape of an underlying continuous survival distribution more accurately than a strong nonparametric baseline when sufficiently many samples are generated. An ablation study confirms the importance of the proposed target-space transformations, which improve event-rate calibration, reduce invalid generated times, and provide consistent gains in predictive discrimination. Codes implementing the proposed model are publicly available.
