EconBase
← All papers

A Comparison of Methods for Treatment Assignment with an Application to Playlist Generation

Carlos Fernández-Loría, Foster Provost, Jesse Anderton, Benjamin Carterette, Praveen Chandar

arXiv 24 Apr 2020 · Econometrics · publishedInformation Systems Research (2022) · 11 citations (OpenAlex)

arXiv:2004.11532 · PDF · DOI · OpenAlex · Extracted main text

Abstract

This study presents a systematic comparison of methods for individual treatment assignment, a general problem that arises in many applications and has received significant attention from economists, computer scientists, and social scientists. We group the various methods proposed in the literature into three general classes of algorithms (or metalearners): learning models to predict outcomes (the O-learner), learning models to predict causal effects (the E-learner), and learning models to predict optimal treatment assignments (the A-learner). We compare the metalearners in terms of (1) their level of generality and (2) the objective function they use to learn models from data; we then discuss the implications that these characteristics have for modeling and decision making. Notably, we demonstrate analytically and empirically that optimizing for the prediction of outcomes or causal effects is not the same as optimizing for treatment assignments, suggesting that in general the A-learner should lead to better treatment assignments than the other metalearners. We demonstrate the practical implications of our findings in the context of choosing, for each user, the best algorithm for playlist generation in order to optimize engagement. This is the first comparison of the three different metalearners on a real-world application at scale (based on more than half a billion individual treatment assignments). In addition to supporting our analytical findings, the results show how large A/B tests can provide substantial value for learning treatment assignment policies, rather than simply choosing the variant that performs best on average.

Citation extraction

50
references
101
in-text mentions
50
distinct cited
1
self-citations
15,827
main-text words

appendix boundary found by none_found · 100% of the source is main text. Read the extracted text to check this.

Most heavily cited references

The works this paper leans on most, across its whole bibliography — not restricted to papers in our corpus. Ranked by composite intensity, which combines how often a work is mentioned, how many sections mention it, and how much of that falls in the main text rather than the appendix.

ReferenceIntensityMentionsSectionsMain text
1Alina Beygelzimer and John Langford (2009) The offset tree for learning with partial labels. In Proceedings of the 15th ACM SIGKDD international conference on Knowledge di…1.00094100%
2Lihong Li, Wei Chu, John Langford, and Robert E Schapire (2010) A contextual-bandit approach to personalized news article recommendation. In Proceedings of the 19th international conference on…1.00053100%
3Baqun Zhang, Anastasios A Tsiatis, Marie Davidian, Min Zhang, and Er… (2012) Estimating optimal treatment regimes from a classification perspective1.00053100%
4Susan Athey and Guido Imbens (2016) Recursive partitioning for heterogeneous causal effects0.92843100%
5Susan Athey and Stefan Wager (2021) Policy learning with observational data0.87462100%
6Toru Kitagawa and Aleksey Tetenov (2018) Who should be treated? empirical welfare maximization methods for treatment choice0.87462100%
7Diego Olaya, Kristof Coussement, and Wouter Verbeke (2020) A survey and benchmarking study of multitreatment uplift modeling0.87452100%
8Yingqi Zhao, Donglin Zeng, A John Rush, and Michael R Kosorok (2012) Estimating individualized treatment rules using outcome weighted learning0.81142100%
9Adam N Elmachtoub and Paul Grigas (2021) Smart “predict, then optimize”0.73732100%
10Charles F Manski (2004) Statistical treatment rules for heterogeneous populations0.73732100%

Showing the top 10 of 50 scored citations.

Cited by, within the corpus

arXiv econ.EM papers that cite this one, ranked by how heavily they lean on it.

Citing paperIntensityMentionsSections
1Unveiling the Potential of Robustness in Selecting Conditional Average Treatment Effect Estimators0.40511