arXiv:2608.07055cs.IR2026-08

用两阶段知识蒸馏实现20K长序列高效推荐,效果不降反升。

Teacher Retains Full Tokens, Student Merges Efficiently: TM20K for E-Commerce Sequence Modeling in Ad Recommendation

论文配图:Teacher Retains Full Tokens, Student Merges Efficiently: TM20K for E-Commerce Sequence Modeling in Ad Recommendation
图 1 · 摘自论文原文
  • 教师和学生均用完整注意力机制,学生通过简单合并压缩序列
  • 在20K序列长度下,广告点击率提升1.036%,延迟仅增5.6%
  • 适合需要超长行为序列建模的电商推荐系统落地

现有推荐系统通过建模超长用户行为序列,同时捕捉长期与短期兴趣,提升用户体验。然而,序列过长会显著增加训练效率和推理吞吐压力。以往方法多采用搜索或聚类压缩序列,损失细粒度信息,或依赖轻量注意力结构,难以充分提取序列特征。本文提出一种两阶段知识蒸馏框架,使教师与学生模型均使用完整注意力机制,以有效扩展序列建模能力。针对学生模型,设计多种简洁且动机明确的令牌合并策略,大幅压缩序列长度同时保持可接受性能。教师模型一次性训练,基于全序列令牌,通过知识蒸馏进一步提升学生模型表现。所提方案TM20K已成功部署于字节跳动电商广告推荐系统,将序列长度扩展至20K,关键业务指标显著提升(如ADSS +1.036%),训练与推理成本与当前线上最优模型相当(如服务延迟仅+5.6%)。

原文摘要 · Abstract (English)

Benefiting from ultra-long behavior sequence modeling, existing recommender systems bring users a better experience via simultaneously considering their long-term and short-term interests. Nevertheless, extended sequence lengths introduce substantial burdens on training efficiency and serving throughput. Prior approaches typically utilize search-based or cluster-based compression on ultra-long sequences at the cost of fine-grained information, or rely on various lightweight target attention structures incapable of sufficient sequential feature extraction. In this paper, we balance the effectiveness and efficiency for ultra-long sequence modeling via full transformer modeling accompanied with a two-stage knowledge distillation framework. First, both teacher and student models take the full attention mechanism rather than pure target-sequence attention for effective sequence scaling. For student models, we propose several simple yet well-motivated token merge approaches, significantly compressing the sequence length while maintaining an acceptable performance. Then, a one-time teacher is heavily trained with full sequence tokens, further boosting the performance of student models via knowledge distillation. The proposed paradigm named TM20K has been successfully deployed in ByteDance's e-commerce advertising recommender system that extends the e-commerce sequence length to 20K, delivering substantial improvements in key business metrics (e.g., ADSS +1.036\%) while keeping the training and serving cost nearly the same as the online state-of-the-art model (e.g., serving latency only +5.6\%).

推荐系统序列建模知识蒸馏长序列

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