Plain Transformers are Surprisingly Powerful Link Predictors
Not on the map. Why?
Abstract (arXiv)
Link prediction is a core challenge in graph machine learning, demanding models that capture rich and complex topological dependencies. While Graph Neural Networks (GNNs) are the standard solution, state-of-the-art pipelines often rely on explicit structural heuristics or memory-intensive node embeddings -- approaches that struggle to generalize or scale to massive graphs. Emerging Graph Transformers (GTs) offer a potential alternative but often incur significant overhead due to complex structural encodings, hindering their applications to large-scale link prediction. We challenge these sophisticated paradigms with PENCIL, an encoder-only plain Transformer that replaces hand-crafted priors with attention over sampled local subgraphs, retaining the scalability and hardware efficiency of standard Transformers. Through experimental and theoretical analysis, we show that PENCIL extracts richer structural signals than GNNs, implicitly generalizing a broad class of heuristics and subgraph-based expressivity. Empirically, PENCIL outperforms heuristic-informed GNNs and is far more parameter-efficient than ID-embedding--based alternatives, while remaining competitive across diverse benchmarks -- even without node features. Our results challenge the prevailing reliance on complex engineering techniques, demonstrating that simple design choices are potentially sufficient to achieve the same capabilities. Our code is publicly available at https://github.com/quang-truong/pencil.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 63 of its 84 references that are in the map. First posted on arXiv 2 Feb 2026.
Builds on
63 of 84 references.
References · 63
- 2017Attention Is All You Needcited by 59,932
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,282
- 2017Graph Attention Networkscited by 8,854
- 2017Inductive Representation Learning on Large Graphscited by 6,284
- 2018How Powerful are Graph Neural Networks?cited by 4,355
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
- 2019Fast Graph Representation Learning with PyTorch Geometriccited by 2,414
- 2020Open Graph Benchmark: Datasets for Machine Learning on Graphscited by 2,148
- 2020Rethinking Attention with Performerscited by 1,584
- 2017Deep Setscited by 1,394
38 more
- 2018Deeper Insights into Graph Convolutional Networks for Semi-Supervised Learningcited by 1,099
- 2016Revisiting Semi-Supervised Learning with Graph Embeddingscited by 970
- 2018Link Prediction Based on Graph Neural Networkscited by 829
- 2013Exact solutions to the nonlinear dynamics of learning in deep linear neural networkscited by 748
- 2022Recipe for a General, Powerful, Scalable Graph Transformercited by 702
- 2020On the Bottleneck of Graph Neural Networks and its Practical Implicationscited by 624
- 2021Understanding over-squashing and bottlenecks on graphs via curvaturecited by 471
- 2021Rethinking Graph Transformers with Spectral Attentioncited by 468
- 2018Invariant and Equivariant Graph Networkscited by 339
- 2021Do Transformers Really Perform Bad for Graph Representation?cited by 318
- 2021Neural Bellman-Ford Networks: A General Graph Neural Network Framework for Link Predictioncited by 273
- 2020Can Graph Neural Networks Count Substructures?cited by 235
- 2021Weisfeiler and Lehman Go Topological: Message Passing Simplicial Networkscited by 219
- 2021Identity-aware Graph Neural Networkscited by 197
- 2022Representing Long-Range Context for Graph Neural Networks with Global Attentioncited by 193
- 2022Structure-Aware Transformer for Graph Representation Learningcited by 191
- 2020The Surprising Power of Graph Neural Networks with Random Node Initializationcited by 190
- 2022Pure Transformers are Powerful Graph Learnerscited by 188
- 2020Random Features Strengthen Graph Neural Networkscited by 184
- 2021From Stars to Subgraphs: Uplifting Any GNN with Local Structure Awarenesscited by 154
- 2019Relational Pooling for Graph Representationscited by 150
- 2023Graph Inductive Biases in Transformers without Message Passingcited by 150
- 2020Labeling Trick: A Theory of Using Graph Neural Networks for Multi-Node Representation Learningcited by 148
- 2022NAGphormer: A Tokenized Graph Transformer for Node Classification in Large Graphscited by 137
- 2024RelBench: A Benchmark for Deep Learning on Relational Databasescited by 90
- 2022Graph Neural Networks for Link Prediction with Subgraph Sketchingcited by 81
- 2023Attending to Graph Transformerscited by 81
- 2022Neo-GNNs: Neighborhood Overlap-aware Graph Neural Networks for Link Predictioncited by 78
- 2023Evaluating Graph Neural Networks for Link Prediction: Current Pitfalls and New Benchmarkingcited by 69
- 2023Neural Common Neighbor with Completion for Link Predictioncited by 64
- 2024Understanding Transformer Reasoning Capabilities via Graph Algorithmscited by 64
- 2023Graph Neural Networks for temporal graphs: State of the art, open challenges, and opportunitiescited by 60
- 2023Relational Deep Learning: Graph Representation Learning on Relational Databasescited by 55
- 2025PyG 2.0: Scalable Learning on Real World Graphscited by 49
- 2022A Practical, Progressively-Expressive GNNcited by 20
- 2024ContextGNN: Beyond Two-Tower Recommendation Systemscited by 19
- 2023Pure Message Passing Can Estimate Common Neighbor for Link Predictioncited by 18
- 2024Learning Long Range Dependencies on Graphs via Random Walkscited by 18
50 most cited of 63 All 63 (JSON)
Built on by
Not cited yet.