English

Prediction-powered Generalization of Causal Inferences

Machine Learning 2024-06-06 v1 Machine Learning

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.

Keywords

Cite

@article{arxiv.2406.02873,
  title  = {Prediction-powered Generalization of Causal Inferences},
  author = {Ilker Demirel and Ahmed Alaa and Anthony Philippakis and David Sontag},
  journal= {arXiv preprint arXiv:2406.02873},
  year   = {2024}
}

Comments

International Conference on Machine Learning (ICML), 2024

R2 v1 2026-06-28T16:53:51.816Z