HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Simple and Efficient Heterogeneous Graph Neural Network

Xiaocheng Yang; Mingyu Yan; Shirui Pan; Xiaochun Ye; Dongrui Fan

Simple and Efficient Heterogeneous Graph Neural Network

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

ICT-GIMLab/SeHGNN/tree/master/large
pytorch
Mentioned in GitHub
ict-gimlab/sehgnn
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
heterogeneous-node-classification-on-acmSeHGNN
Claim Classification Macro-F1: 93.95
Micro-F1: 93.87
heterogeneous-node-classification-on-dblp-2SeHGNN
Macro-F1: 94.86
Micro-F1: 95.24
heterogeneous-node-classification-on-freebaseSeHGNN
Macro-F1: 50.71
Micro-F1: 63.41
heterogeneous-node-classification-on-imdbSeHGNN
Macro-F1: 66.63
Micro-F1: 68.21
heterogeneous-node-classification-on-oagSeHGNN
MRR: 29.11
NDCG: 46.75
heterogeneous-node-classification-on-oag-l1SeHGNN
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.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Simple and Efficient Heterogeneous Graph Neural Network | Papers | HyperAI