arXiv:2606.00422cs.IRcs.LG2026-06被引 4

统一推荐系统的检索与排序,提升效率并增益用户体验。

UniPinRec: Unifying Generative Retrieval and Ranking at Pinterest Scale

  • 共享编码器+任务头,用同一模型处理检索和排序。
  • 上线后在线互动率提升1%,延迟降低11.1%,每秒请求数提高63.6%。
  • 适合大规模推荐系统优化,尤其关注部署效率的团队。

现代推荐系统通常将检索与排序分别训练,尽管二者均依赖大模型处理相同用户行为数据,造成参数、算力和部署成本重复。现有工作虽统一模型架构,但输入格式、训练流程和部署栈仍割裂。我们提出UniPinRec,实现Pinterest规模下检索与排序的全栈统一:统一输入格式、统一模型、统一训练阶段,并部署于现有服务基础设施。一个共享的Transformer将用户行为序列编码为候选无关表示,通过任务专用头部分支出检索(近似最近邻点积)和排序(交叉注意力)。三个关键设计使其可行:(1) 隐式行为建模(MAM)消除交错,支持权重共享且不增加上下文长度;(2) 混合训练样本将行为序列与首页展示候选集配对,联合满足双目标;(3) 跨阶段键值缓存共享,复用检索阶段的用户历史计算,相比独立模型减少总浮点运算量。在Pinterest核心场景上线后,该系统带来约+1%的在线互动率提升,端到端延迟降低11.1%,每秒请求数(QPS)提升63.6%。据我们所知,这是首个在生产系统中实现输入、模型、训练与服务全栈统一的检索-排序融合方案。

原文摘要 · Abstract (English)

Modern recommendation systems predominantly train retrieval and ranking as separate models despite both increasingly relying on large transformers encoding the same user behavior data, duplicating parameters, compute, and serving cost. Prior work unifies the model architecture but not the full pipeline: input formats, training procedures, and serving stacks remain fragmented across stages. We present UniPinRec, which achieves full-stack unification of retrieval and ranking at Pinterest: one input format, one model, one training stage, deployed within existing serving infrastructure. A shared transformer encodes the user action sequence into candidate-independent representations that branch into retrieval (ANN dot-product) and ranking (cross-attention) via task-specific heads. Three ideas make this work: (1) Masked Action Modeling (MAM) eliminates interleaving, enabling weight sharing without doubling context length; (2) Blended training examples pair action sequences with feedview impression slates to satisfy both objectives jointly; (3) Cross-stage KV cache sharing reuses user-history computation from retrieval for ranking, reducing total FLOPs versus serving two independent models. Deployed in the Pinterest core surfaces, UniPinRec delivers approximately +1% online engagement lift while cutting end-to-end serving latency by 11.1% and lifting QPS by 63.6%. To our knowledge, this is the first full-stack unification of retrieval and ranking, covering inputs, model, training and serving, deployed in a production recommendation system.

推荐系统模型统一效率优化工业应用

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