Prediction-powered Generalization of Causal Inferences
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. Cited by 14 papers in the map (data through 7 Oct 2026). Builds on 24 of its 36 references that are in the map. First posted on arXiv 5 Jun 2024.
Builds on
24 of 36 references.
References · 24
- 2016Learning Representations for Counterfactual Inferencecited by 327
- 2023Prediction-Powered Inferencecited by 214
- 2018Removing Hidden Confounding by Experimental Groundingcited by 64
- 2020Causal inference methods for combining randomized trials and observational studies: a reviewcited by 50
- 2022Combining Observational and Randomized Data for Estimating Heterogeneous Treatment Effectscited by 32
- 2018Combining multiple observational data sources to estimate causal effectscited by 29
- 2020Combining Observational and Experimental Datasets Using Shrinkage Estimatorscited by 28
- 2017Generalizing causal inferences from individuals in randomized trials to all trial-eligible individualscited by 24
- 2021Adaptive Combination of Randomized and Observational Datacited by 22
- 2018Extending inferences from a randomized trial to a new target populationcited by 19
- 2020Elastic Integrative Analysis of Randomized Trial and Real-World Data for Treatment Heterogeneity Estimationcited by 17
- 2021Multi-Source Causal Inference Using Control Variatescited by 13
12 more
- 2022Detecting hidden confounding in observational data using multiple environmentscited by 13
- 2023Multiply Robust Federated Estimation of Targeted Average Treatment Effectscited by 13
- 2022Falsification before Extrapolation in Causal Effect Estimationcited by 10
- 2023Falsification of Internal and External Validity in Observational Studies via Conditional Moment Restrictionscited by 10
- 2021Minimax Rates and Adaptivity in Combining Experimental and Observational Datacited by 8
- 2023Hidden yet quantifiable: A lower bound for confounding strength using randomized trialscited by 8
- 2024Detecting critical treatment effect bias in small subgroupscited by 7
- 2023Prognostic Adjustment with Efficient Estimators to Unbiasedly Leverage Historical Data in Randomized Trialscited by 6
- 2022Understanding the Risks and Rewards of Combining Unbiased and Possibly Biased Estimators, with Applications to Causal Inferencecited by 4
- 2024Benchmarking Observational Studies with Experimental Data under Right-Censoringcited by 4
- 2021Designing Experiments Informed by Observational Studiescited by 3
- 2021Conditional Cross-Design Synthesis Estimators for Generalizability in Medicaidcited by 2
Built on by
Citing papers per year: 2024: 4; 2025: 8; 2026: 2.
14 papers.
Sort:
By area:
- 2025Efficient Randomized Experiments Using Foundation Modelscited by 10
- 2025Prediction-Powered Causal Inferencescited by 9
- 2026Multiple-Prediction-Powered Inferencecited by 9
- 2025No Free Lunch: Non-Asymptotic Analysis of Prediction-Powered Inferencecited by 8
- 2024Another look at statistical inference with machine learning-imputed datacited by 6
- 2024Constructing Confidence Intervals for Average Treatment Effects from Multiple Datasetscited by 6
- 2025A Unified Framework for Inference with General Missingness Patterns and Machine Learning Imputationcited by 5
- 2025How can we assess human-agent interactions? Case studies in software agent designcited by 4
- 2024Meta-Analysis with Untrusted Datacited by 2
- 2025Do We Really Even Need Data? A Modern Look at Drawing Inference with Predicted Datacited by 2
- 2024Accounting for Missing Covariates in Heterogeneous Treatment Estimationcited by 0
- 2025Uncovering Bias Mechanisms in Observational Studiescited by 0
- 2026Artificial intelligence surrogates for treatment effect estimation with before-and-after data23 Sep 2026
- 2026Multiple-Prediction-Powered Inference28 Mar 2026
- 2025Do We Really Even Need Data? A Modern Look at Drawing Inference with Predicted Data5 Dec 2025
- 2025Semi-Supervised Treatment Effect Estimation with Unlabeled Covariates for Prediction-Powered Causal Inference11 Nov 2025
- 2025How can we assess human-agent interactions? Case studies in software agent design10 Oct 2025
- 2025A Unified Framework for Inference with General Missingness Patterns and Machine Learning Imputation21 Aug 2025
- 2025Uncovering Bias Mechanisms in Observational Studies1 Jun 2025
- 2025No Free Lunch: Non-Asymptotic Analysis of Prediction-Powered Inference26 May 2025
- 2025Prediction-Powered Causal Inferences10 Feb 2025
- 2025Efficient Randomized Experiments Using Foundation Models6 Feb 2025
- 2024Constructing Confidence Intervals for Average Treatment Effects from Multiple Datasets16 Dec 2024
- 2024Another look at statistical inference with machine learning-imputed data29 Nov 2024
2 more
- 2024Accounting for Missing Covariates in Heterogeneous Treatment Estimation21 Oct 2024
- 2024Meta-Analysis with Untrusted Data12 Jul 2024
- 2025Efficient Randomized Experiments Using Foundation Modelscited by 10
- 2025Prediction-Powered Causal Inferencescited by 9
- 2026Multiple-Prediction-Powered Inferencecited by 9
- 2025No Free Lunch: Non-Asymptotic Analysis of Prediction-Powered Inferencecited by 8
- 2024Another look at statistical inference with machine learning-imputed datacited by 6
- 2024Constructing Confidence Intervals for Average Treatment Effects from Multiple Datasetscited by 6
- 2025A Unified Framework for Inference with General Missingness Patterns and Machine Learning Imputationcited by 5
- 2024Meta-Analysis with Untrusted Datacited by 2
- 2025Do We Really Even Need Data? A Modern Look at Drawing Inference with Predicted Datacited by 2
- 2024Accounting for Missing Covariates in Heterogeneous Treatment Estimationcited by 0
- 2025Uncovering Bias Mechanisms in Observational Studiescited by 0
- 2025Semi-Supervised Treatment Effect Estimation with Unlabeled Covariates for Prediction-Powered Causal Inferencecited by 0
12 most cited of 13 All 13 (JSON)
- 2026Artificial intelligence surrogates for treatment effect estimation with before-and-after data23 Sep 2026
- 2026Multiple-Prediction-Powered Inference28 Mar 2026
- 2025Do We Really Even Need Data? A Modern Look at Drawing Inference with Predicted Data5 Dec 2025
- 2025Semi-Supervised Treatment Effect Estimation with Unlabeled Covariates for Prediction-Powered Causal Inference11 Nov 2025
- 2025A Unified Framework for Inference with General Missingness Patterns and Machine Learning Imputation21 Aug 2025
- 2025Uncovering Bias Mechanisms in Observational Studies1 Jun 2025
- 2025No Free Lunch: Non-Asymptotic Analysis of Prediction-Powered Inference26 May 2025
- 2025Prediction-Powered Causal Inferences10 Feb 2025
- 2025Efficient Randomized Experiments Using Foundation Models6 Feb 2025
- 2024Constructing Confidence Intervals for Average Treatment Effects from Multiple Datasets16 Dec 2024
- 2024Another look at statistical inference with machine learning-imputed data29 Nov 2024
- 2024Accounting for Missing Covariates in Heterogeneous Treatment Estimation21 Oct 2024
12 newest of 13 All 13 (JSON)