Circle Feature Graphormer: Can Circle Features Stimulate Graph Transformer?

Jingsong Lv, Hongyang Chen, Yao Qi, Lei Yu

Main figure for Circle Feature Graphormer: Can Circle Features Stimulate Graph Transformer?

Abstract

In this paper, we introduce two local graph features for missing link prediction tasks on ogbl-citation2. We define the features as Circle Features, which are borrowed from the concept of circle of friends. We propose the detailed computing formulas for the above features. Firstly, we define the first circle feature as modified swing for common graph, which comes from bipartite graph. Secondly, we define the second circle feature as bridge, which indicates the importance of two nodes for different circle of friends. In addition, we firstly propose the above features as bias to enhance graph transformer neural network, such that graph self-attention mechanism can be improved. We implement a Circled Feature aware Graph transformer (CFG) model based on SIEG network, which utilizes a double tower structure to capture both global and local structure features. Experimental results show that CFG achieves the state-of-the-art performance on dataset ogbl-citation2.

Proposed method: CFG

Benchmark Results

Tested on 1 dataset 1 baseline value reported too low

Tables show presented results in global context (best 3 papers + GCN + MLP).

OGB

#ModelMRR ↑Paper
1
CFG
0.8997 ±0.0015
This paper
2
SIEG
0.8987 ±0.0018
This paper
3
LPFormer
0.8981 ±0.0013
4
SUREL
actual #4 vs paper #5 1 place worse
0.8974 ±0.0018
best reported
actual #4 vs paper #5 1 place worse
paper 0.8883 ±0.0018
5
NGNN + SEAL
0.8891 ±0.0022
This paper
7
S3GRL (PoS Plus)
0.8814 ±0.0008
This paper
8
SEAL
0.8767 ±0.0032
This paper
12
GCN
0.8705 ±0.0004
17
AGDN w/GraphSAINT
0.8549 ±0.0029
This paper
20
PLNLP
0.8492 ±0.0029
This paper
71
MLP
0.3917 ±0.0044
SOTA performance (not significantly different from the best) Proposed method (this paper) Baseline / comparison (this paper) Global context (other papers) Best at pub — reported result worse than the best known when published (best reported in green)