用动态状态表示和位置嵌入,提升跨域推荐的效率与精度
Efficient Large-Scale Cross-Domain Sequential Recommendation with Dynamic State Representations
- 引入领域转移位置嵌入,只关注本域内容,减少无效计算
- 设计动态领域状态表示,高效传递关键领域信息
- 适合大规模跨域推荐系统,尤其在多领域场景下表现优越
近期自回归推荐模型(ARMs),如Meta的HSTU模型,已成为超越传统深度学习推荐模型(DLRMs)的重大突破,展现出理想的扩展规律。然而,在多领域场景中,Transformer的注意力机制会同时关注所有领域的项目,成为计算瓶颈。为解决此问题,系统需在跨域与域内知识迁移间实现高效平衡。本文提出一种可扩展的多领域推荐新方法:用两种创新机制替代全跨域注意力。一是过渡感知位置嵌入(TAPE),通过编码领域切换信息,使注意力仅聚焦于本域项目,有效降低无关领域带来的计算开销;二是动态领域状态表示(DDSR),为每个领域维护并访问动态状态,支持高效的知识传递而无需依赖完整注意力图。该方法在检索任务中显著提升性能,通过分别建模并融合跨域与域内表示,实现了对大规模多领域推荐系统的有效优化。
原文摘要 · Abstract (English)
Recently, autoregressive recommendation models (ARMs), such as Meta's HSTU model, have emerged as a major breakthrough over traditional Deep Learning Recommendation Models (DLRMs), exhibiting the highly sought-after scaling law behaviour. However, when applied to multi-domain scenarios, the transformer architecture's attention maps become a computational bottleneck, as they attend to all items across every domain. To tackle this challenge, systems must efficiently balance inter and intra-domain knowledge transfer. In this work, we introduce a novel approach for scalable multi-domain recommendation systems by replacing full inter-domain attention with two innovative mechanisms: 1) Transition-Aware Positional Embeddings (TAPE): We propose novel positional embeddings that account for domain-transition specific information. This allows attention to be focused solely on intra-domain items, effectively reducing the unnecessary computational cost associated with attending to irrelevant domains. 2) Dynamic Domain State Representation (DDSR): We introduce a dynamic state representation for each domain, which is stored and accessed during subsequent token predictions. This enables the efficient transfer of relevant domain information without relying on full attention maps. Our method offers a scalable solution to the challenges posed by large-scale, multi-domain recommendation systems and demonstrates significant improvements in retrieval tasks by separately modelling and combining inter- and intra-domain representations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。