arXiv:2412.13844cs.IRcs.AI2024-12被引 4

让检索模型同时考虑点击率和观看时长,提升推荐精准度。

CRM: Retrieval Model with Controllable Condition

  • 将观看时长等回归目标作为条件特征融入双塔检索模型
  • 在快手超4亿用户系统中通过真实A/B测试验证效果提升
  • 适合需要兼顾多种业务指标的工业级推荐系统使用

推荐系统通常分为检索和排序两个阶段:检索阶段从海量候选中筛选出数百个相关项目,排序阶段则基于多目标估计(如点击率、观看时长)选出最优数十项。传统检索模型因推理时缺乏候选项目信息,仅能使用点击率等分类目标训练,难以融合观看时长等回归目标,限制了检索效果。本文提出可控检索模型(CRM),将回归信息作为条件特征引入双塔检索架构,使检索阶段能更有效地匹配用户兴趣与业务条件。我们在快手短视频推荐系统(服务超4亿用户)中通过真实A/B测试验证了该方法的有效性,并成功部署。

原文摘要 · Abstract (English)

Recommendation systems (RecSys) are designed to connect users with relevant items from a vast pool of candidates while aligning with the business goals of the platform. A typical industrial RecSys is composed of two main stages, retrieval and ranking: (1) the retrieval stage aims at searching hundreds of item candidates satisfied user interests; (2) based on the retrieved items, the ranking stage aims at selecting the best dozen items by multiple targets estimation for each item candidate, including classification and regression targets. Compared with ranking model, the retrieval model absence of item candidate information during inference, therefore retrieval models are often trained by classification target only (e.g., click-through rate), but failed to incorporate regression target (e.g., the expected watch-time), which limit the effectiveness of retrieval. In this paper, we propose the Controllable Retrieval Model (CRM), which integrates regression information as conditional features into the two-tower retrieval paradigm. This modification enables the retrieval stage could fulfill the target gap with ranking model, enhancing the retrieval model ability to search item candidates satisfied the user interests and condition effectively. We validate the effectiveness of CRM through real-world A/B testing and demonstrate its successful deployment in Kuaishou short-video recommendation system, which serves over 400 million users.

推荐系统双塔模型多目标学习工业落地

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