Double machine learning and design in batch adaptive experiments

Li Harrison H. () and Owen Art B. ()
Additional contact information
Li Harrison H.: Department of Statistics, Stanford University, Stanford, California, United States
Owen Art B.: Department of Statistics, Stanford University, Stanford, California, United States

Journal of Causal Inference, 2024, vol. 12, issue 1, 27

Abstract: We consider an experiment with at least two stages or batches and O ( N ) O\left(N) subjects per batch. First, we propose a semiparametric treatment effect estimator that efficiently pools information across the batches, and we show that it asymptotically dominates alternatives that aggregate single batch estimates. Then, we consider the design problem of learning propensity scores for assigning treatment in the later batches of the experiment to maximize the asymptotic precision of this estimator. For two common causal estimands, we estimate this precision using observations from previous batches, and then solve a finite-dimensional concave maximization problem to adaptively learn flexible propensity scores that converge to suitably defined optima in each batch at rate O p ( N − 1 ⁄ 4 ) {O}_{p}\left({N}^{-1/4}) . By extending the framework of double machine learning, we show this rate suffices for our pooled estimator to attain the targeted precision after each batch, as long as nuisance function estimates converge at rate o p ( N − 1 ⁄ 4 ) {o}_{p}\left({N}^{-1/4}) . These relatively weak rate requirements enable the investigator to avoid the common practice of discretizing the covariate space for design and estimation in batch adaptive experiments while maintaining the advantages of pooling. Our numerical study shows that such discretization often leads to substantial asymptotic and finite sample precision losses outweighing any gains from design.

Keywords: adaptive design; propensity score; pooled estimation; partially linear model; convex optimization (search for similar items in EconPapers)
Date: 2024
References: View references in EconPapers View complete reference list from CitEc
Citations:

Downloads: (external link)
https://doi.org/10.1515/jci-2023-0068 (text/html)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:bpj:causin:v:12:y:2024:i:1:p:27:n:1001

DOI: 10.1515/jci-2023-0068

Access Statistics for this article

Journal of Causal Inference is currently edited by Elias Bareinboim, Jin Tian and Iván Díaz

More articles in Journal of Causal Inference from De Gruyter
Bibliographic data for series maintained by Peter Golla ().