Command Palette
Search for a command to run...
Xiaocheng Yang; Mingyu Yan; Shirui Pan; Xiaochun Ye; Dongrui Fan

Abstract
Heterogeneous graph neural networks (HGNNs) have 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) over 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. This paper conducts an in-depth and detailed study of these mechanisms and proposes 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 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.
Code Repositories
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| heterogeneous-node-classification-on-acm | SeHGNN | Claim Classification Macro-F1: 93.95 Micro-F1: 93.87 |
| heterogeneous-node-classification-on-dblp-2 | SeHGNN | Macro-F1: 94.86 Micro-F1: 95.24 |
| heterogeneous-node-classification-on-freebase | SeHGNN | Macro-F1: 50.71 Micro-F1: 63.41 |
| heterogeneous-node-classification-on-imdb | SeHGNN | Macro-F1: 66.63 Micro-F1: 68.21 |
| heterogeneous-node-classification-on-oag | SeHGNN | MRR: 29.11 NDCG: 46.75 |
| heterogeneous-node-classification-on-oag-l1 | SeHGNN | MRR: 84.95 NDCG: 86.01 |
Build AI with AI
From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.