EconBase
← All papers

Leveraging LLMs to Improve Experimental Design: A Generative Stratification Approach

George Gui, Seungwoo Kim

arXiv 30 Sep 2025 · Econometrics

arXiv:2509.25709 · PDF · Extracted main text

Abstract

Pre-experiment stratification, or blocking, is a well-established technique for designing more efficient experiments and increasing the precision of the experimental estimates. However, when researchers have access to many covariates at the experiment design stage, they often face challenges in effectively selecting or weighting covariates when creating their strata. This paper proposes a Generative Stratification procedure that leverages Large Language Models (LLMs) to synthesize high-dimensional covariate data to improve experimental design. We demonstrate the value of this approach by applying it to a set of experiments and find that our method would have reduced the variance of the treatment effect estimate by 10%-50% compared to simple randomization in our empirical applications. When combined with other standard stratification methods, it can be used to further improve the efficiency. Our results demonstrate that LLM-based simulation is a practical and easy-to-implement way to improve experimental design in covariate-rich settings.

Citation extraction

33
references
56
in-text mentions
33
distinct cited
2
self-citations
5,454
main-text words

appendix boundary found by appendix_command · 96% of the source is main text. Read the extracted text to check this.

Most heavily cited references

The works this paper leans on most, across its whole bibliography — not restricted to papers in our corpus. Ranked by composite intensity, which combines how often a work is mentioned, how many sections mention it, and how much of that falls in the main text rather than the appendix.

ReferenceIntensityMentionsSectionsMain text
1Bai, Y (2022) Optimality of matched-pair designs in randomized controlled trials1.000104100%
2Greevy, R., Lu, B., Silber, J. H., and Rosenbaum, P (2004) Optimal multivariate matching before randomization0.92844100%
3Athey, S. and Imbens, G. W (2017) The econometrics of randomized experiments0.92843100%
4Bruhn, M. and McKenzie, D (2009) In pursuit of balance: Randomization in practice in development field experiments0.64422100%
5Morgan, K. L. and Rubin, D. B (2012) Rerandomization to improve covariate balance in experiments0.64422100%
6Barrera-Osorio, F., Linden, L. L., and Saavedra, J. E (2019) Medium- and long-term educational consequences of alternative conditional cash transfer designs: Experimental evidence from colo…0.58531100%
7de Mel, S., McKenzie, D., and Woodruff, C (2019) Labor drops: Experimental evidence on the return to additional labor in microenterprises0.58531100%
8Abel, M., Burger, R., and Piraino, P (2020) The value of reference letters: Experimental evidence from south africa0.51121100%
9Gerber, A., Hoffman, M., Morgan, J., and Raymond, C (2020) One in a million: Field experiments on perceived closeness of the election and voter turnout0.51121100%
10Aher, G. V., Arriaga, R. I., and Kalai, A. T (2023) Using Large Language Models to Simulate Multiple Humans and Replicate Human Subject Studies0.40511100%

Showing the top 10 of 33 scored citations.

Cited by, within the corpus

arXiv econ.EM papers that cite this one, ranked by how heavily they lean on it.

Citing paperIntensityMentionsSections
1LLM Personas as a Substitute for Field Experiments in Method Benchmarking0.40511
2AI-Assisted Variance Reduction in Randomized Experiments0.40511