Wasserstein Random Forests and Applications in Heterogeneous Treatment Effects - Archive ouverte HAL
Pré-Publication, Document De Travail Année : 2022

Wasserstein Random Forests and Applications in Heterogeneous Treatment Effects

Résumé

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.

Dates et versions

hal-03735416 , version 1 (21-07-2022)

Identifiants

Citer

Qiming Du, Gérard Biau, François Petit, Raphaël Porcher. Wasserstein Random Forests and Applications in Heterogeneous Treatment Effects. 2022. ⟨hal-03735416⟩
48 Consultations
0 Téléchargements

Altmetric

Partager

More