通过增强特征交互提升推荐模型规模扩展能力
FuXi-$α$: Scaling Recommendation Model with Feature Interaction Enhanced Transformer
- 分通道自适应注意力建模时序、位置与语义特征
- 多阶段前馈网络强化隐式特征交互,模型越大效果越好
- 华为音乐上线测试,播放量和时长分别提升4.76%与5.10%
受规模化定律和大语言模型启发,大规模推荐模型研究受到广泛关注。现有先进序列推荐模型主要依赖自注意力机制显式建模物品间特征交互,隐式交互则由前馈网络(FFN)处理,但此类方法对时序和位置信息的整合不足,通常通过添加注意力权重或混合潜在表示实现,限制了表达能力。近期模型HSTU进一步弱化隐式交互,制约性能。本文提出新模型FuXi-α,引入自适应多通道自注意力机制,分别建模时序、位置与语义特征,并采用多阶段FFN增强隐式特征交互。离线实验表明,模型性能随规模扩大持续提升;在线A/B测试在华为音乐应用中实现用户平均播放歌曲数提升4.76%,平均听歌时长提升5.10%。代码已开源:https://github.com/USTC-StarTeam/FuXi-alpha。
原文摘要 · Abstract (English)
Inspired by scaling laws and large language models, research on large-scale recommendation models has gained significant attention. Recent advancements have shown that expanding sequential recommendation models to large-scale recommendation models can be an effective strategy. Current state-of-the-art sequential recommendation models primarily use self-attention mechanisms for explicit feature interactions among items, while implicit interactions are managed through Feed-Forward Networks (FFNs). However, these models often inadequately integrate temporal and positional information, either by adding them to attention weights or by blending them with latent representations, which limits their expressive power. A recent model, HSTU, further reduces the focus on implicit feature interactions, constraining its performance. We propose a new model called FuXi-$α$ to address these issues. This model introduces an Adaptive Multi-channel Self-attention mechanism that distinctly models temporal, positional, and semantic features, along with a Multi-stage FFN to enhance implicit feature interactions. Our offline experiments demonstrate that our model outperforms existing models, with its performance continuously improving as the model size increases. Additionally, we conducted an online A/B test within the Huawei Music app, which showed a $4.76\%$ increase in the average number of songs played per user and a $5.10\%$ increase in the average listening duration per user. Our code has been released at https://github.com/USTC-StarTeam/FuXi-alpha.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。