Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Transfer learning for causal forest

About

Transfer learning addresses the challenge of transfering knowledge from one domain to another. Traditional transfer learning focuses on adapting models trained on a source domain (with a lot of observations) to improve performance on a target domain (with few observations). In this work we consider the case of a model shift and we focus on the transfer learning applied to a causal forest namely HTERF. This causal forest aims to estimate the Conditional Average Treatment Effect (CATE). The approach considered is the offset method presented by Wang (2016) adapted to a causal context. This method relies on the use of intermediate models in order to estimate the offset between source and target distributions. Our main result is a bound on the CATE error of HTERF on target depending on the error of the intermediate models. Simulation studies show the good performances of this approach in different settings on simulations and on a real-world dataset.

B\'er\'enice-Alexia Jocteur, V\'eronique Maume-Deschamps, Pierre Ribereau• 2026

Related benchmarks

TaskDatasetResultRank
CATE estimation10D Multi-dimensional simulation Target
RMSE0.12
5
Conditional Average Treatment Effect estimation1D Simulation CATE (test)
RMSE0.003
4
ATE EstimationIHDP target sample modified simulation replications
ATE10.18
3
CATE estimationIHDP target sample modified per replication simulation replications
RMSE0.734
3
CATE estimation10D Multi-dimensional simulation Source--
1
Showing 5 of 5 rows

Other info

Follow for update