HINormer: Representation Learning On Heterogeneous Information Networks with Graph Transformer
Abstract
Recent studies have highlighted the limitations of message-passing based graph neural networks (GNNs),, limited model expressiveness, over-smoothing, over-squashing, etc.\ To alleviate these issues, Graph Transformers (GTs) have been proposed which work in the paradigm that allows message passing to a larger coverage even across the whole graph. Hinging on the global range attention mechanism, GTs have shown a superpower for representation learning on homogeneous graphs. However, the investigation of GTs on heterogeneous information networks (HINs) is still under-exploited. In particular, on account of the existence of heterogeneity, HINs show distinct data characteristics and thus require different treatment. To bridge this gap, in this paper we investigate the representation learning on HINs with Graph Transformer, and propose a novel model named, which capitalizes on a larger-range aggregation mechanism for node representation learning. In particular, assisted by two major modules,, a local structure encoder and a heterogeneous relation encoder, \ can capture both the structural and heterogeneous information of nodes on HINs for comprehensive node representations. We conduct extensive experiments on four HIN benchmark datasets, which demonstrate that our proposed model can outperform the state-of-the-art.
Benchmark Results
Tested on 2 datasets 1 baseline value reported too low
Tables show presented results in global context (best 3 papers + GCN + MLP).
Heterogeneous Graph Benchmarks
| # | Model | Micro-F1 ↑ | Paper |
|---|---|---|---|
| 1 | LMSPS | 0.9566
±0.0020 | |
| 2 | PSHGCN | 0.9561
±0.0012 | |
| 3 | RpHGNN | 0.9555
±0.0029 | |
| 13 | HINormer | 0.9494
±0.0021 | This paper |
| 21 | GCN | 0.9460
±0.0031 | |
| 33 | Transformer | 0.9399
±0.0011 | This paper |
| 176 | MLP | 0.6447
±0.0136 |
| # | Model | Micro-F1 ↑ | Paper |
|---|---|---|---|
| 1 | ELLA | 0.7930
±0.0010 | |
| 2 | MHGCN | 0.7820
±0.0138 | |
| 3 | PSHGCN | 0.7446
±0.0032 | |
| 14 | GCN | 0.6813
±0.0083 | |
| 16 | HINormer | 0.6783
±0.0034 | This paper |
| 26 | RGCN | 0.6651
±0.0028 best reported | best at pub: Heterogeneous Graph Representation Learning with Relation Awareness (2021) best today: Spectral Heterogeneous Graph Convolutions via Positive Noncommutative Polynomials (WWW 2023) |
| 27 | Transformer | 0.6629
±0.0069 | This paper |
| 74 | MLP | 0.5514 |
