ML-assisted Randomization Tests for Detecting Treatment Effects in A/B Experiments
Experimentation is widely utilized for causal inference and data-driven decision-making across disciplines. In an A/B experiment, for example, an online business randomizes two different treatments (e.g., website designs) to their customers and then aims to infer which treatment is better. In this paper, we construct randomization tests for complex treatment effects, including heterogeneity and interference. A key feature of our approach is the use of flexible machine learning (ML) models, where the test statistic is defined as the difference between the cross-validation errors from two ML models, one including the treatment variable and the other without it. This approach combines the predictive power of modern ML tools with the finite-sample validity of randomization procedures, enabling a robust and efficient way to detect complex treatment effects in experimental settings. We demonstrate this combined benefit both theoretically and empirically through applied examples.
Code (0)
등록된 구현이 없습니다.
Tasks
Causal InferenceDecision MakingMethods 이 논문이 사용한 방법론
Similar Papers 제목 키워드 기반
Fit CATE Once: Model-Assisted Randomization Tests Without Sample Splitting
Randomization tests and flexible treatment-effect models offer complementary strengths for analyzing data from randomized panel experiments: the former provide valid inference under the known assignment mechanism, while …
Inference under Covariate-Adaptive Randomization with Multiple Treatments
This paper studies inference in randomized controlled trials with covariate-adaptive randomization when there are multiple treatments. More specifically, we study inference about the average effect of one or more treatme…
regressionRevisiting Randomization with the Cube Method
We propose a novel randomization approach for randomized controlled trials (RCTs), based on the cube method developed by Deville and Till\'e (2004). The cube method allows for the selection of balanced samples across var…
Randomization Inference of Heterogeneous Treatment Effects under Network Interference
We develop randomization-based tests for heterogeneous treatment effects in the presence of network interference. Leveraging the exposure mapping framework, we study a broad class of null hypotheses that represent variou…
validRandomization Inference Tests for Shift-Share Designs
We consider the problem of inference in shift-share research designs. The choice between existing approaches that allow for unrestricted spatial correlation involves tradeoffs, varying in terms of their validity when the…
valid