跳到主要导航 跳到搜索 跳到主要内容

Graph Contrastive Learning with Adaptive Augmentation for Recommendation

  • Mengyuan Jing
  • , Yanmin Zhu*
  • , Tianzi Zang
  • , Jiadi Yu
  • , Feilong Tang
  • *此作品的通讯作者
  • Shanghai Jiao Tong University

科研成果: 书/报告/会议事项章节会议稿件同行评审

摘要

Graph Convolutional Network (GCN) has been one of the most popular technologies in recommender systems, as it can effectively model high-order relationships. However, these methods usually suffer from two problems: sparse supervision signal and noisy interactions. To address these problems, graph contrastive learning is applied for GCN-based recommendation. The general framework of graph contrastive learning is first to perform data augmentation on the input graph to get two graph views and then maximize the agreement of representations in these views. Despite the effectiveness, existing methods ignore the differences in the impact of nodes and edges when performing data augmentation, which will degrade the quality of the learned representations. Meanwhile, they usually adopt manual data augmentation schemes, limiting the generalization of models. We argue that the data augmentation scheme should be learnable and adaptive to the inherent patterns in the graph structure. Thus, the model can learn representations that remain invariant to perturbations of unimportant structures while demanding fewer resources. In this work, we propose a novel Graph Contrastive learning framework with Adaptive data augmentation for Recommendation (GCARec). Specifically, for adaptive augmentation, we first calculate the retaining probability of each edge based on the attention mechanism and then sample edges according to the probability with a Gumbel Softmax. In addition, the adaptive data augmentation scheme is based on the neural network and requires no domain knowledge, making it learnable and generalizable. Extensive experiments on three real-world datasets show that GCARec outperforms state-of-the-art baselines.

源语言英语
主期刊名Machine Learning and Knowledge Discovery in Databases - European Conference, ECML PKDD 2022, Proceedings
编辑Massih-Reza Amini, Stéphane Canu, Asja Fischer, Tias Guns, Petra Kralj Novak, Grigorios Tsoumakas
出版商Springer Science and Business Media Deutschland GmbH
590-605
页数16
ISBN(印刷版)9783031263866
DOI
出版状态已出版 - 2023
已对外发布
活动22nd Joint European Conference on Machine Learning and Principles and Practice of Knowledge Discovery in Databases, ECML PKDD 2022 - Grenoble, 法国
期限: 19 9月 202223 9月 2022

出版系列

姓名Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
13713 LNAI
ISSN(印刷版)0302-9743
ISSN(电子版)1611-3349

会议

会议22nd Joint European Conference on Machine Learning and Principles and Practice of Knowledge Discovery in Databases, ECML PKDD 2022
国家/地区法国
Grenoble
时期19/09/2223/09/22

指纹

探究 'Graph Contrastive Learning with Adaptive Augmentation for Recommendation' 的科研主题。它们共同构成独一无二的指纹。

引用此