arrow
返回

Optimizing Graph Neural Network With Multiaspect Hilbert-Schmidt Independence Criterion

delete2023-12-01
delete7
PRE
AI
Y
Yunfei He
D
Dengcheng Yan
W
Wenxin Xie
张
张议文 (Yiwen Zhang) *
Q
Qiang He
Y
Yun Yang
DOI:10.1109/TNNLS.2022.3171419delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
The graph neural network (GNN) has demonstrated its superior power in various data mining tasks and has been widely applied in diversified fields. The core of GNN is the aggregation and combination functions, and mainstream GNN studies focus on the enhancement of these functions. However, GNNs face a common challenge, i.e., useless features contained in neighbor nodes may be integrated into the target node during the aggregation process. This leads to poor node embedding and undermines downstream tasks. To tackle this problem, this article proposes a novel GNN optimization framework GNN-MHSIC by introducing the nonparametric dependence method Hilbert-Schmidt independence criterion (HSIC) under the guidance of information bottleneck. HSIC is utilized to guide the information propagation among layers of a GNN from multiaspect views. GNN-MHSIC aims to achieve three main objectives: 1) minimizing the HSIC between the input features and the propagation layers; 2) maximizing the HSIC between the propagation layers and the ground truth; and 3) minimizing the HSIC between the propagation layers. With a multiaspect design, GNN-MHSIC can minimize the propagation of redundant information while preserving relevant information about the target node. We prove GNN-MHSIC's finite upper and lower bounds theoretically and evaluate it experimentally with four classic GNN models, including the graph convolutional network, the graph attention network (GAT), the heterogeneous GAT, and the heterogeneous graph (HG) propagation network on three widely used HGs. The results illustrate the usefulness and performance of GNN-MHSIC.
Keyword:
Optimization
Task analysis
Kernel
Graph neural networks
Random variables
Learning systems
Mutual information
Graph neural network (GNN)
heterogeneous information network (HIN)
Hilbert-Schmidt independence criterion (HSIC)
information bottleneck (IB)

期刊

IEEE Transactions on Neural Networks and Learning Systems 封面图
IEEE Transactions on Neural Networks and Learning Systems
IF:
8.9
论文数:
7.6K
被引数:
7.2W

机构

S
Swinburne University of Technology
学者数:
9.3K
论文数: 1.2W
被引数: 2.0W
A
anhui university
学者数:
1.9W
论文数: 1.2W
被引数: 24
引用论文

引用论文

Metagraph-Based Learning on Heterogeneous Graphs
err2021-01-01
err22
errOAAI
errFang, Yuan; Lin, Wenqing; Zheng, Vincent W.; Wu, Min; Shi, Jiaqi; Chang, Kevin Chen-Chuan; Li, Xiao-Li
err分享
err收藏
err分享
err收藏
err分享
err收藏
The BOEL Test as a Screening Device for Otitis Media in Infants
err1978-05-01
err0
PREAI
errCathee J. Huber; Sharon R. Stangler; Donald K. Routh
err分享
err收藏
The heritability of insomnia: A meta‐analysis of twin studies
err2020-12-03
err0
errOAAI
errNicola L. Barclay; Desi Kocevska; Wichor M. Bramer; Eus J. W. Van Someren; Philip Gehrman
err分享
err收藏
Health-related quality of life (HRQoL) from HIV patients’ perspective: comparison of patient-reported outcome (PRO) measures among people living with hiv (PLWH) and other chronic clinical conditions
err2022-03-26
err0
errOAAI
errC. Seguiti; P. F. Salvo; E. Di Stasio; S. Lamonica; A. L. Fedele; S. Manfrida; N. Ciccarelli; B. Corvari; C. De Luca; L. Tartaglione; D. Pitocco; R. Cauda; A. Cingolani
err分享
err收藏
学者 查看更多内容