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: Learning Objectives for Treatment Effect Estimation

Abstract: 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 to form an objective function that isolates the heterogeneous treatment effects, and then optimize the learned objective. This approach has several advantages over existing methods. From a practical perspective, our method is very flexible and easy to use: In both steps, we can use any method of our choice, e.g., penalized regression, a deep net, or boosting; moreover, these methods can be fine-tuned by cross-validating on the learned objective. Meanwhile, in the case of penalized kernel regression, we show that our method has a quasi-oracle property, whereby even if our pilot estimates for marginal effects and treatment propensities are not particularly accurate, we achieve the same regret bounds as an oracle who has a-priori knowledge of these nuisance components. We implement variants of our method based on both penalized regression and convolutional neural networks, and find promising performance relative to existing baselines.
Subjects: Machine Learning (stat.ML); Econometrics (econ.EM); Statistics Theory (math.ST)
Cite as: arXiv:1712.04912 [stat.ML]
  (or arXiv:1712.04912v1 [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.