Current browse context:
stat.ML
Change to browse by:
References & Citations
Statistics > Machine Learning
Title: Wasserstein Random Forests and Applications in Heterogeneous Treatment Effects
(Submitted on 8 Jun 2020 (v1), last revised 15 Feb 2021 (this version, v3))
Abstract: We present new insights into causal inference in the context of Heterogeneous Treatment Effects by proposing natural variants of Random Forests to estimate the key conditional distributions. To achieve this, we recast Breiman's original splitting criterion in terms of Wasserstein distances between empirical measures. This reformulation indicates that Random Forests are well adapted to estimate conditional distributions and provides a natural extension of the algorithm to multivariate outputs. Following the philosophy of Breiman's construction, we propose some variants of the splitting rule that are well-suited to the conditional distribution estimation problem. Some preliminary theoretical connections are established along with various numerical experiments, which show how our approach may help to conduct more transparent causal inference in complex situations.
Submission history
From: Qiming Du [view email][v1] Mon, 8 Jun 2020 16:08:10 GMT (2664kb,D)
[v2] Fri, 23 Oct 2020 13:46:07 GMT (3757kb,D)
[v3] Mon, 15 Feb 2021 09:10:38 GMT (11435kb)
Link back to: arXiv, form interface, contact.