提出BASRec方法,让推荐数据增强同时兼顾相关性与多样性。
Augmenting Sequential Recommendation with Balanced Relevance and Diversity
- 用单序列和跨序列两种方式生成新数据,平衡相关性与多样性。
- 在GRU4Rec上平均提升72.0%,SASRec上33.8%,FMLP-Rec上68.5%。
- 适合做序列推荐且关注数据质量的开发者使用。
通过生成新的有效数据,数据增强已成为缓解序列推荐中数据稀疏问题的有前景方法。现有工作多聚焦于原始数据的增强,却很少关注增强数据中相关性与多样性的失衡问题,导致语义漂移或性能提升有限。本文提出一种新的平衡型数据增强插件BASRec,可生成兼具相关性与多样性的数据。BASRec包含两个模块:单序列增强利用启发式算子的随机性,为单一用户生成多样化序列,并在表示层面融合原序列以保持相关性;同时设计重加权策略,使模型能自适应学习这两类属性。跨序列增强则从两个方向对不同序列表示进行非线性混合,生成既足够多样又保留原始关键语义的虚拟序列表示。这两个模块分别从单用户和跨用户视角帮助模型挖掘细粒度偏好知识。大量实验验证了BASRec的有效性:在GRU4Rec上平均提升72.0%,SASRec上33.8%,FMLP-Rec上68.5%。结果表明,BASRec生成的数据在相关性与多样性之间具有更优平衡。源代码已公开于https://github.com/KingGugu/BASRec。
原文摘要 · Abstract (English)
By generating new yet effective data, data augmentation has become a promising method to mitigate the data sparsity problem in sequential recommendation. Existing works focus on augmenting the original data but rarely explore the issue of imbalanced relevance and diversity for augmented data, leading to semantic drift problems or limited performance improvements. In this paper, we propose a novel Balanced data Augmentation Plugin for Sequential Recommendation (BASRec) to generate data that balance relevance and diversity. BASRec consists of two modules: Single-sequence Augmentation and Cross-sequence Augmentation. The former leverages the randomness of the heuristic operators to generate diverse sequences for a single user, after which the diverse and the original sequences are fused at the representation level to obtain relevance. Further, we devise a reweighting strategy to enable the model to learn the preferences based on the two properties adaptively. The Cross-sequence Augmentation performs nonlinear mixing between different sequence representations from two directions. It produces virtual sequence representations that are diverse enough but retain the vital semantics of the original sequences. These two modules enhance the model to discover fine-grained preferences knowledge from single-user and cross-user perspectives. Extensive experiments verify the effectiveness of BASRec. The average improvement is up to 72.0% on GRU4Rec, 33.8% on SASRec, and 68.5% on FMLP-Rec. We demonstrate that BASRec generates data with a better balance between relevance and diversity than existing methods. The source code is available at https://github.com/KingGugu/BASRec.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。