arrow
返回

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
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

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.
Keyword:
graph neural networks
interpretable machine learning
cluster information flow
subgraph explanation
graph clustering

期刊

Pattern Recognition 封面图
Pattern Recognition
IF:
7.6
论文数:
1.3W
被引数:
4.5W

机构

T
Tsinghua University
学者数:
8.6K
论文数: 4.1K
被引数: 17.7W
T
tongji university
学者数:
7.9W
论文数: 6.0W
被引数: 98
T
Tongji University
学者数:
5.2K
论文数: 2.1K
被引数: 388
C
city university of macau
学者数:
1.3K
论文数: 1.4K
被引数: 1
学者 查看更多机构
引用论文

引用论文

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
err分享
err收藏
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
err分享
err收藏
err分享
err收藏
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
err分享
err收藏
Structured self-attention architecture for graph-level representation learning
err2020-04-01
err29
PREAI
errFan, Xiaolong; Gong, Maoguo; Xie, Yu; Jiang, Fenlong; Li, Hao
err分享
err收藏
Mastering the game of Go without human knowledge在没有人类知识的情况下掌握Go游戏
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
err分享
err收藏
学者 查看更多内容