Privacy-Protected Causal Survival Analysis Under Distribution Shift
Yi Liu, Alexander Levis, Ke Zhu, Shu Yang, Peter B. Gilbert, Larry Han
Abstract
Causal inference across multiple data sources can improve the generalizability and reproducibility of scientific findings. However, for time-to-event outcomes, data integration methods remain underdeveloped, especially when populations are heterogeneous and privacy constraints prevent direct data pooling. We propose a federated learning method for estimating target site-specific causal effects in multi-source survival settings. Our approach dynamically re-weights source contributions to correct for distributional shifts, while preserving privacy. Leveraging semiparametric efficiency theory under a site-specific exchangeability assumption, data-adaptive weighting and flexible machine learning, the method achieves double robustness, and it improves efficiency if at least one source site provides a consistent estimate. Through simulations and two real data applications: (i) multi-site randomized trials of monoclonal antibodies for HIV-1 prevention among cisgender men and transgender persons in the United States, Brazil, Peru, and Switzerland, as well as women in sub-Saharan Africa, and (ii) an analysis of sex disparities across biomarker groups for all-cause mortality using the "flchain" dataset, we demonstrate the validity, efficiency gains, and practical utility of the approach. Our findings highlight the promise of federated methods for efficient, privacy-preserving causal survival analysis under distribution shift.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext dfcef996-11f5-4c27-a85b-76e5b07ef921Cited by top-tier papers1
Ask how each one uses itBuilds on3
- Multiply Robust Federated Estimation of Targeted Average Treatment EffectsLarry Han, Zhu Shen, José R. ZubizarretaNeurIPS 2023 · 26 citations
- Multi-Source Conformal Inference Under Distribution ShiftYi Liu, Alexander Levis, Sharon-Lise T. Normand, Larry HanICML 2024 · 23 citations
- A Unified Framework for the Transportability of Population-Level Causal MeasuresAhmed Boughdiri, Clément Berenfeld, Julie Josse, Erwan ScornetNeurIPS 2025 · 4 citations
Related papers
- Collaborative Heterogeneous Causal Inference Beyond Meta-analysisTianyu Guo, Sai Praneeth Karimireddy, Michael I. JordanICML 2024 · 4 citations
- SurvITE: Learning Heterogeneous Treatment Effects from Time-to-Event DataAlicia Curth, Changhee Lee, Mihaela van der SchaarNeurIPS 2021 · 41 citations
- Federated Causal Inference on Multi-Site Observational Data via Propensity Score AggregationRémi Khellaf, Aurélien Bellet, julie JosseICML 2026 · 6 citations
- Fair Federated Survival AnalysisMd Mahmudur Rahman, Sanjay PurushothamAAAI 2025 · 1 citation
- An Adaptive Kernel Approach to Federated Learning of Heterogeneous Causal EffectsThanh Vinh Vo, Arnab Bhattacharyya, Young Lee, Tze-Yun LeongNeurIPS 2022 · 29 citations
