用大模型生成用户与内容嵌入,提升领英推荐召回效果
Large Scale Retrieval for the LinkedIn Feed using Causal Language Models
- 用大语言模型生成用户和内容的文本嵌入,实现端到端检索
- 量化提示词中的数值特征,使检索与排序层对齐更准确
- 新用户留存显著提升,适合高并发实时推荐系统参考
在领英信息流这类大规模推荐系统中,检索阶段需将数亿候选内容压缩至约2000个,延迟控制在几毫秒内,每秒处理数千次请求。本文提出一种新检索方法:微调大型因果语言模型(Meta的LLaMA 3)作为双编码器,仅基于文本输入生成用户与内容的高质量嵌入。文中描述了完整流程,包括嵌入生成的提示设计、大规模微调技术及低延迟、低成本的在线服务架构。研究发现,将数值特征量化后放入提示词,可有效编码信息,增强检索层与排序层的一致性。系统通过离线指标和线上A/B测试评估,显著提升了用户参与度,尤其对新用户(缺乏强社交关系者)改善明显,表明优质推荐内容有助于留存。该工作展示了生成式大模型在工业级实时高吞吐检索中的可行应用。
原文摘要 · Abstract (English)
In large scale recommendation systems like the LinkedIn Feed, the retrieval stage is critical for narrowing hundreds of millions of potential candidates to a manageable subset for ranking. LinkedIn's Feed serves suggested content from outside of the member's network (based on the member's topical interests), where 2000 candidates are retrieved from a pool of hundreds of millions candidate with a latency budget of a few milliseconds and inbound QPS of several thousand per second. This paper presents a novel retrieval approach that fine-tunes a large causal language model (Meta's LLaMA 3) as a dual encoder to generate high quality embeddings for both users (members) and content (items), using only textual input. We describe the end to end pipeline, including prompt design for embedding generation, techniques for fine-tuning at LinkedIn's scale, and infrastructure for low latency, cost effective online serving. We share our findings on how quantizing numerical features in the prompt enables the information to get properly encoded in the embedding, facilitating greater alignment between the retrieval and ranking layer. The system was evaluated using offline metrics and an online A/B test, which showed substantial improvements in member engagement. We observed significant gains among newer members, who often lack strong network connections, indicating that high-quality suggested content aids retention. This work demonstrates how generative language models can be effectively adapted for real time, high throughput retrieval in industrial applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。