Current browse context:
cs.LG
Change to browse by:
References & Citations
Computer Science > Machine Learning
Title: Causal Effect Inference for Structured Treatments
(Submitted on 3 Jun 2021 (v1), last revised 27 Oct 2021 (this version, v3))
Abstract: We address the estimation of conditional average treatment effects (CATEs) for structured treatments (e.g., graphs, images, texts). Given a weak condition on the effect, we propose the generalized Robinson decomposition, which (i) isolates the causal estimand (reducing regularization bias), (ii) allows one to plug in arbitrary models for learning, and (iii) possesses a quasi-oracle convergence guarantee under mild assumptions. In experiments with small-world and molecular graphs we demonstrate that our approach outperforms prior work in CATE estimation.
Submission history
From: Jean Kaddour [view email][v1] Thu, 3 Jun 2021 15:41:00 GMT (5940kb,D)
[v2] Fri, 18 Jun 2021 11:03:46 GMT (5942kb,D)
[v3] Wed, 27 Oct 2021 22:22:41 GMT (5895kb,D)
Link back to: arXiv, form interface, contact.