使用随机森林进行异质性处理效应的估计与推断

Estimation and Inference of Heterogeneous Treatment Effects using Random Forests

Journal of the American Statistical Association · 2017
被引 2777 · 同刊同年前 1%
ABS 4

中文导读

本文开发了一种非参数因果森林方法,用于估计异质性处理效应,并证明了其渐近正态性,为随机森林在统计推断中的有效应用提供了理论基础。

Abstract

Many scientific and engineering challenges—ranging from personalized medicine to customized marketing recommendations—require an understanding of treatment effect heterogeneity. In this article, we develop a nonparametric causal forest for estimating heterogeneous treatment effects that extends Breiman’s widely used random forest algorithm. In the potential outcomes framework with unconfoundedness, we show that causal forests are pointwise consistent for the true treatment effect and have an asymptotically Gaussian and centered sampling distribution. We also discuss a practical method for constructing asymptotic confidence intervals for the true treatment effect that are centered at the causal forest estimates. Our theoretical results rely on a generic Gaussian theory for a large family of random forest algorithms. To our knowledge, this is the first set of results that allows any type of random forest, including classification and regression forests, to be used for provably valid statistical inference. In experiments, we find causal forests to be substantially more powerful than classical methods based on nearest-neighbor matching, especially in the presence of irrelevant covariates.

计量经济学机器学习因果推断统计学