GNNCL: A Graph Neural Network Recommendation Model Based on Contrastive Learning

被引:3
作者
Chen, Jinguang [1 ]
Zhou, Jiahe [1 ]
Ma, Lili [1 ]
机构
[1] Xian Polytech Univ, Sch Comp Sci, Xian 710048, Peoples R China
关键词
Collaborative filtering; Graph neural network; Recommendation system; Self supervised learning; Graph contrastive learning;
D O I
10.1007/s11063-024-11545-9
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
In the field of recommendation algorithms, the representation learning for users and items has evolved from using single IDs or historical interactions to utilizing higher-order neighbors. This can be achieved by modeling the user-item interaction graph to capture user preferences for items. Despite the promising results achieved by these algorithms, they still suffer from the issue of data sparsity. In order to mitigate the impact of data sparsity, contrastive learning has been adopted in graph collaborative filtering to enhance performance. However, current recommendation algorithms using contrastive learning yield uneven representations after data augmentation and do not consider the potential relationships among users (or items). To address these challenges, we propose a graph neural network-based recommendation model that integrates contrastive learning (GNNCL). This model combines data augmentation with added noise and the exploration of semantic neighbors for nodes. For the structural neighbors on the interaction graph, we introduce a novel and straightforward contrastive learning approach, abandoning previous graph augmentation methods, and introducing uniform noise into the embedding space to create contrastive views. To unearth potential semantic neighbor relationships in the semantic space, we assume that users with similar representations possess semantic neighbor relationships and merge these semantic neighbors into the prototype contrastive learning. We utilize a clustering algorithm to obtain prototypes for users and items and employ the EM algorithm for prototype contrastive learning. Experimental results validate the effectiveness of our approach. Particularly, on the Yelp2018 and Amazon-book datasets, our method exhibits significant performance improvements compared to basic graph collaborative filtering models.
引用
收藏
页数:13
相关论文
共 23 条
  • [1] Chen JW, 2021, Arxiv, DOI arXiv:2010.03240
  • [2] Chen Ting, 2019, 25 AMERICAS C INFORM
  • [3] Gao T., 2022, IEEE Access, V10, P56952, DOI [10.1109/ACCESS.2022.3177272, DOI 10.1109/ACCESS.2022.3177272]
  • [4] Gilmer J, 2017, PR MACH LEARN RES, V70
  • [5] Glorot X., 2010, JMLR WORKSHOP C P, P249, DOI DOI 10.1109/LGRS.2016.2565705
  • [6] Hamilton WL, 2017, ADV NEUR IN, V30
  • [7] LightGCN: Simplifying and Powering Graph Convolution Network for Recommendation
    He, Xiangnan
    Deng, Kuan
    Wang, Xiang
    Li, Yan
    Zhang, Yongdong
    Wang, Meng
    [J]. PROCEEDINGS OF THE 43RD INTERNATIONAL ACM SIGIR CONFERENCE ON RESEARCH AND DEVELOPMENT IN INFORMATION RETRIEVAL (SIGIR '20), 2020, : 639 - 648
  • [8] A Survey on Contrastive Self-Supervised Learning
    Jaiswal, Ashish
    Babu, Ashwin Ramesh
    Zadeh, Mohammad Zaki
    Banerjee, Debapriya
    Makedon, Fillia
    [J]. TECHNOLOGIES, 2021, 9 (01)
  • [9] Khosla P, 2020, ADV NEUR IN, V33
  • [10] Prototypical Graph Contrastive Learning
    Lin, Shuai
    Liu, Chen
    Zhou, Pan
    Hu, Zi-Yuan
    Wang, Shuojia
    Zhao, Ruihui
    Zheng, Yefeng
    Lin, Liang
    Xing, Eric
    Liang, Xiaodan
    [J]. IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2024, 35 (02) : 2747 - 2758