mlsanityData to 7 Oct 2026

Scaling Graph Neural Networks for Friend Recommendation: Multi-Hash User Embeddings and Temporal Neighbor Sampling

arXiv 2608.27413 ↗First posted 27 Aug 2026cs.IR cs.LG cs.SIcited by 0

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

  1. 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,282
  2. 2017Graph Attention Networkscited by 8,854
  3. 2017Inductive Representation Learning on Large Graphscited by 6,284
  4. 2019Fast Graph Representation Learning with PyTorch Geometriccited by 2,414
  5. 2020LightGCN: Simplifying and Powering Graph Convolution Network for Recommendationcited by 1,134
  6. 2018Graph Convolutional Neural Networks for Web-Scale Recommender Systemscited by 1,133
  7. 2021How Attentive are Graph Attention Networks?cited by 818
  8. 2019Neural Graph Collaborative Filteringcited by 660
  9. 2020Temporal Graph Networks for Deep Learning on Dynamic Graphscited by 521
  10. 2020Inductive Representation Learning on Temporal Graphscited by 454
  11. 2019Compositional Embeddings Using Complementary Partitions for Memory-Efficient Recommendation Systemscited by 68
  12. 2022TGL: A General Framework for Temporal GNN Training on Billion-Scale Graphscited by 59
8 more
  1. 2017Hash Embeddings for Efficient Word Representationscited by 54
  2. 2020Learning to Embed Categorical Features without Embedding Tables for Recommendationcited by 38
  3. 2024LiGNN: Graph Neural Networks at LinkedIncited by 31
  4. 2025GiGL: Large-Scale Graph Neural Networks at Snapchatcited by 8
  5. 2024GraphStorm: all-in-one graph machine learning framework for industry applicationscited by 7
  6. 2021Position-based Hash Embeddings For Scaling Graph Neural Networkscited by 3
  7. 2025Ultra Fast Warm Start Solution for Graph Recommendationscited by 2
  8. 2024GNN Applied to Ego-nets for Friend Suggestionscited by 1

Built on by

Not cited yet.