Multiply Robust Federated Estimation of Targeted Average Treatment Effects
Larry Han, Zhu Shen, José R. Zubizarreta
Abstract
Federated or multi-site studies have distinct advantages over single-site studies, including increased generalizability, the ability to study underrepresented populations, and the opportunity to study rare exposures and outcomes. However, these studies are challenging due to the need to preserve the privacy of each individual's data and the heterogeneity in their covariate distributions. We propose a novel federated approach to derive valid causal inferences for a target population using multi-site data. We adjust for covariate shift and covariate mismatch between sites by developing multiply-robust and privacy-preserving nuisance function estimation. Our methodology incorporates transfer learning to estimate ensemble weights to combine information from source sites. We show that these learned weights are efficient and optimal under different scenarios. We showcase the finite sample advantages of our approach in terms of efficiency and robustness compared to existing approaches. Recent methodological developments have focused on privacy-preserving estimation strategies. These strategies typically involve sharing summary-level information from multiple data sources [26, 27, 12, 13, 18] . However, they often require restrictive assumptions such as homogeneous data structures and model specifications (e.g., a common set of observed covariates measured using a common data model), which are not realistic in practice. To address these methodological gaps, we propose a multiply robust and privacy-preserving estimator that leverages multi-site information to estimate causal effects in a target population of interest.
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 f167c58a-db6c-48e0-a616-7679f4e1a863Cited by top-tier papers10
- Multi-Source Conformal Inference Under Distribution ShiftYi Liu, Alexander Levis, Sharon-Lise T. Normand, Larry HanICML 2024 · 23 citations
- Prediction-powered Generalization of Causal InferencesIlker Demirel, Ahmed M. Alaa, Anthony Philippakis, David A. SontagICML 2024 · 18 citations
- Doubly Robust Alignment for Large Language ModelsErhan Xu, Kai Ye, Hongyi Zhou, Luhan Zhu et al.NeurIPS 2025 · 14 citations
- Federated Causal Inference on Multi-Site Observational Data via Propensity Score AggregationRémi Khellaf, Aurélien Bellet, julie JosseICML 2026 · 6 citations
- Privacy-Protected Causal Survival Analysis Under Distribution ShiftYi Liu, Alexander Levis, Ke Zhu, Shu Yang et al.ICLR 2026 · 6 citations
Related papers
- Collaborative Heterogeneous Causal Inference Beyond Meta-analysisTianyu Guo, Sai Praneeth Karimireddy, Michael I. JordanICML 2024 · 4 citations
- Transfer Learning on Heterogeneous Feature Spaces for Treatment Effects EstimationIoana Bica, Mihaela van der SchaarNeurIPS 2022 · 33 citations
- An Adaptive Kernel Approach to Federated Learning of Heterogeneous Causal EffectsThanh Vinh Vo, Arnab Bhattacharyya, Young Lee, Tze-Yun LeongNeurIPS 2022 · 29 citations
- Privacy-Aware Data Integration for Enhanced Quantile Inference under HeterogeneityLeheng Cai, Qirui Hu, Shuyuan WuICML 2026
- Unveiling the Potential of Robustness in Selecting Conditional Average Treatment Effect EstimatorsYiyan Huang, Cheuk Hang Leung, Siyi Wang, Yijun Li et al.NeurIPS 2024 · 2 citations
