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, INT C MACH LEARN, DOI DOI 10.22489/CINC.2017.065-469
[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, P 13 INT C ART INT S, 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 [J].
He, Xiangnan ;
Deng, Kuan ;
Wang, Xiang ;
Li, Yan ;
Zhang, Yongdong ;
Wang, Meng .
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 [J].
Jaiswal, Ashish ;
Babu, Ashwin Ramesh ;
Zadeh, Mohammad Zaki ;
Banerjee, Debapriya ;
Makedon, Fillia .
TECHNOLOGIES, 2021, 9 (01)
[9]  
Khosla P, 2020, ADV NEUR IN, V33
[10]   Prototypical Graph Contrastive Learning [J].
Lin, Shuai ;
Liu, Chen ;
Zhou, Pan ;
Hu, Zi-Yuan ;
Wang, Shuojia ;
Zhao, Ruihui ;
Zheng, Yefeng ;
Lin, Liang ;
Xing, Eric ;
Liang, Xiaodan .
IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2024, 35 (02) :2747-2758