Topology-Informed Graph Transformer

Yuncheol Choi, Sun Woo Park, Minho Lee, Youngho Woo

Main figure for Topology-Informed Graph Transformer

Abstract

Transformers have revolutionized performance in Natural Language Processing and Vision, paving the way for their integration with Graph Neural Networks (GNNs). One key challenge in enhancing graph transformers is strengthening the discriminative power of distinguishing isomorphisms of graphs, which plays a crucial role in boosting their predictive performances. To address this challenge, we introduce'Topology-Informed Graph Transformer (TIGT)', a novel transformer enhancing both discriminative power in detecting graph isomorphisms and the overall performance of Graph Transformers. TIGT consists of four components: A topological positional embedding layer using non-isomorphic universal covers based on cyclic subgraphs of graphs to ensure unique graph representation: A dual-path message-passing layer to explicitly encode topological characteristics throughout the encoder layers: A global attention mechanism: And a graph information layer to recalibrate channel-wise graph features for better feature representation. TIGT outperforms previous Graph Transformers in classifying synthetic dataset aimed at distinguishing isomorphism classes of graphs. Additionally, mathematical analysis and empirical evaluations highlight our model's competitive edge over state-of-the-art Graph Transformers across various benchmark datasets.

Proposed method: TIGT

Benchmark Results

Tested on 6 datasets 20 baseline values reported too low

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

Classic

#ModelMAE ↓Paper
1
TIGT
0.0140 ±0.0010
This paper
2
FloydNet
0.0160
3
GSSC
0.0190 ±0.0010
7
CIN
actual #7 vs paper #66 59 places worse
0.0210 ±0.0010
best reported
actual #7 vs paper #66 59 places worse
paper 0.0790 ±0.0060
paper 0.0790 ±0.0060 Δ at pub 0.0580
11
GRIT
0.0230 ±0.0010
This paper
22
Graphormer
actual #22 vs paper #56 34 places worse
0.0250
best reported
actual #22 vs paper #56 34 places worse
paper 0.0520 ±0.0050
23
Graphormer-GD
0.0250 ±0.0040
This paper
29
Graphormer-URPE
0.0280 ±0.0020
This paper
49
Cy2C-GNNs
0.0420 ±0.0010
This paper
63
PNA
actual #63 vs paper #89 26 places worse
0.1880 ±0.0040
actual #63 vs paper #89 26 places worse
paper 0.1880 ±0.0040
paper 0.1880 ±0.0040 Δ at pub 0.1320 Δ today 0.1210
65
GPS
0.0700 ±0.0040
This paper
67
GIN-AK+
0.0800 ±0.0010
This paper
68
CRaW1
0.0850 ±0.0040
This paper
69
GIN
0.0880 ±0.0020
This paper
71
GatedGCN+LSPE
0.0900 ±0.0010
This paper
72
K-Subgraph SAT
0.0940 ±0.0080
This paper
74
GSN
0.1010 ±0.0100
This paper
77
EGT
0.1080 ±0.0090
This paper
78
GAT
0.1110 ±0.0020
This paper
79
GCN
0.1130 ±0.0020
This paper
80
GraphSAGE
0.1260 ±0.0030
84
SAN
0.1390 ±0.0060
This paper
87
DGN
0.1680 ±0.0030
This paper
98
GatedGCN
actual #98 vs paper #98
0.2820 ±0.0150
actual #98 vs paper #98
paper 0.2820 ±0.0150
paper 0.2820 ±0.0150 Δ at pub 0.1440

LRGB

#ModelMAE ↓Paper
1
GCN
actual #1 vs paper #183 182 places worse
0.2421 ±0.0016
best reported
actual #1 vs paper #183 182 places worse
paper 0.3496 ±0.0013
paper 0.3496 ±0.0013 Δ at pub 0.1036 Δ today 0.1075
2
IPR-MPNN
0.2422 ±0.0007
3
TANGO-GPS
0.2422 ±0.0014
5
GIN
0.2429 ±0.0019
7
GatedGCN
actual #7 vs paper #182 175 places worse
0.2431 ±0.0020
best reported
actual #7 vs paper #182 175 places worse
paper 0.3420 ±0.0013
paper 0.3420 ±0.0013 Δ at pub 0.0943 Δ today 0.0989
32
GRIT
0.2460 ±0.0012
This paper
49
GINE
actual #49 vs paper #184 135 places worse
0.2473 ±0.0017
best reported
actual #49 vs paper #184 135 places worse
paper 0.3547 ±0.0045
paper 0.3547 ±0.0045 Δ at pub 0.1074 Δ today 0.1074
70
TIGT
0.2485 ±0.0015
This paper
69
GatedGCN+RWSE
actual #69 vs paper #179 110 places worse
0.2485 ±0.0022
best reported
actual #69 vs paper #179 110 places worse
paper 0.3357 ±0.0006
paper 0.3357 ±0.0006 Δ at pub 0.0828 Δ today 0.0872
85
GPS
0.2500 ±0.0012
This paper
109
Cy2C-GNNs
0.2521 ±0.0012
This paper
113
Transformer+LapPE
0.2529 ±0.0160
This paper
123
SAN+RWSE
0.2545 ±0.0012
This paper
155
SAN+LapPE
actual #155 vs paper #155
0.2680 ±0.0040
best reported
actual #155 vs paper #155
paper 0.2683 ±0.0043
best at pub: LRGB (NeurIPS 2022)

Difference within reported uncertainty; not counted as underperformance.

paper 0.2683 ±0.0043 Δ today 0.0003
201
MLP
0.4351 ±0.0008
#ModelAP ↑Paper
1
S^2GCN
0.7311 ±0.0066
2
S4G
0.7293 ±0.0004
3
BuNN
0.7276 ±0.0065
4
GCN
actual #4 vs paper #142 138 places worse
0.7261 ±0.0011
best reported
actual #4 vs paper #142 138 places worse
paper 0.5930 ±0.0023
paper 0.5930 ±0.0023 Δ at pub 0.0930 Δ today 0.1331
26
GIN
0.7059 ±0.0089
34
GatedGCN
actual #34 vs paper #149 115 places worse
0.7006 ±0.0033
best reported
actual #34 vs paper #149 115 places worse
paper 0.5864 ±0.0035
paper 0.5864 ±0.0035 Δ at pub 0.0901 Δ today 0.1142
36
GRIT
0.6988 ±0.0082
This paper
73
GatedGCN+RWSE
actual #73 vs paper #134 61 places worse
0.6685 ±0.0062
best reported
actual #73 vs paper #134 61 places worse
paper 0.6069 ±0.0035
paper 0.6069 ±0.0035 Δ at pub 0.0616
74
TIGT
0.6679 ±0.0074
This paper
78
GINE
actual #78 vs paper #162 84 places worse
0.6621 ±0.0067
best reported
actual #78 vs paper #162 84 places worse
paper 0.5498 ±0.0079
paper 0.5498 ±0.0079 Δ at pub 0.1123 Δ today 0.1123
89
SAN+RWSE
actual #89 vs paper #111 22 places worse
0.6562 ±0.0075
best reported
actual #89 vs paper #111 22 places worse
paper 0.6439 ±0.0075

Difference within reported uncertainty; not counted as underperformance.

paper 0.6439 ±0.0075 Δ at pub 0.0123
87
GPS
actual #87 vs paper #91 4 places worse
0.6562 ±0.0115
best reported
actual #87 vs paper #91 4 places worse
paper 0.6535 ±0.0041

Difference within reported uncertainty; not counted as underperformance.

paper 0.6535 ±0.0041 Δ at pub 0.0027
116
SAN+LapPE
0.6384 ±0.0121
This paper
118
Transformer+LapPE
0.6326 ±0.0126
This paper
170
Cy2C-GNNs
0.5193 ±0.0025
This paper
181
MLP
0.4060 ±0.0021

GNNBenchmark

#ModelAccuracy ↑Paper
1
Schrödinger
0.9913 ±0.0004
2
CNN
0.9907 ±0.0007
3
MPNN
0.9895 ±0.0006
11
GatedGCN
actual #11 vs paper #90 79 places worse
0.9855 ±0.0002
best reported
actual #11 vs paper #90 79 places worse
paper 0.9734 ±0.0014
32
GIN
actual #32 vs paper #110 78 places worse
0.9833 ±0.0011
best reported
actual #32 vs paper #110 78 places worse
paper 0.9648 ±0.0025
best at pub: Graph-Coupled Oscillator Networks (ICML 2022)
paper 0.9648 ±0.0025 Δ at pub 0.0075 Δ today 0.0184
39
TIGT
0.9823 ±0.0013
This paper
47
EGT
0.9817 ±0.0009
This paper
56
GRIT
0.9811 ±0.0011
This paper
59
GPS
actual #59 vs paper #61 2 places worse
0.9808 ±0.0011
best reported
actual #59 vs paper #61 2 places worse
paper 0.9805 ±0.0013

Difference within reported uncertainty; not counted as underperformance.

paper 0.9805 ±0.0013 Δ today 0.0003
74
CRaW1
0.9794 ±0.0005
This paper
75
PNA
0.9794 ±0.0012
This paper
81
Cy2C-GNNs
0.9777 ±0.0000
This paper
86
GAT
actual #86 vs paper #121 35 places worse
0.9755 ±0.0003
best reported
actual #86 vs paper #121 35 places worse
paper 0.9554 ±0.0021
95
GraphSAGE
0.9731 ±0.0010
126
MLP
0.9534 ±0.0014
135
GCN
actual #135 vs paper #138 3 places worse
0.9405 ±0.0005
best reported
actual #135 vs paper #138 3 places worse
paper 0.9071 ±0.0022
paper 0.9071 ±0.0022 Δ at pub 0.0198 Δ today 0.0335
#ModelAccuracy ↑Paper
1
RandAlign
0.7942 ±0.0007
2
differential encoding
0.7907 ±0.0027
3
PPGT
0.7856 ±0.0070
7
GRIT
0.7647 ±0.0088
This paper
31
TIGT
actual #31 vs paper #32 1 place worse
0.7396 ±0.0036
best reported
actual #31 vs paper #32 1 place worse
paper 0.7396 ±0.0036

Difference within reported uncertainty; not counted as underperformance.

paper 0.7396 ±0.0036 Δ today 0.0000
38
PNA
actual #38 vs paper #67 29 places worse
0.7311 ±0.0011
best reported
actual #38 vs paper #67 29 places worse
paper 0.7035 ±0.0063
paper 0.7035 ±0.0063 Δ at pub 0.0276
40
DGN
actual #40 vs paper #40
0.7284 ±0.0042
best reported
actual #40 vs paper #40
paper 0.7284 ±0.0042

Difference within reported uncertainty; not counted as underperformance.

paper 0.7284 ±0.0042 Δ at pub 0.0000
44
GIN-AK+
actual #44 vs paper #50 6 places worse
0.7239 ±0.0038
best reported
actual #44 vs paper #50 6 places worse
paper 0.7219 ±0.0013

Difference within reported uncertainty; not counted as underperformance.

paper 0.7219 ±0.0013 Δ at pub 0.0020
46
GPS
actual #46 vs paper #49 3 places worse
0.7236 ±0.0032
best reported
actual #46 vs paper #49 3 places worse
paper 0.7230 ±0.0036

Difference within reported uncertainty; not counted as underperformance.

paper 0.7230 ±0.0036 Δ today 0.0006
62
GatedGCN
actual #62 vs paper #90 28 places worse
0.7133 ±0.0039
best reported
actual #62 vs paper #90 28 places worse
paper 0.6731 ±0.0031
paper 0.6731 ±0.0031 Δ at pub 0.0402
81
CRaW1
0.6901 ±0.0026
This paper
82
EGT
0.6870 ±0.0041
This paper
92
GraphSAGE
0.6690 ±0.0020
101
GAT
actual #101 vs paper #112 11 places worse
0.6611 ±0.0098
best reported
actual #101 vs paper #112 11 places worse
paper 0.6422 ±0.0046
paper 0.6422 ±0.0046 Δ at pub 0.0189
109
GIN
actual #109 vs paper #129 20 places worse
0.6510 ±0.0050
best reported
actual #109 vs paper #129 20 places worse
paper 0.5525 ±0.0153
paper 0.5525 ±0.0153 Δ at pub 0.0985
112
Cy2C-GNNs
0.6429 ±0.0001
This paper
124
GCN
actual #124 vs paper #128 4 places worse
0.5839 ±0.0073
best reported
actual #124 vs paper #128 4 places worse
paper 0.5571 ±0.0038
paper 0.5571 ±0.0038 Δ at pub 0.0268
126
MLP
0.5634 ±0.0018
#ModelAccuracy ↑Paper
26
TIGT
1.0000
This paper
8
GCN
actual #8 vs paper #72 64 places worse
1.0000
best reported
actual #8 vs paper #72 64 places worse
paper 0.1000
best at pub: GNNBenchmark (JMLR 2023)
paper 0.1000 Δ at pub 0.9000
25
GPS+LapPE+RWSE (1 layer)
1.0000
This paper
1
PPGN
1.0000
2
RW-AgentNet
1.0000
3
AgentNet
1.0000
35
GraphSAGE
0.9993 ±0.0047
40
GIN
actual #40 vs paper #72 32 places worse
0.9933 ±0.0133
best reported
actual #40 vs paper #72 32 places worse
paper 0.1000
best at pub: GNNBenchmark (JMLR 2023)
paper 0.1000 Δ at pub 0.8933
43
Cy2C-GIN-1
0.9833 ±0.0333
This paper
52
GPS+RWSE (2 layers)
0.9333 ±0.1155
This paper
53
Cy2C-GCN-1
0.9130 ±0.0160
This paper
58
RP-GIN
actual #58 vs paper #61 3 places worse
0.5330 ±0.1290
best reported
actual #58 vs paper #61 3 places worse
paper 0.3760 ±0.1290

Difference within reported uncertainty; not counted as underperformance.

paper 0.3760 ±0.1290 Δ at pub 0.1570
60
Cy2C-GIN-2
0.4667 ±0.3820
This paper
80
GAT-1
0.1000
This paper
81
GAT-2
0.1000
This paper
82
GAT-5
0.1000
This paper
83
GAT-10
0.1000
This paper
84
GRIT+RRWP (1 layer)
0.1000
This paper
88
Cy2C-GIN-5
0.0917 ±0.0569
This paper
89
Cy2C-GIN-10
0.0749 ±0.0321
This paper
90
GPS (2 layers)
0.0667 ±0.0943
This paper
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) Best at pub — reported value improved over the best known when published Improved at pub — later paper surpassed this result Best today — OK at publication; today's leaderboard best is elsewhere