Pre-training Language Model for Friend Recommendation: A Case Study of Large Social Graph
Abstract
Despite the demonstrated success of Pre-trained Language Models (PLM) in enhancing various recommendation tasks, their performance in friend recommendation systems, which are heavily influenced by complex social network dynamics, remains unproven. On the other hand, for social networks like Facebook, graph-based machine learning approaches in friend suggestion, like Graph Neural Networks (GNNs), struggle with the computational demands posed by social graphs. In is paper we investigate the application of PLM and GNN in the context of PYMK features within Facebook's social graph. We propose a representation learning scheme that captures both the local structure and semantic content within a second-degree connection space. Our approach leverages pre-trained transformer models with a feature aggregation scheme that enables efficient node representation learning in large social networks. We detail our model's architecture and discuss our training methods, accompanied by detailed experiments. Our model has improved the amount of friending on Facebook, facilitating new connections for users each day in online experiment.
Fields of Study & Subjects
Recommender Systems and Techniques
Advanced Graph Neural Networks
Complex Network Analysis Techniques
Publication details
- Journal / source
- ACM International Conference on Web Search and Data Mining (WSDM)
- Pages
- 1351–1352
- Publication date
- 16 February 2026
- Language
- EN