Timezone: »

 
Poster
Sample Constrained Treatment Effect Estimation
Raghavendra Addanki · David Arbour · Tung Mai · Cameron Musco · Anup Rao

Tue Nov 29 09:00 AM -- 11:00 AM (PST) @ Hall J #330
in Poster Session 1 »
Treatment effect estimation is a fundamental problem in causal inference. We focus on designing efficient randomized controlled trials, to accurately estimate the effect of some treatment on a population of $n$ individuals. In particular, we study \textit{sample-constrained treatment effect estimation}, where we must select a subset of $s \ll n$ individuals from the population to experiment on. This subset must be further partitioned into treatment and control groups. Algorithms for partitioning the entire population into treatment and control groups, or for choosing a single representative subset, have been well-studied. The key challenge in our setting is jointly choosing a representative subset and a partition for that set. We focus on both individual and average treatment effect estimation, under a linear effects model. We give provably efficient experimental designs and corresponding estimators, by identifying connections to discrepancy minimization and leverage-score-based sampling used in randomized numerical linear algebra. Our theoretical results obtain a smooth transition to known guarantees when $s$ equals the population size. We also empirically demonstrate the performance of our algorithms.
Keywords: Causal Inference · experimental design · average treatment effect estimation · individual treatment effect estimation · randomized numerical linear algebra · discrepancy minimization
[ Paper]  [ Poster]  [ OpenReview] 

Author Information

Raghavendra Addanki (Adobe Research)
David Arbour (Adobe Research)
Tung Mai (Adobe Research)
Cameron Musco (University of Massachusetts Amherst)
Anup Rao (School of Computer Science, Georgia Tech)

More from the Same Authors