GCF: Generalized Causal Forest for Heterogeneous Treatment Effect Estimation Using Nonparametric Methods
Heterogeneous treatment effect (HTE) estimation with continuous treatment is essential in multiple disciplines, such as the online marketplace and pharmaceutical industry. The existing machine learning (ML) 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 kernel-based Double/Debiased ML estimators. Moreover, we propose a distance-based splitting criterion in the functional space of Partial DRFs to capture the heterogeneity for 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 compared to SOTA on synthetic data and proprietary real-world data sets.
Code (0)
등록된 구현이 없습니다.
Tasks
Heterogeneous Treatment Effect EstimationSimilar Papers 제목 키워드 기반
GCF: Generalized Causal Forest for Heterogeneous Treatment Effect Estimation in Online Marketplace
Uplift modeling is a rapidly growing approach that utilizes causal inference and machine learning methods to directly estimate the heterogeneous treatment effects, which has been widely applied to various online marketpl…
Causal InferenceDecision MakingHeterogeneous Treatment Effect EstimationEstimation and Inference of Heterogeneous Treatment Effects using Random Forests
Many scientific and engineering challenges -- ranging from personalized medicine to customized marketing recommendations -- require an understanding of treatment effect heterogeneity. In this paper, we develop a non-para…
MarketingvalidEstimating heterogeneous treatment effects with right-censored data via causal survival forests
Forest-based methods have recently gained in popularity for non-parametric treatment effect estimation. Building on this line of work, we introduce causal survival forests, which can be used to estimate heterogeneous tre…
Targeting relative risk heterogeneity with causal forests
The identification of heterogeneous treatment effects (HTE) across subgroups is of significant interest in clinical trial analysis. Several state-of-the-art HTE estimation methods, including causal forests, apply recursi…
causalfe: Causal Forests with Fixed Effects in Python
The causalfe package provides a Python implementation of Causal Forests with Fixed Effects (CFFE) for estimating heterogeneous treatment effects in panel data settings. Standard causal forest methods struggle with panel …