1
Return

Addressing Structural Distribution Shift in Explanations for Graph Neural Networks

delete2026-05-04
delete0
PRE
AI
Z
Zhuomin Chen
H
Hojat Allah Salehi
E
Esteban Schafir
X
Xu Zheng
J
Jiaxing Zhang
韦华 (Hua Wei)
J
Jingchao Ni
F
Farhad Shirani
D
Dongsheng Luo
DOI:10.1109/tpami.2026.3690304delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Graph Neural Networks (GNNs) are essential for processing graph-structured data and have wide applications in critical domains. The increasing use of GNNs in high-stakes scenarios requires robust explainability to ensure trust and transparency in decision-making. A common approach to explaining GNNs is to identify subgraphs, <italic xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">a.k.a.</i> explanations, that significantly influence model predictions. However, this task is challenging due to the distribution shifts from the original training graphs to the explanation subgraphs, a factor that is largely overlooked in the existing research. These shifts arise because GNNs are trained on original graphs, while explanation subgraphs often differ in properties such as the number of nodes or structural patterns. As a result, GNNs may struggle to generalize to explanation subgraphs with a different distribution from its training data. In this paper, we systematically investigate the Out-Of-Distribution (OOD) problem through theoretical analysis and empirical studies. To address this challenge, we first develop a theoretical framework that formalizes the notion of explanation subgraphs through <italic xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">sufficiency</i> and <italic xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">minimality</i> criteria, ensuring both prediction preservation and structural compactness. Our analysis reveals a fundamental distributional disparity between explanation subgraphs and original graphs, leading to a novel concept of <italic xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">proxy graphs</i> proposed in this work. Proxy graphs maintain the essential explanatory information while conforming to the original data distribution through a combination of parametric and non-parametric optimization approaches. Empirical evaluations on diverse datasets show that our method improves the quality and reliability of GNN explanations, advancing the field of GNN explainability.
Keywords:
Explainable artificial intelligence
graph neural networks
deep learning

Journal

IEEE Transactions on Pattern Analysis and Machine Intelligence cover
IEEE Transactions on Pattern Analysis and Machine Intelligence
IF:
18.6
Papers:
831
Citations:
9.8W

Organization

U
University of Houston
Scholars:
923
Papers: 537
Citations: 1.7W
N
New Jersey Institute of Technology
Scholars:
4.1K
Papers: 4.5K
Citations: 4.6K
A
arizona state university
Scholars:
3.0K
Papers: 1.6K
Citations: 0
F
Florida International University
Scholars:
7.2K
Papers: 5.8K
Citations: 1.1W
Cited Papers

Cited Papers

Citing Papers

Citing Papers