基于多图神经网络的会话感知推荐模型

Multi-graph neural network-based session perception recommendation model

针对基于会话的推荐算法主要依赖目标会话中的信息,而未充分利用其他会话中的协同信息的问题,提出了一种基于多图神经网络的会话感知推荐(MGSP)模型。首先,根据目标会话与训练集中的所有会话构建物品转移图(ITG)和协同关联图(CRG),并基于这两张图应用图神经网络(GNN)来汇聚节点的信息,得到两类的节点表示;然后,经过双层注意力模块对两类节点表示建模,获取会话级别的表示;最后,使用注意力机制进行信息融合,得到最终的会话表示,并预测下一个交互物品。分别在电商和民航两个场景下进行了对比实验,实验结果表明,相较最优的基准模型,MGSP模型在电商数据集各项指标上的提高超过1个百分点,在民航数据集各项指标上的提高约为3个百分点,验证了MGSP模型的有效性。

The session-based recommendation algorithms mainly rely on the information from the target session,but fail to fully utilize the collaborative information from other sessions.In order to solve this problem,a Multi-Graph neural networkbased Session Perception recommendation(MGSP)model was proposed.Firstly,according to the target session and all sessions in the training set,Item-Transition Graph(ITG)and Collaborative Relation Graph(CRG)were constructed.Based on these two graphs,the Graph Neural Network(GNN)was applied to aggregate the information of the nodes in order to obtain two types of node representations.Then,after the two-layer attention module modelling two type node representations,the session-level representation was obtained.Finally,by using the attention mechanism to fuse the information,the ultimate session representation was gained,and the next interaction item was predicted.The comparison experiments were carried out in two scenarios of e-commerce and civil aviation.Experimental results show that,the proposed algorithm is superior to the optimal benchmark model,with an increase of more than 1 percentage point and 3 percentage point in the indicators on the e-commerce and civil aviation datasets respectively,verifying the effectiveness of the proposed model.