超长用户历史推荐模型,百亿参数仍保持高效推理
Massive Memorization with Hundreds of Trillions of Parameters for Sequential Transducer Generative Recommenders
- 分两阶段建模:先压缩历史为数百个摘要标记,再注意力匹配候选项
- 支持百万级用户历史,下游训练与推理成本不变
- 已在亿级用户平台落地,显著提升离线与线上指标
现代大规模推荐系统依赖用户交互历史序列提升性能。大语言模型与序列建模技术(如Transformer架构)的兴起推动了显著进展(如HSTU、SIM、TWIN模型)。尽管扩展至超长用户历史(1万至10万项)通常能提升模型表现,但会带来延迟、每秒查询数(QPS)和GPU成本的巨大挑战。现有模型未能有效解决工业可扩展性问题。本文提出新型两阶段建模框架VIrtual Sequential Target Attention(VISTA),将传统从候选项到用户历史项的目标注意力分解为两个阶段:(1) 将用户历史压缩为数百个摘要标记;(2) 候选项对这些标记进行注意力计算。这些摘要标记嵌入被缓存至存储系统,作为下游训练与推理的序列特征。该设计极大提升了可扩展性,使VISTA可在保持下游训练与推理成本恒定的前提下,支持终身用户历史(最高达一百万项),这对工业场景至关重要。实验表明,本方法在离线与在线指标上均有显著提升,并已在行业领先推荐平台成功部署,服务数十亿用户。
原文摘要 · Abstract (English)
Modern large-scale recommendation systems rely heavily on user interaction history sequences to enhance the model performance. The advent of large language models and sequential modeling techniques, particularly transformer-like architectures, has led to significant advancements recently (e.g., HSTU, SIM, and TWIN models). While scaling to ultra-long user histories (10k to 100k items) generally improves model performance, it also creates significant challenges on latency, queries per second (QPS) and GPU cost in industry-scale recommendation systems. Existing models do not adequately address these industrial scalability issues. In this paper, we propose a novel two-stage modeling framework, namely VIrtual Sequential Target Attention (VISTA), which decomposes traditional target attention from a candidate item to user history items into two distinct stages: (1) user history summarization into a few hundred tokens; followed by (2) candidate item attention to those tokens. These summarization token embeddings are then cached in storage system and then utilized as sequence features for downstream model training and inference. This novel design for scalability enables VISTA to scale to lifelong user histories (up to one million items) while keeping downstream training and inference costs fixed, which is essential in industry. Our approach achieves significant improvements in offline and online metrics and has been successfully deployed on an industry leading recommendation platform serving billions of users.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。