《计算机应用》唯一官方网站

• •    下一篇

基于图辅助学习的会话推荐

唐廷杰1,2,黄佳进3,秦进1,2   

  1. 1.公共大数据国家重点实验室(贵州大学) 2.贵州大学 计算机科学与技术学院 3.北京工业大学 信息学部
  • 收稿日期:2023-09-13 修回日期:2023-12-13 发布日期:2024-03-19 出版日期:2024-03-19
  • 通讯作者: 黄佳进
  • 作者简介:唐廷杰(1999—),男,贵州镇远人,硕士研究生,主要研究方向:推荐系统;黄佳进(1977—),男,贵州遵义人,助理研究员,博士,主要研究方向:推荐系统;秦进(1978—),男,贵州毕节人,副教授,博士,主要研究方向:强化学习、计算智能。
  • 基金资助:
    贵州省科技计划项目(黔科合基础[2020]1Y275)

Session-based recommendation with graph auxiliary learning

TANG Tingjie1,2, HUANG Jiajin3, QIN Jin1,2   

  1. 1. State Key Laboratory of Public Big Data (Guizhou University) 2. College of Computer Science and Technology, Guizhou University 3. Faculty of Information Technology, Beijing University of Technology
  • Received:2023-09-13 Revised:2023-12-13 Online:2024-03-19 Published:2024-03-19
  • About author:TANG Tingjie, born in 1999, M.S. candidate. His research interests include recommender system. HUANG Jiajin, born in 1977, Ph.D., assistant researcher. His research interests include recommender system. QIN Jin, born in 1978, Ph.D., associate professor. His research interests include reinforcement learning, computational intelligence.
  • Supported by:
    Guizhou Science and Technology Plan Project (Qiankehe Fund[2020]1Y275)

摘要: 针对现有的自监督对比任务未能充分利用原始数据中的丰富语义以及缺乏通用性的问题,提出一种新的基于图辅助学习的会话推荐(SR-GAL)模型。首先,在图神经网络(GNN)的基础上引入具有表示一致性的编码通道,从原始数据中挖掘更有价值的自监督信号;其次,为了充分利用这些自监督信号,设计了与目标任务关系紧密的预测性辅助任务和约束性辅助任务;最后,开发了一个简单且与GNN模型无关的辅助学习框架,将两个辅助任务与推荐任务统一起来,以提高GNN模型的推荐性能。在多个真实数据集上的实验结果表明,SR-GAL模型优于较先进的模型,并且具有良好的可扩展性和通用性。与次优的对比模型CGSNet(Contrastive Graph Self-attention Network)相比,在Diginetica数据集上,准确率P@20和平均倒数排名MRR@20提升了0.58%和1.61%;在Tmall数据集上,P@20和MRR@20分别提升了12.65%和8.41%,验证了该模型的有效性。

关键词: 推荐系统, 会话推荐, 图神经网络, 辅助任务, 自监督学习

Abstract: Aiming at the problems that the existing self-supervised contrastive tasks failed to make full use of the rich semantics in the original data and lacked versatility, a new Session-based Recommendation with Graph Auxiliary Learning (SR-GAL) was proposed. First, an encoding channel with representation consistency was introduced on top of the Graph Neural Network (GNN) to mine more valuable self-supervised signals from raw data. Secondly, in order to make full use of these self-supervised signals, two auxiliary tasks, predictive and constraint, that were closely related to the target task were designed. Finally, a simple and GNN model-agnostic auxiliary learning framework was developed to unify the two auxiliary tasks with the recommendation task to improve the recommendation performance of the GNN model. Experimental results on multiple real datasets show that the SR-GAL model outperforms more advanced models and has good extensibility and versatility. Compared with the suboptimal comparison model CGSNet (Contrastive Graph Self-attention Network), on Diginetica dataset, the precision P@20 and the mean reciprocal ranks MRR@20 increased by 0.58% and 1.61%; on the Tmall dataset, P@20 and MRR@20 increased by 12.65% and 8.41% respectively, verifying the effectiveness of the model.

Key words: recommendation system, session-based recommendation, graph neural network, auxiliary task, self-supervised learning

中图分类号: