arXiv:2506.06239cs.IR2025-06KDD被引 4

兼顾召回率与语义相关性,提升推荐系统精准度。

Optimizing Recall or Relevance? A Multi-Task Multi-Head Approach for Item-to-Item Retrieval in Recommendation

  • 多任务多头架构同时优化召回与语义相关性
  • 召回率最高提升14.4%,语义相关性提升56.6%
  • 适合追求推荐多样性与长期用户体验的场景

物品到物品(I2I)检索旨在基于触发物品找出相关且高吸引力的物品,是现代推荐系统的核心。现有工业级模型主要依赖共浏览数据并以召回率为优化目标,过度强调短期共现模式,忽视语义相关性,导致难以发现新兴趣和促进内容多样性。为此,我们提出MTMH模型,通过多任务学习损失函数显式权衡召回与语义相关性,并采用多头架构同时检索高度共现和语义相关的物品。在服务数十亿用户的商业平台数据上评估,相比现有最优模型,MTMH可使召回率最高提升14.4%,语义相关性提升56.6%。线上实验验证其能同时提升短期消费指标和长期用户体验指标。本工作为联合优化I2I召回与语义相关性提供了系统性方法,对整体推荐性能提升具有重要意义。

原文摘要 · Abstract (English)

The task of item-to-item (I2I) retrieval is to identify a set of relevant and highly engaging items based on a given trigger item. It is a crucial component in modern recommendation systems, where users' previously engaged items serve as trigger items to retrieve relevant content for future engagement. However, existing I2I retrieval models in industry are primarily built on co-engagement data and optimized using the recall measure, which overly emphasizes co-engagement patterns while failing to capture semantic relevance. This often leads to overfitting short-term co-engagement trends at the expense of long-term benefits such as discovering novel interests and promoting content diversity. To address this challenge, we propose MTMH, a Multi-Task and Multi-Head I2I retrieval model that achieves both high recall and semantic relevance. Our model consists of two key components: 1) a multi-task learning loss for formally optimizing the trade-off between recall and semantic relevance, and 2) a multi-head I2I retrieval architecture for retrieving both highly co-engaged and semantically relevant items. We evaluate MTMH using proprietary data from a commercial platform serving billions of users and demonstrate that it can improve recall by up to 14.4% and semantic relevance by up to 56.6% compared with prior state-of-the-art models. We also conduct live experiments to verify that MTMH can enhance both short-term consumption metrics and long-term user-experience-related metrics. Our work provides a principled approach for jointly optimizing I2I recall and semantic relevance, which has significant implications for improving the overall performance of recommendation systems.

推荐系统多任务学习召回优化语义相关性

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