Graph contrastive learning for recommendation with generative data augmentation

被引:3
作者
Li, Xiaoge [1 ,2 ,3 ]
Wang, Yin [1 ,2 ,3 ]
Wang, Yihan [1 ,2 ,3 ]
An, Xiaochun [1 ,2 ,3 ]
机构
[1] Xian Univ Posts & Telecommun, Coll Comp Sci, 618 West Changan St, Xian, Shaanxi, Peoples R China
[2] Shaanxi Key Lab Network Data Anal & Intelligent Pr, 618 West Changan St, Xian 710121, Shaanxi, Peoples R China
[3] Xian Key Lab Big Data & Intelligent Comp, 618 West Changan St, Xian 710121, Shaanxi, Peoples R China
基金
英国科研创新办公室;
关键词
Collaborative filtering; Recommendation systems; Graph neural networks; Data augmentation; Generative models;
D O I
10.1007/s00530-024-01375-z
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Graph Neural Networks (GNNs) have been successfully adopted in recommender systems by virtue of the message-passing that implicitly captures collaborative effect. However, in practical recommendation scenarios, user-item interaction data is often sparse and exhibits a skewed distribution. To address these issues, some contrastive learning methods based on data augmentation are applied to recommender systems to enhance the representation of users and items. Nevertheless, many data enhancements solely rely on graph topology, missing crucial structural information and potentially biasing the model. In this paper, we propose a contrastive learning recommendation framework called GDA-GCL based on a generative model data augmentation strategy. Specifically, we use the Conditional Variational Autoencoder(CVAE) generative model to learn the distribution of neighbor node features conditioned on the features of the central node. Due to the randomness of resampling, we design a mirror graph comparison strategy to generate different comparison views, which introduces additional high-quality training signals into the GNN paradigm. Experimental results on three real-world public datasets demonstrate that GDA-GCL achieves significant improvement in performance over various baseline methods. Extensive analysis, including ablation studies, has demonstrated the effectiveness and robustness of our proposed generative data-augmented contrastive recommendation framework in addressing the data sparsity issue in recommendation systems.
引用
收藏
页数:13
相关论文
共 45 条
[1]   User Cold-Start Recommendation via Inductive Heterogeneous Graph Neural Network [J].
Cai, Desheng ;
Qian, Shengsheng ;
Fang, Quan ;
Hu, Jun ;
Xu, Changsheng .
ACM TRANSACTIONS ON INFORMATION SYSTEMS, 2023, 41 (03)
[2]  
Chami I, 2019, Arxiv, DOI arXiv:1910.12933
[3]   Generative Adversarial Framework for Cold-Start Item Recommendation [J].
Chen, Hao ;
Wang, Zefan ;
Huang, Feiran ;
Huang, Xiao ;
Xu, Yue ;
Lin, Yishi ;
He, Peng ;
Li, Zhoujun .
PROCEEDINGS OF THE 45TH INTERNATIONAL ACM SIGIR CONFERENCE ON RESEARCH AND DEVELOPMENT IN INFORMATION RETRIEVAL (SIGIR '22), 2022, :2565-2571
[4]   GNNCL: A Graph Neural Network Recommendation Model Based on Contrastive Learning [J].
Chen, Jinguang ;
Zhou, Jiahe ;
Ma, Lili .
NEURAL PROCESSING LETTERS, 2024, 56 (02)
[5]  
Chen T, 2020, Arxiv, DOI arXiv:1905.04579
[6]   Deep Neural Networks for YouTube Recommendations [J].
Covington, Paul ;
Adams, Jay ;
Sargin, Emre .
PROCEEDINGS OF THE 10TH ACM CONFERENCE ON RECOMMENDER SYSTEMS (RECSYS'16), 2016, :191-198
[7]  
Feng WZ, 2020, ADV NEUR IN, V33
[8]   Large-Scale Learnable Graph Convolutional Networks [J].
Gao, Hongyang ;
Wang, Zhengyang ;
Ji, Shuiwang .
KDD'18: PROCEEDINGS OF THE 24TH ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY & DATA MINING, 2018, :1416-1424
[9]  
Hamilton WL, 2017, ADV NEUR IN, V30
[10]   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