NICE: Scale-Stable Perturbations for Graph Neural Network Explanations via Noise Corruption
Organizations: Central South University · Hong Kong Polytechnic University · Griffith University
Abstract
Post-hoc Graph Neural Network (GNN) explainers commonly follow a Perturb-Query paradigm, inferring the importance of graph elements based on queried predictions to perturbed inputs. However, such perturbations often introduce substantial distribution shift, undermining the reliability of the queried predictions used to derive explanations. While existing efforts mainly improve perturbed graphs or stabilize model predictions on them, we revisit the perturbation mechanism itself. We show that the widely used Element-wise Masking(EM) suppresses edge-induced messages toward zero, causing deterministic scale contraction that accumulates across message-passing layers, a phenomenon we term Scale Drift. Consequently, prediction changes under EM may conflate information corruption with deviations in propagation scale. As a scale-stable alternative to EM, we introduce Noise Corruption (NC), which perturbs each message through matched-norm random-direction corruption while preserving the expected squared message norm. Building on NC, we propose NICE, a Noise Corruption-based explanation framework, which learns a Stochastic Restoration Boundary (SRB) under NC-induced uncertainty, balancing target-prediction restoration against compactness. Furthermore, Boundary-Integrated Gradient (BIG) converts this boundary into edge attributions by accumulating each edge's contribution to reducing restoration risk along the restoration path. Experiments across multiple benchmarks demonstrate stronger explanation performance and model faithfulness while confirming that NC substantially reduces the Scale Drift induced by masking.
Figures & tables
| Metric | Method | Mutag | Benzene | Alkane | Fluoride | Indole | PAINS | R-Count | R-Max | Rank |
| F1 | Random | 20.06 0.55 | 30.80 0.40 | 8.90 1.09 | 24.52 0.43 | 29.49 0.10 | 29.00 0.23 | 41.23 0.08 | 34.95 0.47 | 10 |
| Saliency | 49.00 0.07 | 47.29 0.01 | 0.00 0.00 | 48.09 0.00 | 50.03 0.00 | 49.97 0.01 | 37.97 0.01 | 46.97 0.00 | 6 | |
| GuidedBP | 37.50 0.00 | 45.31 0.01 | 15.17 0.00 | 41.68 0.00 | 42.67 0.00 | 51.66 0.01 | 38.77 0.01 | 47.03 0.00 | 5 | |
| GNNExplainer | 31.79 1.90 | 25.88 1.00 | 32.16 2.21 | 30.95 0.49 | 42.22 0.50 | 38.97 0.04 | 51.50 0.09 | 45.80 1.01 | 8 | |
| PGExplainer | 51.67 0.24 | 64.96 0.18 | 26.39 0.00 | 61.79 0.40 | 46.65 0.03 | 45.98 0.00 | 52.65 0.00 | 53.17 0.00 | 3 | |
| Refine | 43.60 0.82 | 42.88 0.31 | 11.48 0.00 | 30.36 2.42 | 37.39 0.06 | 54.18 0.07 | 53.73 0.00 | 51.67 0.00 | 4 |
| Dataset | Method | GT | Rand | GT | Rand |
| R-Count | EM | 0.31 0.09 | 0.63 0.22 | 0.57 0.15 | 0.59 0.12 |
| NC | 0.07 0.02 | 0.08 0.02 | 0.02 0.01 | 0.04 0.04 | |
| Imp. | 77.42% | 87.30% | 96.49% | 93.22% | |
| R-Max | EM | 0.45 0.10 | 0.61 0.11 | 0.36 0.16 | 0.48 0.18 |
| NC | 0.16 0.03 | 0.16 0.04 | 0.14 0.05 | 0.13 0.06 | |
| Method | Mean | Std. | Rest. | AUC |
| Full SRB | 0.23 0.06 | 0.06 0.02 | 0.27 0.11 | 0.87 0.00 |
| w/o Var. | 0.37 0.08 | 0.19 0.07 | 0.41 0.13 | 0.83 0.06 |
| w/o Comp. | 0.25 0.11 | 0.09 0.04 | 0.84 0.42 | 0.81 0.02 |
Appendix figures & tables9 assets
Supplementary material from the paper’s appendix.
Appendix
| Symbol | Description |
| Graph and target GNN | |
| Input graph with node set and edge set . | |
| Set of nodes in graph . | |
| Set of edges in graph . | |
| The -th node in graph . | |
| Edge between nodes and . | |
| Symbol | Description |
| Noise Corruption | |
| Matched-norm random corruption of . | |
| Distribution of the corrupted message . | |
| Collection of corruption variables sampled. | |
| Joint distribution of corruption variables. | |
| GNN prediction under restoration configuration and NC realization . | |
| Statistic | Graphs | Average Nodes | Average Edges | Node Features | Classes | Train/Test |
| Mutag | 1,768 | ~29.15 | ~60.83 | 14 | 2 | 1597/177 |
| Benzene | 12,000 | ~20.58 | ~43.64 | 14 | 2 | 10800/1200 |
| Alkane | 1,125 | ~21.39 | ~45.38 | 14 | 2 | 1012/113 |
| Fluoride | 8,671 | ~21.36 | ~45.37 | 14 | 2 | 7803/868 |
| B-XAIC | 5,000 | ~34.60 | ~75.19 | 11 | 2 | 45000/5000 |
| Dataset | Target Acc. (%) | Positive Ratio (%) | Ground-Truth Explanation | GT Edge Ratio (%) Mean Std. |
| Mutagencity | 100.00 | 36.20 | and groups | |
| Benzene | 93.17 | 50.00 | Benzene ring | |
| Alkane | 100.00 | 33.33 | Alkane chain and {}{}\mathrm{C}\hbox to0.0pt{\raisebox{0.86108pt}{{-}}\hss}\raisebox{-0.86108pt}{{-}}\mathrm{O} group | |
| Fluoride | 96.31 | 17.61 | Fluorine atom and {}{}\mathrm{C}\hbox to0.0pt{\raisebox{0.86108pt}{{-}}\hss}\raisebox{-0.86108pt}{{-}}\mathrm{O} group | |
| Indole | 98.68 | 36.70 | Fused benzene–pyrrole structure | |
| PAINS | 91.84 | 32.94 | Detected PAINS alert substructure |
| Metric | Method | Mutag | Benzene | Alkane | Fluoride | Indole | PAINS | R-Count | R-Max |
| Prec | Random | 15.58 0.68 | 29.90 0.57 | 5.17 0.63 | 19.23 0.34 | 31.46 0.08 | 31.74 0.19 | 68.52 0.15 | 30.96 1.03 |
| Saliency | 38.70 0.24 | 44.98 0.02 | 0.00 0.00 | 37.79 0.00 | 51.52 0.00 | 52.39 0.02 | 63.18 0.01 | 41.14 0.00 | |
| GuidedBP | 30.74 0.00 | 43.35 0.02 | 8.86 0.00 | 33.73 0.00 | 44.85 0.00 | 54.74 0.01 | 64.43 0.01 | 41.67 0.00 | |
| GNNExplainer | 24.47 1.51 | 19.68 0.44 | 24.84 1.87 | 23.20 0.68 | 43.41 0.55 | 42.31 0.10 | 85.03 1.08 | 42.59 1.33 | |
| PGExplainer | 39.74 0.22 | 66.20 0.13 | 15.53 0.00 | 49.96 0.33 | 48.76 0.03 | 49.00 0.00 | 88.02 0.00 | 51.15 0.00 | |
| Refine | 34.43 0.75 | 42.77 0.47 | 6.17 0.00 | 19.84 2.11 | 40.89 0.08 | 58.01 0.05 | 86.40 0.08 | 48.65 0.00 |
| Dataset | Method | GT | Rand | GT | Rand |
| Mutag | EM | 0.77 0.12 | 0.46 0.19 | 0.53 0.17 | 0.29 0.02 |
| NC | 0.45 0.14 | 0.38 0.12 | 0.52 0.11 | 0.44 0.11 | |
| Imp. | 41.56% | 17.39% | 1.89% | -51.72% | |
| Benzene | EM | 0.03 0.07 | 0.11 0.09 | 0.77 0.26 | 0.38 0.18 |
| NC | 0.03 0.03 | 0.06 0.04 | 0.74 0.15 | 0.43 0.26 | |
GNN Explainability
Graph neural network explainability studies how to identify and communicate the features, nodes, edges, and subgraphs behind graph-model predictions.
See all 82 papers