Multi-Source Survival Domain Adaptation
Ammar Shaker, Carolin Lawrence
Abstract
Survival analysis is the branch of statistics that studies the relation between the characteristics of living entities and their respective survival times, taking into account the partial information held by censored cases. A good analysis can, for example, determine whether one medical treatment for a group of patients is better than another. With the rise of machine learning, survival analysis can be modeled as learning a function that maps studied patients to their survival times. To succeed with that, there are three crucial issues to be tackled. First, some patient data is censored: we do not know the true survival times for all patients. Second, data is scarce, which led past research to treat different illness types as domains in a multi-task setup. Third, there is the need for adaptation to new or extremely rare illness types, where little or no labels are available. In contrast to previous multi-task setups, we want to investigate how to efficiently adapt to a new survival target domain from multiple survival source domains. For this, we introduce a new survival metric and the corresponding discrepancy measure between survival distributions. These allow us to define domain adaptation for survival analysis while incorporating censored data, which would otherwise have to be dropped. Our experiments on two cancer data sets reveal a superb performance on target domains, a better treatment recommendation, and a weight matrix with a plausible explanation.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers2
- The Boundaries of Fair AI in Medical Image Prognosis: A Causal PerspectiveThai-Hoang Pham, Jiayuan Chen, Seungyeon Lee, Yuanlong Wang et al.NeurIPS 2025 · 3 citations
- Doubly Protected Estimation for Survival Outcomes Utilizing External Controls for Randomized Clinical TrialsChenyin Gao, Shu Yang, Mingyang Shan, Wenyu Ye et al.ICML 2025
Builds on2
Related papers
- SurvITE: Learning Heterogeneous Treatment Effects from Time-to-Event DataAlicia Curth, Changhee Lee, Mihaela van der SchaarNeurIPS 2021 · 41 citations
- Deep Copula-Based Survival Analysis for Dependent Censoring with Identifiability GuaranteesWeijia Zhang, Chun Kai Ling, Xuanhui ZhangAAAI 2024 · 12 citations
- Fair and Interpretable Models for Survival AnalysisMd. Mahmudur Rahman, Sanjay PurushothamKDD 2022 · 11 citations
- An Effective Meaningful Way to Evaluate Survival ModelsShiang Qi, Neeraj Kumar, Mahtab Farrokh, Weijie Sun et al.ICML 2023 · 28 citations
- From Biased Selective Labels to Pseudo-Labels: An Expectation-Maximization Framework for Learning from Biased DecisionsTrenton Chang, Jenna WiensICML 2024 · 1 citation
