Efficient Learning of Survival Analysis Models Under Outcome-Dependent Truncation and Censoring

A new arXiv paper (2610.12538v1) proves that Cox regression can be efficiently learned under outcome-dependent truncation and right-censoring using first-order optimization when the baseline hazard is known, despite the problem's nonconvexity. The authors show that when the baseline hazard is learned jointly, identifiable and correctly specified models can contain strict suboptimal local minima in the negative log-likelihood. Across experiments on four survival datasets, accounting for truncation reduced the prediction bias of naive Cox regression, extended to neural networks, and delivered orders of magnitude speedups over standard expectation maximization.

arXiv:2610.12538v1 Announce Type: new Abstract: Modern machine learning models are typically trained on data assumed to be representative of the predictive tasks the model will face in the future, but practical datasets violate this assumption. We focus on outcome-dependent truncation in which an observation appears in the training data depending on its label. Our work connects two largely distinct literatures: efficient learning under truncation and survival analysis. Existing methods for truncated survival analysis do not characterize the optimization landscape and instead rely on iterative procedures such as expectation maximization. We address this gap and prove that Cox regression can be efficiently learned under truncated and right-censored data using first-order optimization when the baseline hazard is known despite nonconvexity . In contrast, when the baseline hazard is learned jointly, we construct identifiable and correctly specified models whose negative log-likelihood contains strict suboptimal local minima. On experiments with four survival datasets, we show that accounting for truncation reduces the prediction bias induced by naive Cox regression and extends to neural networks. We also show orders of magnitude speedups compared to standard expectation maximization.