Simple and Efficient Heterogeneous Graph Neural Network

No Thumbnail Available
File version
Author(s)
Yang, X
Yan, M
Pan, S
Ye, X
Fan, D
Griffith University Author(s)
Primary Supervisor
Other Supervisors
Editor(s)
Date
2023
Size
File type(s)
Location

Washington, United States

License
Abstract

Heterogeneous graph neural networks (HGNNs) have the powerful capability to embed rich structural and semantic information of a heterogeneous graph into node representations. Existing HGNNs inherit many mechanisms from graph neural networks (GNNs) designed for homogeneous graphs, especially the attention mechanism and the multi-layer structure. These mechanisms bring excessive complexity, but seldom work studies whether they are really effective on heterogeneous graphs. In this paper, we conduct an in-depth and detailed study of these mechanisms and propose the Simple and Efficient Heterogeneous Graph Neural Network (SeHGNN). To easily capture structural information, SeHGNN pre-computes the neighbor aggregation using a light-weight mean aggregator, which reduces complexity by removing overused neighbor attention and avoiding repeated neighbor aggregation in every training epoch. To better utilize semantic information, SeHGNN adopts the single-layer structure with long metapaths to extend the receptive field, as well as a transformer-based semantic fusion module to fuse features from different metapaths. As a result, SeHGNN exhibits the characteristics of a simple network structure, high prediction accuracy, and fast training speed. Extensive experiments on five real-world heterogeneous graphs demonstrate the superiority of SeHGNN over the state-of-the-arts on both accuracy and training speed.

Journal Title
Conference Title

Proceedings of the AAAI Conference on Artificial Intelligence

Book Title
Edition
Volume

37

Issue

9

Thesis Type
Degree Program
School
Publisher link
Patent number
Funder(s)
Grant identifier(s)
Rights Statement
Rights Statement
Item Access Status
Note
Access the data
Related item(s)
Subject

Neural networks

Artificial intelligence

Persistent link to this record
Citation

Yang, X; Yan, M; Pan, S; Ye, X; Fan, D, Simple and Efficient Heterogeneous Graph Neural Network, Proceedings of the AAAI Conference on Artificial Intelligence, 2023, 37 (9), pp. 10816-10824