将语义信息融入推荐模型,提升冷启动场景下的推荐准确率。
SISA-Rec: A Semantically Integrated Sequential Recommender with Contrastive Alignment

- 用门控融合模块结合物品ID与BERT文本嵌入,增强语义表征。
- 在亚马逊美妆和玩具数据集上,点击率和排名指标均显著优于基线。
- 特别适合历史交互数据少的冷启动用户,对新用户更友好。
推荐系统帮助用户从海量选项中发现相关物品。现有基于Transformer的序列推荐主要依赖物品编号,未能充分利用物品的语义信息,导致在稀疏和冷启动场景下表现不佳。为此,本文提出SISA-Rec(语义集成序列推荐),一种将语义上下文直接融入序列建模的Transformer框架。该方法通过门控融合模块结合物品ID嵌入与BERT文本嵌入,将语义相似性注入自注意力机制,并采用基于注意力的聚合模块构建全面的用户表征。最后,联合使用贝叶斯个性化排序(BPR)与对比对齐损失,实现行为空间与语义空间的一致性对齐。实验在两个高度稀疏的亚马逊数据集(美妆和玩具与游戏)上进行,两者均具有99.93%的稀疏度。结果表明,SISA-Rec在所有评估指标上均优于当前最优基线模型。相较于BERT4Rec,SISA-Rec在美妆数据集上将HR@10提升16.6%,NDCG@10提升10.3%;在玩具与游戏数据集上分别提升23.1%和17.9%。冷启动分析显示,该模型在交互记录极少的用户上性能提升最大,验证了语义信息在行为数据匮乏时的关键价值。整体结果表明,将语义信息整合至注意力机制能显著提升推荐的准确性与可靠性。
原文摘要 · Abstract (English)
Recommendation systems help users recommend relevant items from a large collection of choices. Present work on transformer-based sequential recommendation learns user preferences from interaction logs, but it mostly focuses on item identifiers and doesn't fully use the semantic meaning of items. This limitation becomes a major challenge in sparse and cold-start scenarios where historical interaction data is limited. To solve this problem, we introduce SISA-Rec (Semantically Integrated Sequential Recommendation), a transformer-based framework that embeds semantic context directly into sequential modeling. Our approach fuses item ID embeddings with BERT-based text embeddings via a gated fusion module, injects semantic similarity into the self-attention mechanism, and leverages an attention-based aggregation module to construct comprehensive user representations. Finally, a joint learning objective which combines Bayesian Personalized Ranking (BPR) and contrastive alignment loss, aligns the underlying behavioral and semantic spaces. Experiments were conducted on the two highly sparse Amazon Beauty and Amazon Toys \& Games datasets, both having 99.93\% sparsity. The results show that SISA-Rec outperforms state-of-the-art baseline models across all evaluation metrics. Compared with the BERT4Rec \cite{petrov2022systematic}, SISA-Rec improves HR@10 by 16.6\% and NDCG@10 by 10.3\% on Amazon Beauty, and HR@10 by 23.1\% and NDCG@10 by 17.9\% on Amazon Toys \& Games. Cold-start analysis further shows that the proposed model achieves the largest improvements for users with limited interaction historical records. This showcases the value of semantic information when user behavior data is scarce. Overall, the results demonstrate that integrating semantic information into the attention mechanism leads to more accurate and reliable recommendations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。