Graph Neural Network Causal Explanation via Neural Causal Models

Arman Behnam, Binghui Wang
7/12/2024
cs.LGcs.AIstat.ML

No official code found. Best implementation: ArmanBehnam/CXGNN (19 stars, MIT, last commit 2024-12-13).

Abstract

Graph neural network (GNN) explainers identify the important subgraph that ensures the prediction for a given graph. Until now, almost all GNN explainers are based on association, which is prone to spurious correlations. We propose {\name}, a GNN causal explainer via causal inference. Our explainer is based on the observation that a graph often consists of a causal underlying subgraph. {\name} includes three main steps: 1) It builds causal structure and the corresponding structural causal model (SCM) for a graph, which enables the cause-effect calculation among nodes. 2) Directly calculating the cause-effect in real-world graphs is computationally challenging. It is then enlightened by the recent neural causal model (NCM), a special type of SCM that is trainable, and design customized NCMs for GNNs. By training these GNN NCMs, the cause-effect can be easily calculated. 3) It uncovers the subgraph that causally explains the GNN predictions via the optimized GNN-NCMs. Evaluation results on multiple synthetic and real-world graphs validate that {\name} significantly outperforms existing GNN explainers in exact groundtruth explanation identification

View on arXivView PDF

Code Implementations(1)

ArmanBehnam/CXGNNCommunity100%
MIT
193PythonFeb 20, 20241 years agoMIT

Similar Papers

Cite this paper

@article{behnam2024graph,
  title  = {Graph Neural Network Causal Explanation via Neural Causal Models},
  author = {Arman Behnam and Binghui Wang},
  year   = {2024},
  eprint = {2407.09378},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2407.09378v1},
  journal = {ECCV 2024 2024}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2407.09378.svg)](https://codeofpaper.com/paper/2407.09378)

Discussion