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 the latter can capture complex patterns of effect heterogeneity. We develop model-assisted randomization tests that combine these strengths without sample splitting. The key idea is to estimate an unsigned version of the conditional average treatment effect (CATE) from the covariance structure of residualized outcomes, while leaving the realized assignments for randomization inference. The remaining sign can be chosen to best fit the observed outcomes. We establish identification and consistency for the proposed unsigned CATE estimators, as well as validity for the CATE-assisted randomization tests. Across synthetic and semi-synthetic experiments, the CATE-assisted randomization tests control Type I error and achieve higher power than covariate-adjusted and sample-split alternatives. Finally, we show that the assignment-free CATE estimates can be used to discover heterogeneous subgroups and test subgroup-specific treatment effects.
Code (0)
등록된 구현이 없습니다.
Similar Papers 제목 키워드 기반
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…
Causal InferenceDecision MakingRandomization Inference: Theory and Applications
We review approaches to statistical inference based on randomization. Permutation tests are treated as an important special case. Under a certain group invariance property, referred to as the ``randomization hypothesis,'…
validAsymptotic Validity and Finite-Sample Properties of Approximate Randomization Tests
Randomization tests rely on simple data transformations and possess an appealing robustness property. In addition to being finite-sample valid if the data distribution is invariant under the transformation, these tests c…
ClusteringvalidUnconditional Randomization Tests for Interference
When conducting causal inference or designing policy, researchers are often concerned with the existence and extent of interference between units, which may be influenced by factors such as distance, proximity, and conne…
Causal InferencevalidRandomization Tests for Conditional Group Symmetry
Symmetry plays a central role in the sciences, machine learning, and statistics. While statistical tests for the presence of distributional invariance with respect to groups have a long history, tests for conditional sym…