Journal of Computer Applications ›› 2024, Vol. 44 ›› Issue (8): 2357-2364.DOI: 10.11772/j.issn.1001-9081.2023081063

• Artificial intelligence • Previous Articles     Next Articles

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

Tingjie TANG1,2, Jiajin HUANG3(), Jin QIN1,2, Hui LU1,2   

  1. 1.State Key Laboratory of Public Big Data (Guizhou University),Guiyang Guizhou 550025,China
    2.College of Computer Science and Technology,Guizhou University,Guiyang Guizhou 550025,China
    3.Faculty of Information Technology,Beijing University of Technology,Beijing 100124,China
  • Received:2023-08-07 Revised:2023-10-10 Accepted:2023-10-17 Online:2023-12-18 Published:2024-08-10
  • Contact: Jiajin HUANG
  • About author:TANG Tingjie , born in 1999, M. S. candidate. His researchinterests include recommender system.
    HUANG Jiajin , born in 1977, Ph. D., assistant research fellow.His research interests include recommender system.
    QIN Jin , born in 1978, Ph. D., associate professor. His researchinterests include reinforcement learning, computational intelligence.
    LU Hui, born in 1998, M. S. candidate. His research interestsinclude dynamic clustering.
  • Supported by:
    This work is partially supported by Guizhou Science and TechnologyPlan Project( Qiankehe Fund[2020]1Y275).

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

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

  1. 1.公共大数据国家重点实验室(贵州大学), 贵阳 550025
    2.贵州大学 计算机科学与技术学院, 贵阳 550025
    3.北京工业大学 信息学部, 北京 100124
  • 通讯作者: 黄佳进
  • 作者简介:唐廷杰(1999—),男,贵州黔东南人,硕士研究生,主要研究方向:推荐系统
    黄佳进(1977—),男,贵州遵义人,助理研究员,博士,主要研究方向:推荐系统 jhuang@bjut.edu.cn
    秦进(1978—),男,贵州毕节人,副教授,博士,主要研究方向:强化学习、计算智能
    陆辉(1998—),男,贵州安顺人,硕士研究生,主要研究方向:动态聚类。
  • 基金资助:
    贵州省科技计划项目(黔科合基础[2020]1Y275)

Abstract:

Aiming at the problem that the Multi-Layer Perceptron (MLP) architecture can not capture the co-occurrence relationship in the context of session sequence, a session-based recommendation model based on Graph Co-occurrence Enhanced MLP (GCE-MLP) was proposed. Firstly, the sequential dependency of the session sequence was captured by the MLP 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. Secondly, 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. Experimental results on multiple real datasets show that the recommendation performance of the GCE-MLP is better than those of the current mainstream models, which verifies the effectiveness of GCE-MLP. Compared with the optimal MLP architecture model FMLP-Rec(Filter-enhanced MLP for Recommendation), GCE-MLP achieves the P@20 of 54.08% and the MRR@20 of 18.87% for Diginetica dataset, which are respectively increased by about 2.14 and 1.43 percentage points; GCE-MLP achieves the P@20 of 71.77% and the MRR@20 of 31.78% for Yoochoose dataset, which are respectively increased by about 0.48 and 1.77 percentage points.

Key words: recommendation system, session-based recommendation, Multi-Layer Perceptron (MLP), co-occurrence relationship, representation learning

摘要:

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

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

CLC Number: