NeurIPS 2024

Towards Principled Graph Transformers

Luis Müller, Christopher Morris

Main figure for Towards Principled Graph Transformers

Abstract

The expressive power of graph learning architectures based on the $k$-dimensional Weisfeiler--Leman hierarchy is well understood. However, such architectures often fail to deliver solid predictive performance on real-world tasks, limiting their practical impact. In contrast, global attention-based models such as graph transformers demonstrate strong performance in practice. However, comparing their expressivity with the hierarchy remains challenging, particularly since attention-based architectures rely on positional or structural encodings. To address this, we show that the recently proposed Edge Transformer, a global attention model operating on node pairs instead of nodes, has expressive power when provided with the right tokenization. Empirically, we demonstrate that the Edge Transformer surpasses other theoretically aligned architectures regarding predictive performance and is competitive with state-of-the-art models on algorithmic reasoning and molecular regression tasks while not relying on positional or structural encodings. Our code is available at.

Proposed method: ET

Benchmark Results

Tested on 2 datasets 1 baseline value reported too low

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

Classic

#ModelMAE ↓Paper
1
TIGT
0.0140 ±0.0010
2
FloydNet
0.0160
3
GSSC
0.0190 ±0.0010
4
PPGN++
0.0200 ±0.0010
This paper
7
CIN
actual #7 vs paper #8 1 place worse
0.0210 ±0.0010
best reported
actual #7 vs paper #8 1 place worse
paper 0.0220 ±0.0020

Difference within reported uncertainty; not counted as underperformance.

paper 0.0220 ±0.0020 Δ at pub 0.0010
19
SignNet
0.0240 ±0.0030
This paper
18
ET+RRWP
actual #18 vs paper #58 40 places worse
0.0240
best reported
actual #18 vs paper #58 40 places worse
paper 0.0590 ±0.0040
paper 0.0590 ±0.0040 Δ today 0.0350
23
Graphormer-GD
0.0250 ±0.0040
This paper
27
ET
0.0260 ±0.0030
This paper
69
GIN
0.0880 ±0.0020
88
GIN(E)
0.1800 ±0.0060
This paper

OGB

#ModelMAE ↓Paper
1
DeMol
0.0603
2
TGT-At
0.0671
3
TGT-At (+RDKit)
0.0683
12
GPTrans-L
0.0809
This paper
15
ET+RRWP
0.0832
This paper
18
ET
0.0840
This paper
23
GraphGPS Medium
0.0858
This paper
27
GRIT
0.0859
This paper
29
EGT
actual #29 vs paper #33 4 places worse
0.0862
best reported
31
Graphormer
0.0864
This paper
45
TokenGT Lap
0.0910
This paper
51
GraphGPS Small
0.0938
This paper
54
TokenGT ORF
0.0962
This paper
64
GIN
0.1195
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 today — OK at publication; today's leaderboard best is elsewhere