arxivstat.MLcs.LG2026-07-12
Representation Learning for Semiparametric Causal Mediation Analysis under No Essential Heterogeneity
Roberto Faleh, Sofia Morelli, Holger Brandt
We propose a two-stage estimator for structural mediation parameters that combines deep representation learning with G-estimation under the "no essential heterogeneity" (NEH) assumption. We call the method UNIT. In the first stage,TARNet estimates the heterogeneous effect of a ra…