GEN: A Practical Alternative to Graph Transformers for Long-Range Graph Modeling

Shuo Wang, Ge Cheng, Yun Zhang

Main figure for GEN: A Practical Alternative to Graph Transformers for Long-Range Graph Modeling

Abstract

Message Passing Neural Networks (MPNNs) model local relations effectively but struggle to propagate information over long distances. Graph Transformers (GTs) mitigate this via global self-attention, yet their quadratic cost in the number of nodes limits scalability. We propose Graph Elimination Networks (GENs), an MPNN variant that approximates GT-like long-range modeling while maintaining high efficiency. GENs combine edge-wise and hop-wise self-attention in parallel; their multiplicative composition yields an attention kernel separable across edge and hop factors within a bounded K-hop receptive field. To enable hop-wise attention, we introduce the Graph Elimination Algorithm (GEA), which prevents double counting across hops, ensuring that each round injects the k-hop incremental contribution exactly once. Taking differences between successive rounds recovers the k-hop increment and yields disentangled multi-hop features as inputs for hop-wise attention. This preserves clearer structural distinctions across hop distances and enables more faithful modeling of pairwise dependencies between distant nodes within the K-hop neighborhood. On the Long-Range Graph Benchmark (LRGB), GENs outperform strong MPNN baselines by 7.7 and 6.0 percentage points (pp) on PascalVOC-SP and COCO-SP, and achieve performance on par with or better than state-of-the-art Graph Transformers. On OGBN-Products, GENs support full-batch training/inference, while sparse-attention baselines like Exphormer struggle with memory limits under comparable budgets, highlighting GENs as a practical alternative for large, sparse graphs.

Proposed method: GENs

Benchmark Results

Tested on 1 dataset

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

LRGB

#ModelMRR ↑Paper
1
FloydNet
0.6143
2
Exphormer
actual #2 vs paper #36 34 places worse
0.4737 ±0.0024
best reported
actual #2 vs paper #36 34 places worse
paper 0.3637 ±0.0020
paper 0.3637 ±0.0020 Δ today 0.1100
3
Exphormer + LapPE
0.4737 ±0.0024
5
GPS
actual #5 vs paper #42 37 places worse
0.3498 ±0.0005
actual #5 vs paper #42 37 places worse
paper 0.3498 ±0.0005
paper 0.3498 ±0.0005 Δ at pub 0.0161 Δ today 0.1205
8
GatedGCN
actual #8 vs paper #42 34 places worse
0.3495 ±0.0010
actual #8 vs paper #42 34 places worse
paper 0.3495 ±0.0010
paper 0.3495 ±0.0010 Δ at pub 0.0277 Δ today 0.1175
16
GIN
0.4617 ±0.0005
28
GCN
actual #28 vs paper #52 24 places worse
0.3424 ±0.0007
actual #28 vs paper #52 24 places worse
paper 0.3424 ±0.0007
paper 0.3424 ±0.0007 Δ at pub 0.0190 Δ today 0.1102
40
GENs
0.3520 ±0.0010
This paper
42
IPR-MPNN
0.3516 ±0.0102
This paper
47
DRew
0.3444 ±0.0017
This paper
50
GAT
0.3441 ±0.0008
This paper
53
MixHop
0.3414 ±0.0006
This paper
55
Graph-Mamba
0.3395 ±0.0013
This paper
59
MAGNA
0.3364 ±0.0008
This paper
68
DAGNN
0.3287 ±0.0020
This paper
SOTA performance (not significantly different from the best) Proposed method (this paper) Baseline / comparison (this paper) Global context (other papers) Improved at pub — later paper surpassed this result Best today — OK at publication; today's leaderboard best is elsewhere