GCF: Generalized Causal Forest for Heterogeneous Treatment Effect
  Estimation in Online Marketplace
  Uplift modeling is a rapidly growing approach that utilizes machine learning
and causal inference methods to estimate the heterogeneous treatment effects.
It has been widely adopted and applied to online marketplaces to assist
large-scale decision-making in recent years. The existing popular methods, like
forest-based modeling, either work only for discrete treatments or make
partially linear or parametric assumptions that may suffer from model
misspecification. To alleviate these problems, we extend causal forest (CF)
with non-parametric dose-response functions (DRFs) that can be estimated
locally using a kernel-based doubly robust estimator. Moreover, we propose a
distance-based splitting criterion in the functional space of conditional DRFs
to capture the heterogeneity for the continuous treatments. We call the
proposed algorithm generalized causal forest (GCF) as it generalizes the use
case of CF to a much broader setup. We show the effectiveness of GCF by
comparing it to popular uplift modeling models on both synthetic and real-world
datasets. We implement GCF in Spark and successfully deploy it into DiDi's
real-time pricing system. Online A/B testing results further validate the
superiority of GCF.
