arXiv:2504.06609cs.IRcs.AI2025-04被引 7

通过用户历史行为建模跨交互特征,提升大规模搜索预排序效果。

InteractRank: Personalized Web-Scale Search Pre-Ranking with Cross Interaction Features

  • 在双塔模型中引入基于用户历史的查询-物品交叉特征。
  • 线上实验提升6.5%互动率,较基线显著改善。
  • 适合需要高效个性化排序的推荐系统开发者。

现代搜索系统采用多阶段架构以高效提供个性化结果,关键阶段包括检索、预排序、全排序和融合。预排序阶段需将数十万项压缩至数千项,通常依赖计算高效的双塔模型,但难以捕捉复杂交互。尽管查询-物品交叉特征对全排序至关重要,但将其融入预排序面临效率挑战。本文提出InteractRank,一种在Pinterest上线的新型双塔预排序模型,通过在打分函数中结合基于用户历史行为的查询-物品交互特征与双塔点积,显著提升性能且几乎不增加延迟和计算成本。真实世界A/B测试显示,相比BM25基线,互动指标提升6.5%;相比普通双塔基线,提升3.7%。我们还展示了实时用户序列建模等组件,并通过离线消融实验分析其贡献。代码已开源。

原文摘要 · Abstract (English)

Modern search systems use a multi-stage architecture to deliver personalized results efficiently. Key stages include retrieval, pre-ranking, full ranking, and blending, which refine billions of items to top selections. The pre-ranking stage, vital for scoring and filtering hundreds of thousands of items down to a few thousand, typically relies on two tower models due to their computational efficiency, despite often lacking in capturing complex interactions. While query-item cross interaction features are paramount for full ranking, integrating them into pre-ranking models presents efficiency-related challenges. In this paper, we introduce InteractRank, a novel two tower pre-ranking model with robust cross interaction features used at Pinterest. By incorporating historical user engagement-based query-item interactions in the scoring function along with the two tower dot product, InteractRank significantly boosts pre-ranking performance with minimal latency and computation costs. In real-world A/B experiments at Pinterest, InteractRank improves the online engagement metric by 6.5% over a BM25 baseline and by 3.7% over a vanilla two tower baseline. We also highlight other components of InteractRank, like real-time user-sequence modeling, and analyze their contributions through offline ablation studies. The code for InteractRank is available at https://github.com/pinterest/atg-research/tree/main/InteractRank.

搜索排序双塔模型用户行为建模

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