千亿级视觉平台用大模型理解用户行为序列,提升新内容推荐效果
PinFM: Foundation Model for User Activity Sequences at a Billion-scale Visual Discovery Platform
- 基于200亿参数Transformer预训练用户行为序列,支持多场景应用
- 通过序列改写使新物品点击率提升20%,系统每秒可处理百万级推荐
- 采用去重交叉注意力等优化技术,吞吐量提升600%,适合工业级部署
用户行为序列已成为推荐系统中最重要信号之一。我们提出针对千亿级视觉发现平台的通用模型PinFM,通过200亿以上参数的Transformer在大规模用户行为数据上进行预训练,再针对具体应用场景微调,高效集成至现有系统。尽管该预训练-微调范式在视觉与自然语言领域已成熟,但其在工业推荐系统中面临诸多挑战:模型需在极低延迟和成本约束下每秒评分数百万个物品,同时捕捉用户行为与其他特征的交互,并处理预训练阶段未出现的新物品。为此,我们开发了创新性技术和优化方案,如去重交叉注意力变换器(DCAT),在Pinterest内部数据上将吞吐量提升600%。实验表明,通过调整输入序列,PinFM能有效学习用户序列与候选物品间的交互,使新物品参与度提升20%。目前,PinFM已部署于多个核心应用,服务超5亿用户。
原文摘要 · Abstract (English)
User activity sequences have emerged as one of the most important signals in recommender systems. We present a foundational model, PinFM, for understanding user activity sequences across multiple applications at a billion-scale visual discovery platform. We pretrain a transformer model with 20B+ parameters using extensive user activity data, then fine-tune it for specific applications, efficiently coupling it with existing models. While this pretraining-and-fine-tuning approach has been popular in other domains, such as Vision and NLP, its application in industrial recommender systems presents numerous challenges. The foundational model must be scalable enough to score millions of items every second while meeting tight cost and latency constraints imposed by these systems. Additionally, it should capture the interactions between user activities and other features and handle new items that were not present during the pretraining stage. We developed innovative techniques to address these challenges. Our infrastructure and algorithmic optimizations, such as the Deduplicated Cross-Attention Transformer (DCAT), improved our throughput by 600% on Pinterest internal data. We demonstrate that PinFM can learn interactions between user sequences and candidate items by altering input sequences, leading to a 20% increase in engagement with new items. PinFM is now deployed to help improve the experience of more than half a billion users across various applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。