High Fidelity Image Counterfactuals with Probabilistic Causal Models
About
We present a general causal generative modelling framework for accurate estimation of high fidelity image counterfactuals with deep structural causal models. Estimation of interventional and counterfactual queries for high-dimensional structured variables, such as images, remains a challenging task. We leverage ideas from causal mediation analysis and advances in generative modelling to design new deep causal mechanisms for structured variables in causal models. Our experiments demonstrate that our proposed mechanisms are capable of accurate abduction and estimation of direct, indirect and total effects as measured by axiomatic soundness of counterfactuals.
Fabio De Sousa Ribeiro, Tian Xia, Miguel Monteiro, Nick Pawlowski, Ben Glocker• 2023
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Counterfactual Effectiveness | CCTA (coronary computed tomography angiography) (test) | Proximal Effectiveness Error4.36 | 27 | |
| Counterfactual Effectiveness | MIMIC Chest X-ray 192×192 (test) | |ΔAUC| (Sex)0.03 | 13 | |
| Age Prediction | MIMIC Chest X-ray 192x192 | MAE (yr)0.144 | 10 | |
| Sex Prediction | MIMIC Chest X-ray 192x192 | Absolute Delta AUC37 | 10 | |
| Disease prediction | MIMIC Chest X-ray 192x192 | |ΔAUC| (%)0.59 | 10 | |
| Race Prediction | MIMIC Chest X-ray 192x192 | Absolute Delta AUC (%)8.64 | 10 | |
| Counterfactual intervention evaluation | PadChest (test) | LLA16.1 | 9 | |
| Counterfactual intervention evaluation | CCTA (test) | NCPA17.27 | 9 | |
| Counterfactual Generation | MIMIC Chest X-ray 192x192 (test) | Composition MAE3.0954 | 4 |
Showing 9 of 9 rows