Journal of Computer Applications

    Next Articles

Session-based recommendation based on graph co-occurrence enhanced multi-layer perceptron

  

  • Received:2023-08-07 Revised:2023-10-10 Online:2023-12-18 Published:2023-12-18
  • Contact: Jia Jin 无Huang

基于图共现增强多层感知机的会话推荐

唐廷杰1,黄佳进2,秦进3,陆辉4   

  1. 1. 公共大数据国家重点实验室,贵州大学计算机科学与技术学院
    2. 北京工业大学 国际WIC研究院
    3. 贵州大学
    4. 贵州大学计算机科学与技术学院
  • 通讯作者: 黄佳进
  • 基金资助:
    贵州省科技计划项目

Abstract: Abstract: Aiming at the problem that the multi-layer perceptron (MLP) architecture could not capture the co-occurrence relationship in the context of session sequence, a graph co-occurrence enhanced MLP for session-based recommendation (GCE-MLP) model was proposed. Firstly, the sequential dependency of the session sequence was captured by the multi-layer perceptron architecture, and at the same time, the co-occurrence relationship in the sequence context was obtained through the co-occurrence relationship learning layer, and the session representation was obtained through the information fusion module. Second, a specific feature selection layer was designed to amplify the diversity of input features of different relation learning layers. Finally, the representation learning of sessional interest was further enhanced by maximizing the mutual information between two relational representations via a noise contrastive task. Experiments on multiple real data sets show that the recommendation performance of the GCE-MLP model is better than the current mainstream models. Compared with the optimal MLP architecture model FMLP-Rec, P@20 reaches 54.08% and MRR@20 reaches 18.87% in Diginetica, which are respectively increased by about 2 percentage points and 1 percentage point; P@20 reaches 71.77% and MRR@20 reaches 31.78% in Yoochoose, respectively increased by about 0.5 percentage point and 2 percentage points, which verifies the effectiveness of the model

Key words: recommendation system, session-based recommendation, multi-layer perceptron, co-occurrence relationship, representation learning

摘要: 摘 要: 针对多层感知机(MLP)架构无法捕获会话序列上下文中的共现关系的问题,提出了一种基于图共现增强多层感知机的会话推荐(GCE-MLP)模型。首先,利用多层感知机架构捕获会话序列的顺序依赖关系,同时通过共现关系学习层获得序列上下文中的共现关系,并通过信息融合模块得到会话表示。其次,设计了特定的特征选择层,旨在扩大不同关系学习层输入特征的差异性。最后,通过噪声对比任务最大化两种关系表征之间的互信息,进一步增强对会话兴趣的表征学习。在多个真实数据集上的实验表明GCE-MLP模型的推荐性能优于目前主流的模型。与最优的MLP架构模型FMLP-Rec相比,在Diginetica数据集上,P@20最高达到了54.08%,MRR@20最高达到了18.87%,分别提升了约2个百分点和1个百分点;在Yoochoose数据集上,P@20最高达到了71.77%,MRR@20最高达到了31.78%,分别提升了约0.5个百分点和2个百分点,验证了该模型的有效性。

关键词: 关键词: 推荐系统, 会话推荐, 多层感知机, 共现关系, 表征学习