计算机科学与探索 ›› 2025, Vol. 19 ›› Issue (2): 502-512.DOI: 10.3778/j.issn.1673-9418.2401029

• 人工智能·模式识别 • 上一篇    下一篇

结合图同构和混合阶残差门控图神经网络的会话推荐

王永贵,于琦   

  1. 辽宁工程技术大学 电子与信息工程学院,辽宁 葫芦岛 125105
  • 出版日期:2025-02-01 发布日期:2025-01-23

Graph Isomorphism and Hybrid-Order Residual Gated Graph Neural Network for Session-Based Recommendation

WANG Yonggui, YU Qi   

  1. School of Electronic and Information Engineering, Liaoning Technical University, Huludao, Liaoning 125105, China
  • Online:2025-02-01 Published:2025-01-23

摘要: 基于会话推荐的目的是依据当前会话的先前动作来预测用户的下一个动作。针对现有基于图神经网络的会话推荐模型存在的不足之处,提出一种结合图同构和混合阶残差门控图神经网络的会话推荐模型(GIHR-GNN)。使用图同构网络聚合相邻项目的特征向量,有效融合全局和局部信息,解决图神经网络善于捕获节点之间的局部连接而忽略全局信息的问题,并通过门控融合函数聚合用户的长短期兴趣以更好地捕捉用户兴趣的动态变化。使用混合阶门控图神经网络对位置嵌入向量进行处理以捕获用户长时间后重新交互所反映出的用户意图,并在此基础之上添加残差模块,解决深层网络的退化问题。将未去噪和去噪后的用户长期兴趣表示进行对比学习,缓解了数据稀疏和噪声干扰的问题。在Tmall和RetailRocket两个数据集上进行多次实验,并与先进基线模型进行比较,结果表明该模型在Tmall数据集上P@20指标和MRR@20指标至少提升了3.26%和10.33%,在RetailRocket数据集上P@20指标和MRR@20指标至少提升了0.55%和2.57%,证明了GIHR-GNN模型的有效性。

关键词: 会话推荐, 图同构网络, 混合阶残差门控图神经网络, 对比学习

Abstract: Session-based recommendation aims to predict which item will be clicked next for the current session. Aiming at the shortcomings of existing session recommendation models based on graph neural network, this paper proposes a model named graph isomorphism and hybrid-order residual gated graph neural network for session-based recommendation (GIHR-GNN). Firstly, graph isomorphic network is used to aggregate feature vectors of adjacent items, which can effectively fuse global and local information, solving the problem that graph neural network is good at capturing local connections between nodes and ignoring global information. And the user􀆳s long-term and short-term interests are aggregated by gated fusion to capture dynamic changes of user interests. Secondly, hybrid-order gated graph neural network is used to process the position embedding to capture the user intention reflected by the user􀆳s re-interaction after a long time, and the residual module is added to solve the degradation problem of deep network. Finally, contrastive learning is applied to comparing the user􀆳s long-term interest representations without denoising and after denoising, alleviating the problems of data sparsity and noise interference. Experiments are performed on Tmall and RetailRocket datasets. Compared with baseline model, GIHR-GNN increases at least 3.26% and 10.33% in P@20 and MRR@20 on Tmall, 0.55% and 2.57% in P@20 and MRR@20 on RetailRocket, which proves the effectiveness of GIHR-GNN.

Key words: session recommendation, graph isomorphic network, hybrid-order residual gated graph neural network, contrastive learning