FSX: Message Flow Sensitivity Enhanced Structural Explainer for Graph Neural Networks

TL;DR

FSX is a hybrid explainability framework for GNNs that combines message flow analysis with cooperative game theory to produce accurate, interpretable explanations of model predictions by identifying influential subgraphs.

Contribution

It introduces a novel flow-sensitivity analysis and a flow-aware cooperative game approach to improve GNN explanations with better fidelity and efficiency.

Findings

FSX outperforms existing methods in explanation fidelity.

It significantly reduces explanation runtime.

Provides detailed insights into GNN structural reasoning.

Abstract

Despite the widespread success of Graph Neural Networks (GNNs), understanding the reasons behind their specific predictions remains challenging. Existing explainability methods face a trade-off that gradient-based approaches are computationally efficient but often ignore structural interactions, while game-theoretic techniques capture interactions at the cost of high computational overhead and potential deviation from the model's true reasoning path. To address this gap, we propose FSX (Message Flow Sensitivity Enhanced Structural Explainer), a novel hybrid framework that synergistically combines the internal message flows of the model with a cooperative game approach applied to the external graph data. FSX first identifies critical message flows via a novel flow-sensitivity analysis: during a single forward pass, it simulates localized node perturbations and measures the resulting changes in message flow intensities. These sensitivity-ranked flows are then projected onto the input graph to define compact, semantically meaningful subgraphs. Within each subgraph, a flow-aware cooperative game is conducted, where node contributions are evaluated fairly through a Shapley-like value that incorporates both node-feature importance and their roles in sustaining or destabilizing the identified critical flows. Extensive evaluation across multiple datasets and GNN architectures demonstrates that FSX achieves superior explanation fidelity with significantly reduced runtime, while providing unprecedented insights into the structural logic underlying model predictions--specifically, how important sub-structures exert influence by governing the stability of key internal computational pathways.