mlsanityData to 7 Oct 2026

Plain Transformers are Surprisingly Powerful Link Predictors

arXiv 2602.01553 ↗First posted 2 Feb 2026cs.LG cs.AIcited by 0

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

  1. 2017Attention Is All You Needcited by 59,932
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  3. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
  4. 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,282
  5. 2017Graph Attention Networkscited by 8,854
  6. 2017Inductive Representation Learning on Large Graphscited by 6,284
  7. 2018How Powerful are Graph Neural Networks?cited by 4,355
  8. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
  9. 2019Fast Graph Representation Learning with PyTorch Geometriccited by 2,414
  10. 2020Open Graph Benchmark: Datasets for Machine Learning on Graphscited by 2,148
  11. 2020Rethinking Attention with Performerscited by 1,584
  12. 2017Deep Setscited by 1,394
38 more
  1. 2018Deeper Insights into Graph Convolutional Networks for Semi-Supervised Learningcited by 1,099
  2. 2016Revisiting Semi-Supervised Learning with Graph Embeddingscited by 970
  3. 2018Link Prediction Based on Graph Neural Networkscited by 829
  4. 2013Exact solutions to the nonlinear dynamics of learning in deep linear neural networkscited by 748
  5. 2022Recipe for a General, Powerful, Scalable Graph Transformercited by 702
  6. 2020On the Bottleneck of Graph Neural Networks and its Practical Implicationscited by 624
  7. 2021Understanding over-squashing and bottlenecks on graphs via curvaturecited by 471
  8. 2021Rethinking Graph Transformers with Spectral Attentioncited by 468
  9. 2018Invariant and Equivariant Graph Networkscited by 339
  10. 2021Do Transformers Really Perform Bad for Graph Representation?cited by 318
  11. 2021Neural Bellman-Ford Networks: A General Graph Neural Network Framework for Link Predictioncited by 273
  12. 2020Can Graph Neural Networks Count Substructures?cited by 235
  13. 2021Weisfeiler and Lehman Go Topological: Message Passing Simplicial Networkscited by 219
  14. 2021Identity-aware Graph Neural Networkscited by 197
  15. 2022Representing Long-Range Context for Graph Neural Networks with Global Attentioncited by 193
  16. 2022Structure-Aware Transformer for Graph Representation Learningcited by 191
  17. 2020The Surprising Power of Graph Neural Networks with Random Node Initializationcited by 190
  18. 2022Pure Transformers are Powerful Graph Learnerscited by 188
  19. 2020Random Features Strengthen Graph Neural Networkscited by 184
  20. 2021From Stars to Subgraphs: Uplifting Any GNN with Local Structure Awarenesscited by 154
  21. 2019Relational Pooling for Graph Representationscited by 150
  22. 2023Graph Inductive Biases in Transformers without Message Passingcited by 150
  23. 2020Labeling Trick: A Theory of Using Graph Neural Networks for Multi-Node Representation Learningcited by 148
  24. 2022NAGphormer: A Tokenized Graph Transformer for Node Classification in Large Graphscited by 137
  25. 2024RelBench: A Benchmark for Deep Learning on Relational Databasescited by 90
  26. 2022Graph Neural Networks for Link Prediction with Subgraph Sketchingcited by 81
  27. 2023Attending to Graph Transformerscited by 81
  28. 2022Neo-GNNs: Neighborhood Overlap-aware Graph Neural Networks for Link Predictioncited by 78
  29. 2023Evaluating Graph Neural Networks for Link Prediction: Current Pitfalls and New Benchmarkingcited by 69
  30. 2023Neural Common Neighbor with Completion for Link Predictioncited by 64
  31. 2024Understanding Transformer Reasoning Capabilities via Graph Algorithmscited by 64
  32. 2023Graph Neural Networks for temporal graphs: State of the art, open challenges, and opportunitiescited by 60
  33. 2023Relational Deep Learning: Graph Representation Learning on Relational Databasescited by 55
  34. 2025PyG 2.0: Scalable Learning on Real World Graphscited by 49
  35. 2022A Practical, Progressively-Expressive GNNcited by 20
  36. 2024ContextGNN: Beyond Two-Tower Recommendation Systemscited by 19
  37. 2023Pure Message Passing Can Estimate Common Neighbor for Link Predictioncited by 18
  38. 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.