Controllable GNN Explanations via Multi-Metric Preference Selection
Organizations: Indian Institute of Technology Gandhinagar
Abstract
Mechanisms for generating GNN explanations are crucial for building trust and mitigating biases in Graph Neural Networks (GNNs), especially in high-stakes scenarios. Most current methods optimize only for fidelity under the sparsity constraint. However, this discounts the need for interpretable explanations (those that consist of familiar motif patterns) and stable explanations (those that remain unchanged under structural perturbations). We propose a novel approach that optimizes GNN explanations across these metrics, exposing their relative weighing as a control. Experiments on various real-world datasets, including MUTAG, BA-2Motif, BAMultiShapes, and PROTEINS, suggest that our method produces higher-fidelity explanations than a state-of-the-art baseline on MUTAG and PROTEINS across all evaluated budgets, and on BA-2Motif at larger budgets, while being faster in the regime of small explanation budgets. They also indicate that our method produces more stable explanations for the MUTAG and BA-2Motif datasets. We also examine the relationship among different metrics via their induced trade-off surface and explore its dependence on the nature of the motif library.
Figures & tables
| VGAE | GNTK | |||
|---|---|---|---|---|
| Configuration | Fidelity | Stability | Fidelity | Stability |
| interp_off | 0.5433 | 44.32 | 0.5450 | 78 117 |
| interp_precomputed | 0.5449 | 45.69 | 0.5374 | 81 305 |
| Dataset | Metric | ( only) | 95% CI | ( ) | Gain from |
|---|---|---|---|---|---|
| MUTAG ( ) | VGAE Fidelity | 0.765 | [0.592, 0.889] | 0.783 | +0.018 |
| VGAE Interpretability | 0.922 | [0.871, 0.962] | 0.930 | +0.007 | |
| VGAE Stability | 0.678 | [0.430, 0.844] | 0.682 | +0.004 | |
| GNTK Fidelity | 0.718 | [0.500, 0.865] | 0.773 | +0.056 | |
| GNTK Interpretability | 0.946 | [0.903, 0.983] | 0.951 | +0.005 | |
| GNTK Stability | 0.919 | [0.848, 0.972] | 0.945 | +0.026 |
Appendix figures & tables8 assets
Supplementary material from the paper’s appendix.
Appendix
| Dataset | Graphs | Split | Seed | Hidden | Optimiser | Epochs | Accuracy (errors) |
|---|---|---|---|---|---|---|---|
| MUTAG | 188 | — | 300 | Adam | 50 | 0.798 (38) | |
| BA-2Motif | 1000 | — | 300 | Adam | 50 | 1.000 (0) | |
| BAMultiShapes | 1000 | 12345 | 64 | Adam | 100 | 0.938 (62) | |
| PROTEINS | 1113 | 12345 | 128 | Adam | 100 | 0.845 (172) |
| Raw | Scaled | Effective | ||||
|---|---|---|---|---|---|---|
| Dataset | ||||||
| MUTAG | 0.425 | 1.354 | 4.25 | 1.354 | 3.54 | |
| BA-2Motif | 0.681 | 1.103 | 6.81 | 1.103 | 5.67 | |
| BAMultiShapes | 0.201 | 2.01 | 1.67 | |||
| PROTEINS | 0.312 | 1.312 | 3.12 | 1.312 | 2.60 | |
| Phase | s/graph | Role |
|---|---|---|
| setup | 0.003 | model forward passes, normalization constants |
| stage1_main | 1.108 | Stage 1 search on the target graph |
| smoothing | 10.642 | sampling perturbations, Stage 1 on the retained |
| stability_norm | 0.001 | stability normalization constants |
| stage2_main | 4.260 | Stage 2 search on the target graph |
| stab_eval | 10.679 | evaluation only : re-samples to score stability |
| Explainer | Metric | ||||
|---|---|---|---|---|---|
| MUTAG | |||||
| GNNExplainer | Fidelity | 0.218 | 0.256 | 0.285 | 0.322 |
| GNNExplainer | Interpretability | 0.460 | 0.489 | 0.498 | 0.524 |
| GNNExplainer | Stability (VGAE) | 0.665 | 1.248 | 1.701 | 2.571 |
| GNNExplainer | Stability (GNTK) | 1 280 | 2 110 | 3 144 | 5 647 |
| PGExplainer | Fidelity | 0.234 | 0.256 | 0.278 | 0.331 |
| Dataset | Method | ||||
|---|---|---|---|---|---|
| MUTAG | Ours ( VGAE ) | ||||
| MUTAG | Ours ( GNTK ) | ||||
| BA-2Motif | Ours ( VGAE ) | ||||
| BA-2Motif | Ours ( GNTK ) | ||||
| BAMultiShapes | Ours ( VGAE ) | ||||
| BAMultiShapes | Ours ( GNTK ) |
| Dataset | Method | ||||
|---|---|---|---|---|---|
| MUTAG | Ours ( VGAE ) | ||||
| MUTAG | Ours ( GNTK ) | ||||
| BA-2Motif | Ours ( VGAE ) | ||||
| BA-2Motif | Ours ( GNTK ) | ||||
| BAMultiShapes | Ours ( VGAE ) | ||||
| BAMultiShapes | Ours ( GNTK ) |
| Dataset | Method | ||||
|---|---|---|---|---|---|
| MUTAG | Ours ( VGAE ) | ||||
| MUTAG | Ours ( GNTK ) | ||||
| BA-2Motif | Ours ( VGAE ) | ||||
| BA-2Motif | Ours ( GNTK ) | ||||
| BAMultiShapes | Ours ( VGAE ) | ||||
| BAMultiShapes | Ours ( GNTK ) |
GNN Explainability
Graph neural network explainability studies how to identify and communicate the features, nodes, edges, and subgraphs behind graph-model predictions.
See all 80 papers