English

High-dimensional regression adjustments in randomized experiments

Methodology 2022-06-08 v3 Machine Learning

Abstract

We study the problem of treatment effect estimation in randomized experiments with high-dimensional covariate information, and show that essentially any risk-consistent regression adjustment can be used to obtain efficient estimates of the average treatment effect. Our results considerably extend the range of settings where high-dimensional regression adjustments are guaranteed to provide valid inference about the population average treatment effect. We then propose cross-estimation, a simple method for obtaining finite-sample-unbiased treatment effect estimates that leverages high-dimensional regression adjustments. Our method can be used when the regression model is estimated using the lasso, the elastic net, subset selection, etc. Finally, we extend our analysis to allow for adaptive specification search via cross-validation, and flexible non-parametric regression adjustments with machine learning methods such as random forests or neural networks.

Keywords

Cite

@article{arxiv.1607.06801,
  title  = {High-dimensional regression adjustments in randomized experiments},
  author = {Stefan Wager and Wenfei Du and Jonathan Taylor and Robert Tibshirani},
  journal= {arXiv preprint arXiv:1607.06801},
  year   = {2022}
}

Comments

To appear in the Proceedings of the National Academy of Sciences. The present draft does not reflect final copyediting by the PNAS staff

R2 v1 2026-06-22T15:01:59.340Z