arXiv:2411.10057cs.IRcs.AI2024-11被引 23

用Transformer重构推荐检索,提升用户使用时长

KuaiFormer: Transformer-Based Retrieval at Kuaishou

  • 改用Transformer预测用户下一步行为,而非传统点击率评分
  • 上线后日活超4亿,平均使用时长显著增长
  • 为工业级大模型部署提供技术与业务双视角参考

在大规模内容推荐系统中,检索是流水线的第一阶段,需从数十亿项目中筛选数千个候选项传递给排序模块。传统方法主要采用基于嵌入的双塔深度神经网络检索(EBR)。尽管近年来研究聚焦于将Transformer应用于检索任务,但其在工业场景中的实际部署仍面临重大挑战。本文提出KuaiFormer,一种部署于快手应用短视频推荐系统的新型Transformer检索框架。KuaiFormer通过将检索过程从传统的打分估计(如点击率预测)转变为Transformer驱动的下一步行为预测范式,实现更高效实时的兴趣获取和多兴趣提取,显著提升检索性能。该系统自2024年5月起正式接入快手App,服务超过4亿日活跃用户,显著提升了用户平均每日使用时长。本文同时探讨了在大规模推荐系统中部署Transformer的技术与商业实践,总结了实际落地中的关键挑战,为相关工程师与研究者优化大规模内容推荐系统提供了宝贵经验。

原文摘要 · Abstract (English)

In large-scale content recommendation systems, retrieval serves as the initial stage in the pipeline, responsible for selecting thousands of candidate items from billions of options to pass on to ranking modules. Traditionally, the dominant retrieval method has been Embedding-Based Retrieval (EBR) using a Deep Neural Network (DNN) dual-tower structure. However, applying transformer in retrieval tasks has been the focus of recent research, though real-world industrial deployment still presents significant challenges. In this paper, we introduce KuaiFormer, a novel transformer-based retrieval framework deployed in a large-scale content recommendation system. KuaiFormer fundamentally redefines the retrieval process by shifting from conventional score estimation tasks (such as click-through rate estimate) to a transformer-driven Next Action Prediction paradigm. This shift enables more effective real-time interest acquisition and multi-interest extraction, significantly enhancing retrieval performance. KuaiFormer has been successfully integrated into Kuaishou App's short-video recommendation system since May 2024, serving over 400 million daily active users and resulting in a marked increase in average daily usage time of Kuaishou users. We provide insights into both the technical and business aspects of deploying transformer in large-scale recommendation systems, addressing practical challenges encountered during industrial implementation. Our findings offer valuable guidance for engineers and researchers aiming to leverage transformer models to optimize large-scale content recommendation systems.

推荐系统Transformer检索模型工业部署

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。