arXiv:2505.21811cs.IRcs.AI2025-05KDD被引 14

优化自注意力机制,提升跨域序列推荐效果

Revisiting Self-attention for Cross-domain Sequential Recommendation

  • 将跨域学习建模为多目标问题,动态降低跨域注意力得分
  • 在SASRec和Bert4Rec上分别提升召回率9.8%和16.0%
  • 无需调参、计算开销小,可直接嵌入现有模型

序列推荐是现代推荐系统中的主流范式,其中跨域序列推荐(CDSR)旨在基于用户在多个领域的交互行为预测未来行为。现有框架多基于自注意力变换器,并通过引入领域特定模块来改进。然而,这些额外组件忽略了变换器中已有的自注意力模块——一种天然强大的行为相关性学习工具。本文提出从增强自注意力的角度提升CDSR性能,引入帕累托最优自注意力,将跨域学习视为多目标优化问题:在优化推荐任务的同时,动态最小化跨域注意力分数。所提方法(AutoCDSR)自动实现知识迁移,既缓解负迁移,又促进辅助域间互补知识交换。进一步提出更高效的AutoCDSR+变体。该方法易于实现且可即插即用,无显著计算开销,无需复杂调参。在SASRec和Bert4Rec上,平均分别提升Recall@10 9.8%、16.0%,NDCG@10 12.0%、16.7%。代码已开源。

原文摘要 · Abstract (English)

Sequential recommendation is a popular paradigm in modern recommender systems. In particular, one challenging problem in this space is cross-domain sequential recommendation (CDSR), which aims to predict future behaviors given user interactions across multiple domains. Existing CDSR frameworks are mostly built on the self-attention transformer and seek to improve by explicitly injecting additional domain-specific components (e.g. domain-aware module blocks). While these additional components help, we argue they overlook the core self-attention module already present in the transformer, a naturally powerful tool to learn correlations among behaviors. In this work, we aim to improve the CDSR performance for simple models from a novel perspective of enhancing the self-attention. Specifically, we introduce a Pareto-optimal self-attention and formulate the cross-domain learning as a multi-objective problem, where we optimize the recommendation task while dynamically minimizing the cross-domain attention scores. Our approach automates knowledge transfer in CDSR (dubbed as AutoCDSR) -- it not only mitigates negative transfer but also encourages complementary knowledge exchange among auxiliary domains. Based on the idea, we further introduce AutoCDSR+, a more performant variant with slight additional cost. Our proposal is easy to implement and works as a plug-and-play module that can be incorporated into existing transformer-based recommenders. Besides flexibility, it is practical to deploy because it brings little extra computational overheads without heavy hyper-parameter tuning. AutoCDSR on average improves Recall@10 for SASRec and Bert4Rec by 9.8% and 16.0% and NDCG@10 by 12.0% and 16.7%, respectively. Code is available at https://github.com/snap-research/AutoCDSR.

序列推荐自注意力跨域学习推荐系统

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