Transfer learning for causal forest - Archive ouverte HAL
Pré-Publication, Document De Travail Année : 2024

Transfer learning for causal forest

Résumé

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.
Fichier principal
Vignette du fichier
sample.pdf (596.14 Ko) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-04700817 , version 1 (19-09-2024)

Identifiants

  • HAL Id : hal-04700817 , version 1

Citer

Bérénice-Alexia Jocteur, Véronique Maume-Deschamps, Pierre Ribereau. Transfer learning for causal forest. 2024. ⟨hal-04700817⟩
24 Consultations
31 Téléchargements

Partager

More