首页 | 本学科首页   官方微博 | 高级检索  
     

结合自监督学习的图神经网络会话推荐
引用本文:王永贵,赵晓暄.结合自监督学习的图神经网络会话推荐[J].计算机工程与应用,2023,59(3):244-252.
作者姓名:王永贵  赵晓暄
作者单位:辽宁工程技术大学 电子与信息工程学院,辽宁 葫芦岛 125105
基金项目:国家自然科学基金面上项目(61772249);
摘    要:基于会话的推荐方法由于短期用户交互数据有限,与传统推荐方法相比,其性能更容易受到数据稀疏性问题的影响。为增强会话数据以缓解数据稀疏对会话推荐性能的影响,提出一种结合自监督学习的图神经网络会话推荐(Ss-GNN)模型。构建会话图并建立基于图注意力网络的会话推荐任务来获取项目级表示和会话级表示;从会话级表示的角度出发,利用用户的一般兴趣和当前兴趣来构建辅助任务获取自监督信号;利用自监督学习实现推荐任务和辅助任务之间的互信息最大化,以增强会话数据,从而提升推荐性能。在Yoochoose和Tmall两个公开数据集上进行实验,与基线模型相比,提出的模型在Yoochoose上P@20和MRR@20至少提升了0.94%和0.79%,在Tmall上P@20和MRR@20至少提升了9.61%和4.67%,证明了Ss-GNN模型的有效性。

关 键 词:会话推荐  自监督学习  数据稀疏性  图神经网络

Self-Supervised Graph Neural Networks for Session-Based Recommendation
WANG Yonggui,ZHAO Xiaoxuan.Self-Supervised Graph Neural Networks for Session-Based Recommendation[J].Computer Engineering and Applications,2023,59(3):244-252.
Authors:WANG Yonggui  ZHAO Xiaoxuan
Affiliation:College of Electronics and Information Engineering, Liaoning Technical University, Huludao, Liaoning 125105, China
Abstract:Compared with traditional recommendation methods, session-based recommendation methods are more susceptible to the data sparsity problem due to the limited data of short-term user interaction. In order to enhance session data and alleviate the impact of datas parseness on session recommendation performance, a model named self-supervised graph neural networks for session-based recommendation(Ss-GNN) is proposed. It constructs asession graph and establishes a session recommendation task based on graph attention network to obtain the item-level and session-level representation. Then, from the perspective of the session-level representation, it uses the user’s general and current interests to construct an auxiliary task to obtain self-supervised signals.It utilizes self-supervised learning to maximize mutual information between the recommendation task and the auxiliary task to enhance session data and improve recommendation performance. Experiments are conducted on two public data sets, Yoochoose and Tmall. Compared with the baseline model, the proposed model improves at least 0.94% and 0.79% in P@20 and MRR@20 on Yoochoose, 9.61% and 4.67% in P@20 and MRR@20 on Tmall, which proves the effectiveness of Ss-GNN model.
Keywords:session recommendation  self-supervised learning  data sparsity  graph neural network  
点击此处可从《计算机工程与应用》浏览原始摘要信息
点击此处可从《计算机工程与应用》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号