We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:


Current browse context:


Change to browse by:

References & Citations


(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo ScienceWISE logo

Statistics > Machine Learning

Title: Quasi-Oracle Estimation of Heterogeneous Treatment Effects

Abstract: Flexible estimation of heterogeneous treatment effects lies at the heart of many statistical challenges, such as personalized medicine and optimal resource allocation. In this paper, we develop a general class of two-step algorithms for heterogeneous treatment effect estimation in observational studies. We first estimate marginal effects and treatment propensities in order to form an objective function that isolates the causal component of the signal. Then, we optimize this data-adaptive objective function. Our approach has several advantages over existing methods. From a practical perspective, our method is flexible and easy to use: In both steps, we can use any loss-minimization method, e.g., penalized regression, deep neural networks, or boosting; moreover, these methods can be fine-tuned by cross validation. Meanwhile, in the case of penalized kernel regression, we show that our method has a quasi-oracle property: Even if the pilot estimates for marginal effects and treatment propensities are not particularly accurate, we achieve the same error bounds as an oracle who has a priori knowledge of these two nuisance components. We implement variants of our approach based on penalized regression, kernel ridge regression, and boosting in a variety of simulation setups, and find promising performance relative to existing baselines.
Comments: Biometrika, forthcoming
Subjects: Machine Learning (stat.ML); Econometrics (econ.EM); Statistics Theory (math.ST)
Cite as: arXiv:1712.04912 [stat.ML]
  (or arXiv:1712.04912v4 [stat.ML] for this version)

Submission history

From: Xinkun Nie [view email]
[v1] Wed, 13 Dec 2017 18:32:13 GMT (100kb,D)
[v2] Mon, 25 Jun 2018 17:11:54 GMT (81kb,D)
[v3] Mon, 4 Feb 2019 16:51:33 GMT (85kb,D)
[v4] Thu, 6 Aug 2020 06:31:20 GMT (130kb,D)

Link back to: arXiv, form interface, contact.