Survival analysis under label shift
Abstract
Let P represent the source population with complete data, containing covariate and response , and Q the target population, where only the covariate is available. We consider a setting with both label shift and label censoring. Label shift assumes that the marginal distribution of differs between and , while the conditional distribution of given remains the same. Label censoring refers to the case where the response in is subject to random censoring. Our goal is to leverage information from the label-shifted and label-censored source population to conduct statistical inference in the target population . We propose a parametric model for given in and estimate the model parameters by maximizing an approximate likelihood. This allows for statistical inference in and accommodates a range of classical survival models. Under the label shift assumption, the likelihood depends not only on the unknown parameters but also on the unknown distribution of in and in , which we estimate nonparametrically. The asymptotic properties of the estimator are rigorously established and the effectiveness of the method is demonstrated through simulations and a real data application. This work is the first to combine survival analysis with label shift, offering a new research direction in this emerging topic.
Cite
@article{arxiv.2506.21190,
title = {Survival analysis under label shift},
author = {Yuxiang Zong and Yanyuan Ma and Ingrid Van Keilegom},
journal= {arXiv preprint arXiv:2506.21190},
year = {2025}
}