Caron, Alberto;
Baio, Gianluca;
Manolopoulou, Ioanna;
(2022)
Interpretable Deep Causal Learning for Moderation Effects.
arXiv
Preview |
Text
2206.10261v3.pdf - Other Download (1MB) | Preview |
Abstract
In this extended abstract paper, we address the problem of interpretability and targeted regularization in causal machine learning models. In particular, we focus on the problem of estimating individual causal/treatment effects under observed confounders, which can be controlled for and moderate the effect of the treatment on the outcome of interest. Black-box ML models adjusted for the causal setting perform generally well in this task, but they lack interpretable output identifying the main drivers of treatment heterogeneity and their functional relationship. We propose a novel deep counterfactual learning architecture for estimating individual treatment effects that can simultaneously: i) convey targeted regularization on, and produce quantify uncertainty around the quantity of interest (i.e., the Conditional Average Treatment Effect); ii) disentangle baseline prognostic and moderating effects of the covariates and output interpretable score functions describing their relationship with the outcome. Finally, we demonstrate the use of the method via a simple simulated experiment.
Type: | Working / discussion paper |
---|---|
Title: | Interpretable Deep Causal Learning for Moderation Effects |
Open access status: | An open access version is available from UCL Discovery |
Publisher version: | https://doi.org/10.48550/arXiv.2206.10261 |
Language: | English |
Additional information: | This version is the version of record. For information on re-use, please refer to the publisher’s terms and conditions. |
Keywords: | cs.LG, cs.LG, cs.AI, stat.ML |
UCL classification: | UCL UCL > Provost and Vice Provost Offices > UCL BEAMS UCL > Provost and Vice Provost Offices > UCL BEAMS > Faculty of Maths and Physical Sciences UCL > Provost and Vice Provost Offices > UCL BEAMS > Faculty of Maths and Physical Sciences > Dept of Statistical Science |
URI: | https://discovery-pp.ucl.ac.uk/id/eprint/10178273 |
Archive Staff Only
View Item |