AI-Assisted Variance Reduction in Randomized Experiments
David Arbour, Eli Ben-Michael, Avi Feller, Apoorva Lal, Lo-Hua Yuan
Abstract
Generative AI and large language models can produce realistic predictions of human behavior from rich, unstructured inputs with little to no task-specific training data. Recent work uses these ''digital twin'' predictions to supplement human responses in surveys and experiments. We study the special case of using AI-generated predictions to reduce variance in randomized experiments. We argue that doing so requires no new estimators and that researchers can simply include AI predictions as covariates in standard regression adjustment, analogous to adjusting for a prognostic score. A benefit of this approach is a ''do no harm'' property whereby the adjusted estimator reverts to the unadjusted difference in means when predictions are uninformative. Other methods, such as variants of prediction-powered inference, do not have this guarantee. We provide implementation guidance, including how to obtain continuous scores from discrete LLM outputs and how to use LLMs to featurize unstructured inputs as auxiliary covariates. We demonstrate these ideas in simulations and three empirical applications: a survey mega-study, an email marketing A/B test, and a large-scale technology platform experiment. Overall, efficiency gains are real if modest, with greater benefits in studies that contain substantial text and other unstructured data. We also confirm the do no harm property empirically. Given these gains and limited costs, we recommend adjusting for AI-generated predictions as a regular empirical practice.
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 c81a62d5-4e06-418f-bf6b-7fd6f5944a07Builds on7
- TabPFN: A Transformer That Solves Small Tabular Classification Problems in a SecondNoah Hollmann, Samuel Müller, Katharina Eggensperger, Frank HutterICLR 2023 · 96 citations
- Using Imperfect Surrogates for Downstream Inference: Design-based Supervised Learning for Social Science Applications of Large Language ModelsNaoki Egami, Musashi Hinck, Brandon M. Stewart, Hanying WeiNeurIPS 2023 · 74 citations
- Machine Learning for Variance Reduction in Online ExperimentsYongyi Guo, Dominic Coey, Mikael Konutgan, Wenting Li et al.NeurIPS 2021 · 47 citations
- Efficient Randomized Experiments Using Foundation ModelsPiersilvio De Bartolomeis, Javier Abad, Guanbo Wang, Konstantin Donhauser et al.NeurIPS 2025 · 22 citations
- Valid Inference with Imperfect Synthetic DataYewon Byun, Shantanu Gupta, Zachary C. Lipton, Rachel Leah Childers et al.NeurIPS 2025 · 8 citations
Related papers
- Generative Augmented InferenceCheng Lu, Mengxin Wang, Dennis Zhang, Heng ZhangICML 2026 · 1 citation
- Prediction-Powered Adaptive Inference with Pretrained AI Models for Contextual BanditsGabriel Sargent, Wei Sun, Zhengwu Zhang, Yufeng LiuICML 2026
- Estimating Distributional Treatment Effects in Randomized Experiments: Machine Learning for Variance ReductionUndral Byambadalai, Tatsushi Oka, Shota YasuiICML 2024 · 7 citations
- Leveraging semantic similarity for experimentation with AI-generated treatmentsLei Shi, David Arbour, Raghavendra Addanki, Ritwik Sinha et al.NeurIPS 2025 · 1 citation
- Efficient Inference for Noisy LLM-as-a-Judge EvaluationYiqun Chen, Sizhu Lu, Sijia Li, Moran Guo et al.ICML 2026 · 5 citations
