Efficient Graph Collaborative Filtering via Contrastive Learning

被引:5
作者
Pan, Zhiqiang [1 ]
Chen, Honghui [1 ]
机构
[1] Natl Univ Def Technol, Sci & Technol Informat Syst Engn Lab, Changsha 410073, Peoples R China
关键词
recommender systems; efficient recommendation; collaborative filtering; graph convolution networks; contrastive learning;
D O I
10.3390/s21144666
中图分类号
O65 [分析化学];
学科分类号
070302 ; 081704 ;
摘要
Collaborative filtering (CF) aims to make recommendations for users by detecting user's preference from the historical user-item interactions. Existing graph neural networks (GNN) based methods achieve satisfactory performance by exploiting the high-order connectivity between users and items, however they suffer from the poor training efficiency problem and easily introduce bias for information propagation. Moreover, the widely applied Bayesian personalized ranking (BPR) loss is insufficient to provide supervision signals for training due to the extremely sparse observed interactions. To deal with the above issues, we propose the Efficient Graph Collaborative Filtering (EGCF) method. Specifically, EGCF adopts merely one-layer graph convolution to model the collaborative signal for users and items from the first-order neighbors in the user-item interactions. Moreover, we introduce contrastive learning to enhance the representation learning of users and items by deriving the self-supervisions, which is jointly trained with the supervised learning. Extensive experiments are conducted on two benchmark datasets, i.e., Yelp2018 and Amazon-book, and the experimental results demonstrate that EGCF can achieve the state-of-the-art performance in terms of Recall and normalized discounted cumulative gain (NDCG), especially on ranking the target items at right positions. In addition, EGCF shows obvious advantages in the training efficiency compared with the competitive baselines, making it practicable for potential applications.
引用
收藏
页数:17
相关论文
共 40 条
[11]  
Kabbur S, 2013, 19TH ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY AND DATA MINING (KDD'13), P659
[12]   Momentum Contrast for Unsupervised Visual Representation Learning [J].
He, Kaiming ;
Fan, Haoqi ;
Wu, Yuxin ;
Xie, Saining ;
Girshick, Ross .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2020), 2020, :9726-9735
[13]   Modeling of Recommendation System Based on Emotional Information and Collaborative Filtering [J].
Kim, Tae-Yeun ;
Ko, Hoon ;
Kim, Sung-Hwan ;
Kim, Ho-Da .
SENSORS, 2021, 21 (06) :1-25
[14]  
King DB, 2015, ACS SYM SER, V1214, P1, DOI 10.1021/bk-2015-1214.ch001
[15]  
Kipf T. N., 2017, INT C LEARN REPR ICL, DOI [10.1051/0004-6361/201527329, DOI 10.48550/ARXIV.1609.02907]
[16]  
Koren Y., 2008, ACM C KNOWLEDGE DISC, P426, DOI [10.1145/1401890.1401944, DOI 10.1145/1401890.1401944]
[17]   A Personalized QoS Prediction Approach for CPS Service Recommendation Based on Reputation and Location-Aware Collaborative Filtering [J].
Kuang, Li ;
Yu, Long ;
Huang, Lan ;
Wang, Yin ;
Ma, Pengju ;
Li, Chuanbin ;
Zhu, Yujia .
SENSORS, 2018, 18 (05)
[18]  
Li Y., 2018, 180703748 ARXIV
[19]   Variational Autoencoders for Collaborative Filtering [J].
Liang, Dawen ;
Krishnan, Rahul G. ;
Hoffman, Matthew D. ;
Jebara, Tony .
WEB CONFERENCE 2018: PROCEEDINGS OF THE WORLD WIDE WEB CONFERENCE (WWW2018), 2018, :689-698
[20]  
Logeswaran L., 2018, P INT C LEARN REPR I