English

Survival analysis under label shift

Methodology 2025-06-27 v1

Abstract

Let P represent the source population with complete data, containing covariate Z\mathbf{Z} and response TT, and Q the target population, where only the covariate Z\mathbf{Z} is available. We consider a setting with both label shift and label censoring. Label shift assumes that the marginal distribution of TT differs between PP and QQ, while the conditional distribution of Z\mathbf{Z} given TT remains the same. Label censoring refers to the case where the response TT in PP is subject to random censoring. Our goal is to leverage information from the label-shifted and label-censored source population PP to conduct statistical inference in the target population QQ. We propose a parametric model for TT given Z\mathbf{Z} in QQ and estimate the model parameters by maximizing an approximate likelihood. This allows for statistical inference in QQ 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 TT in PP and Z\mathbf{Z} in QQ, 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.

Keywords

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}
}
R2 v1 2026-07-01T03:34:22.748Z