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.
Benchmark Results
Tested on 1 dataset
Tables show presented results in global context (best 3 papers + GCN + MLP).
LRGB
| # | Model | MRR ↑ | Paper |
|---|---|---|---|
| 1 | FloydNet | 0.6143 | |
| 2 | Exphormer | 0.4737
±0.0024 best reported | best at pub: Exphormer: Sparse Transformers for Graphs (ICML 2023) best today: Benchmarking Positional Encodings for GNNs and Graph Transformers (KDD 2024) |
| 3 | Exphormer + LapPE | 0.4737
±0.0024 | |
| 5 | GPS | 0.3498
±0.0005 | prev best: Recipe for a General, Powerful, Scalable Graph Transformer (NeurIPS 2022) best today: Learning Long Range Dependencies on Graphs via Random Walks (ICLR 2024) |
| 8 | GatedGCN | 0.3495
±0.0010 | prev best: Recipe for a General, Powerful, Scalable Graph Transformer (NeurIPS 2022) best today: Benchmarking Positional Encodings for GNNs and Graph Transformers (KDD 2024) |
| 16 | GIN | 0.4617
±0.0005 | |
| 28 | GCN | 0.3424
±0.0007 | prev best: Recipe for a General, Powerful, Scalable Graph Transformer (NeurIPS 2022) best today: Benchmarking Positional Encodings for GNNs and Graph Transformers (KDD 2024) |
| 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 |
