将推荐系统Transformer扩展至十亿参数,显著提升用户听歌时长和喜爱率。
Scaling Recommender Transformers to One Billion Parameters
- 将用户行为建模为自回归生成任务,分解为反馈预测与下一物品预测
- 训练出最大达10亿参数的Transformer推荐模型,在真实平台验证效果
- 在线测试显示听歌时长提升2.26%,用户喜欢概率提高6.37%
尽管大规模Transformer已在自然语言处理、计算机视觉等领域广泛应用,但将其扩展到推荐系统仍面临挑战。近期生成式推荐框架通过将推荐建模为序列转换任务,改善了计算可扩展性,但现有最大编码器仅约1.76亿参数,远小于现代语言模型的数百亿甚至万亿参数规模。本文提出一种训练超大Transformer推荐模型的方法,实现高达十亿参数的规模。我们发现用户历史的自回归学习可自然分解为反馈预测与下一物品预测两个子任务,且该分解在不同规模的Transformer上均具有效扩展性。此外,我们在一个大型音乐平台部署该模型,线上A/B测试显示,总听歌时长提升2.26%,用户喜爱概率提高6.37%,据我们所知,这是该平台历史上基于深度学习系统的最大推荐质量提升。
原文摘要 · Abstract (English)
While large transformer models have been successfully used in many real-world applications such as natural language processing, computer vision, and speech processing, scaling transformers for recommender systems remains a challenging problem. Recently, Generative Recommenders framework was proposed to scale beyond typical Deep Learning Recommendation Models (DLRMs). Reformulation of recommendation as sequential transduction task led to improvement of scaling properties in terms of compute. Nevertheless, the largest encoder configuration reported by the HSTU authors amounts only to ~176 million parameters, which is considerably smaller than the hundreds of billions or even trillions of parameters common in modern language models. In this work, we present a recipe for training large transformer recommenders with up to a billion parameters. We show that autoregressive learning on user histories naturally decomposes into two subtasks, feedback prediction and next-item prediction, and demonstrate that such a decomposition scales effectively across a wide range of transformer sizes. Furthermore, we report a successful deployment of our proposed architecture on a large-scale music platform serving millions of users. According to our online A/B tests, this new model increases total listening time by +2.26% and raises the likelihood of user likes by +6.37%, constituting (to our knowledge) the largest improvement in recommendation quality reported for any deep learning-based system in the platform's history.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。