Abstract:Transformers have recently emerged as powerful neural networks for graph learning, showcasing state-of-the-art performance on several graph property prediction tasks. However, these results have been limited to small-scale graphs, where the computational feasibility of the global attention mechanism is possible. The next goal is to scale up these architectures to handle very large graphs on the scale of millions or even billions of nodes. With large-scale graphs, global attention learning is proven impractical due to its quadratic complexity w.r.t. the number of nodes. On the other hand, neighborhood sampling techniques become essential to manage large graph sizes, yet finding the optimal trade-off between speed and accuracy with sampling techniques remains challenging. This work advances representation learning on single large-scale graphs with a focus on identifying model characteristics and critical design constraints for developing scalable graph transformer (GT) architectures. We argue such GT requires layers that can adeptly learn both local and global graph representations while swiftly sampling the graph topology. As such, a key innovation of this work lies in the creation of a fast neighborhood sampling technique coupled with a local attention mechanism that encompasses a 4-hop reception field, but achieved through just 2-hop operations. This local node embedding is then integrated with a global node embedding, acquired via another self-attention layer with an approximate global codebook, before finally sent through a downstream layer for node predictions. The proposed GT framework, named LargeGT, overcomes previous computational bottlenecks and is validated on three large-scale node classification benchmarks. We report a 3x speedup and 16.8% performance gain on ogbn-products and snap-patents, while we also scale LargeGT on ogbn-papers100M with a 5.9% performance improvement.

Breaking the Bottleneck on Graphs with Structured State Spaces

Discovering the Representation Bottleneck of Graph Neural Networks from Multi-order Interactions

NGAT: Attention in Breadth and Depth Exploration for Semi-Supervised Graph Representation Learning

What Can We Learn from State Space Models for Machine Learning on Graphs?

Discovering the Representation Bottleneck of Graph Neural Networks

Graph Mamba: Towards Learning on Graphs with State Space Models

Discovering and Explaining the Representation Bottleneck of Graph Neural Networks from Multi-order Interactions

From Stars to Subgraphs: Uplifting Any GNN with Local Structure Awareness

SpikeGraphormer: A High-Performance Graph Transformer with Spiking Graph Attention

Reliable Representations Make A Stronger Defender: Unsupervised Structure Refinement for Robust GNN

Global Minima, Recoverability Thresholds, and Higher-Order Structure in GNNS

Exploiting Spiking Dynamics with Spatial-temporal Feature Normalization in Graph Learning

NodeFormer: A Scalable Graph Structure Learning Transformer for Node Classification

Representing Long-Range Context for Graph Neural Networks with Global Attention

AnchorGT: Efficient and Flexible Attention Architecture for Scalable Graph Transformers

Rethinking Structure Learning For Graph Neural Networks

Graph Transformers for Large Graphs

Graph Structural-topic Neural Network

Theoretically Improving Graph Neural Networks Via Anonymous Walk Graph Kernels

Spatial-Temporal Dynamic Hypergraph Information Bottleneck for Brain Network Classification

Limiting Over-Smoothing and Over-Squashing of Graph Message Passing by Deep Scattering Transforms