Scaling Graph Neural Networks for Friend Recommendation: Multi-Hash User Embeddings and Temporal Neighbor Sampling
Not on the map. Why?
Abstract (arXiv)
Friend recommendation is inherently graph-structured: the relevance of a potential connection depends on multi-hop social context rather than user attributes alone. However, deploying message-passing GNNs on a production-scale social graph with hundreds of millions of users and tens of billions of edges requires addressing numerous modeling and systems challenges. We present a scalable end-to-end GNN ranking system for production social graphs, focusing on two design choices that are critical in this setting: multi-hash ID embeddings and temporal neighbor sampling. Multi-hash embeddings are common for high-cardinality features, but industrial GNN systems typically either ignore trainable IDs or accept full embedding tables, exceeding 200 GB for our graph. We integrate multi-hash as the primary node representation, reducing the ID-embedding table size by more than 98 percent while preserving ranking quality. Temporal neighbor sampling is well understood in principle, but existing implementations scan full adjacency lists, which is a non-starter for users with tens of thousands of friends. We implement timestamp-sorted CSR storage with binary search, reducing the per-node temporal sampling cost from O(deg(v) + k) to O(log(deg(v)) + k). Beyond these components, we show that this combination scales and yields measurable production impact. On a graph with 194M users and 28B edges, offline ablations isolate each design choice's contribution. In an online A/B test, our system increases friend additions from recommendations by 16 percent and unique friend adders by 11.5 percent over a strong production baseline. We release our framework for distributed training and inference on large temporal graphs.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 20 of its 28 references that are in the map. First posted on arXiv 27 Aug 2026.
Builds on
20 of 28 references.
References · 20
- 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
- 2019Fast Graph Representation Learning with PyTorch Geometriccited by 2,414
- 2020LightGCN: Simplifying and Powering Graph Convolution Network for Recommendationcited by 1,134
- 2018Graph Convolutional Neural Networks for Web-Scale Recommender Systemscited by 1,133
- 2021How Attentive are Graph Attention Networks?cited by 818
- 2019Neural Graph Collaborative Filteringcited by 660
- 2020Temporal Graph Networks for Deep Learning on Dynamic Graphscited by 521
- 2020Inductive Representation Learning on Temporal Graphscited by 454
- 2019Compositional Embeddings Using Complementary Partitions for Memory-Efficient Recommendation Systemscited by 68
- 2022TGL: A General Framework for Temporal GNN Training on Billion-Scale Graphscited by 59
8 more
- 2017Hash Embeddings for Efficient Word Representationscited by 54
- 2020Learning to Embed Categorical Features without Embedding Tables for Recommendationcited by 38
- 2024LiGNN: Graph Neural Networks at LinkedIncited by 31
- 2025GiGL: Large-Scale Graph Neural Networks at Snapchatcited by 8
- 2024GraphStorm: all-in-one graph machine learning framework for industry applicationscited by 7
- 2021Position-based Hash Embeddings For Scaling Graph Neural Networkscited by 3
- 2025Ultra Fast Warm Start Solution for Graph Recommendationscited by 2
- 2024GNN Applied to Ego-nets for Friend Suggestionscited by 1
Built on by
Not cited yet.