arrow
Return

Enhancing graph learning interpretability through modulating cluster information flow

delete2026-01-29
delete0
PRE
AI
J
Jiayi Yang
W
Wei Ye
X
Xin Sun
R
Rui Fan
韩
韩军功 (Jungong Han)
DOI:10.1016/j.patcog.2026.113178delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Interpretable graph learning is essential for scientific applications that depend on learning models to extract reliable insights from graph-structured data. Recent efforts to explain GNN predictions focus on identifying vital substructures, such as subgraphs. However, existing approaches tend to misclassify the neighboring irrelevant nodes as part of the vital subgraphs. To address this, we propose Cluster Information Flow Graph Neural Networks (CIFlow-GNN), a built-in model-level method that provides accurate interpretable subgraph explanations by modulating the cluster information flow. CIFlow-GNN incorporates two modules, i.e., the graph clustering module and the cluster prototype module. The graph clustering module partitions the nodes according to their connectivity in the graph topology and their similarity in cluster features. Specifically, we introduce a cluster feature loss to regulate information flow at the cluster level. We prove that the proposed cluster feature loss is a lower bound of the InfoNCE loss. Optimizing the cluster feature loss reduces the mutual information among clusters and achieves the modulation of cluster information flow. Subsequently, the graph prototype module uses prototypes as a bridge to select important clusters as vital subgraphs by integrating information across all graphs. To ensure accurate correspondence between clusters and prototypes, we further modulate the cluster information flow at the prototype level. Experimental studies on both synthetic and real-world datasets demonstrate that our proposed CIFlow-GNN can identify vital subgraphs effectively and efficiently.
Keywords:
graph neural networks
interpretable machine learning
cluster information flow
subgraph explanation
graph clustering

Journal

Pattern Recognition cover
Pattern Recognition
IF:
7.6
Papers:
1.3W
Citations:
4.5W

Organization

T
Tsinghua University
Scholars:
8.6K
Papers: 4.1K
Citations: 17.7W
T
tongji university
Scholars:
7.9W
Papers: 6.0W
Citations: 98
T
Tongji University
Scholars:
5.2K
Papers: 2.1K
Citations: 388
C
city university of macau
Scholars:
1.3K
Papers: 1.4K
Citations: 1
researcher View more organizations
Cited Papers

Cited Papers

A unified framework for convolution-based graph neural networks
err2024-11-01
err0
PREAI
errPan, Xuran; Han, Xiaoyan; Wang, Chaofei; Li, Zhuo; Song, Shiji; Huang, Gao; Wu, Cheng
errShare
errSave
Molecular substructure graph attention network for molecular property identification in drug discovery
err2022-08-01
err22
PREAI
errYe, Xian-bin; Guan, Quanlong; Luo, Weiqi; Fang, Liangda; Lai, Zhao-Rong; Wang, Jun
errShare
errSave
errShare
errSave
Self-attention empowered graph convolutional network for structure learning and node embedding
err2024-09-01
err2
errOAAI
errJiang, Mengying; Liu, Guizhong; Su, Yuanchao; Wu, Xinliang
errShare
errSave
Structured self-attention architecture for graph-level representation learning
err2020-04-01
err29
PREAI
errFan, Xiaolong; Gong, Maoguo; Xie, Yu; Jiang, Fenlong; Li, Hao
errShare
errSave
Mastering the game of Go without human knowledge
err2017-10-19
err0
PREAI
errDavid Silver; Julian Schrittwieser; Karen Simonyan; Ioannis Antonoglou; Aja Huang; Arthur Guez; Thomas Hubert; Lucas Baker; Matthew Lai; Adrian Bolton; Yutian Chen; Timothy Lillicrap; Fan Hui; Laurent Sifre; George van den Driessche; Thore Graepel; Demis Hassabis
errShare
errSave
researcher View more