Generalized random forests

From MaRDI portal
Publication:666599

DOI10.1214/18-AOS1709zbMATH Open1418.62102arXiv1610.01271OpenAlexW2962727190MaRDI QIDQ666599FDOQ666599


Authors: Susan Athey, Julie Tibshirani, Stefan Wager Edit this on Wikidata


Publication date: 6 March 2019

Published in: The Annals of Statistics (Search for Journal in Brave)

Abstract: We propose generalized random forests, a method for non-parametric statistical estimation based on random forests (Breiman, 2001) that can be used to fit any quantity of interest identified as the solution to a set of local moment equations. Following the literature on local maximum likelihood estimation, our method considers a weighted set of nearby training examples; however, instead of using classical kernel weighting functions that are prone to a strong curse of dimensionality, we use an adaptive weighting function derived from a forest designed to express heterogeneity in the specified quantity of interest. We propose a flexible, computationally efficient algorithm for growing generalized random forests, develop a large sample theory for our method showing that our estimates are consistent and asymptotically Gaussian, and provide an estimator for their asymptotic variance that enables valid confidence intervals. We use our approach to develop new methods for three statistical tasks: non-parametric quantile regression, conditional average partial effect estimation, and heterogeneous treatment effect estimation via instrumental variables. A software implementation, grf for R and C++, is available from CRAN.


Full work available at URL: https://arxiv.org/abs/1610.01271




Recommendations




Cites Work


Cited In (only showing first 100 items - show all)

Uses Software





This page was built for publication: Generalized random forests

Report a bug (only for logged in users!)Click here to report a bug for this page (MaRDI item Q666599)