Prediction-powered Generalization of Causal Inferences
Ilker Demirel, Ahmed M. Alaa, Anthony Philippakis, David A. Sontag
Abstract
Causal inferences from a randomized controlled trial (RCT) may not pertain to a target population where some effect modifiers have a different distribution. Prior work studies generalizing the results of a trial to a target population with no outcome but covariate data available. We show how the limited size of trials makes generalization a statistically infeasible task, as it requires estimating complex nuisance functions. We develop generalization algorithms that supplement the trial data with a prediction model learned from an additional observational study (OS), without making any assumptions on the OS. We theoretically and empirically show that our methods facilitate better generalization when the OS is "high-quality", and remain robust when it is not, and e.g., have unmeasured confounding.
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 c7ee0860-eb49-4e52-86d5-6d041a666da3Cited by top-tier papers8
- Efficient Randomized Experiments Using Foundation ModelsPiersilvio De Bartolomeis, Javier Abad, Guanbo Wang, Konstantin Donhauser et al.NeurIPS 2025 · 22 citations
- Multiple-Prediction-Powered InferenceCharlie Cowen-Breen, Alekh Agarwal, Stephen Bates, William W. Cohen et al.ICLR 2026 · 11 citations
- Prediction-Powered Causal InferencesRiccardo Cadei, Ilker Demirel, Piersilvio De Bartolomeis, Lukas Lindorfer et al.NeurIPS 2025 · 9 citations
- No Free Lunch: Non-Asymptotic Analysis of Prediction-Powered InferencePranav Mani, Peng Xu, Zachary Lipton, Michael OberstICML 2026 · 8 citations
- A Unified Framework for the Transportability of Population-Level Causal MeasuresAhmed Boughdiri, Clément Berenfeld, Julie Josse, Erwan ScornetNeurIPS 2025 · 4 citations
Builds on3
- Multiply Robust Federated Estimation of Targeted Average Treatment EffectsLarry Han, Zhu Shen, José R. ZubizarretaNeurIPS 2023 · 26 citations
- Detecting hidden confounding in observational data using multiple environmentsRickard Karlsson, Jesse H. KrijtheNeurIPS 2023 · 22 citations
- Falsification before Extrapolation in Causal Effect EstimationZeshan M. Hussain, Michael Oberst, Ming-Chieh Shih, David A. SontagNeurIPS 2022 · 11 citations
Related papers
- Externally Valid Policy Evaluation from Randomized Trials Using Additional Observational DataSofia Ek, Dave ZachariahNeurIPS 2024
- A Two-Stage Pretraining-Finetuning Framework for Treatment Effect Estimation with Unmeasured ConfoundingChuan Zhou, Yaxuan Li, Chunyuan Zheng, Haiteng Zhang et al.KDD 2025 · 6 citations
- Learning Adjustment Sets from Observational and Limited Experimental DataSofia Triantafillou, Gregory F. CooperAAAI 2021 · 7 citations
- How and Why to Use Experimental Data to Evaluate Methods for Observational Causal InferenceAmanda Gentzel, Purva Pruthi, David D. JensenICML 2021 · 22 citations
- Quantifying Ignorance in Individual-Level Causal-Effect Estimates under Hidden ConfoundingAndrew Jesson, Sören Mindermann, Yarin Gal, Uri ShalitICML 2021 · 66 citations
