论文标题

通过富含邻里的对比度学习改善图形协作过滤

Improving Graph Collaborative Filtering with Neighborhood-enriched Contrastive Learning

论文作者

Lin, Zihan, Tian, Changxin, Hou, Yupeng, Zhao, Wayne Xin

论文摘要

最近,已经提出了图形协作过滤方法作为一种有效的建议方法,该方法可以通过对用户项目交互图进行建模来捕获用户对项目的偏好。为了减少数据稀疏性的影响,在图形协作过滤中采用了对比度学习以增强性能。但是,这些方法通常通过随机抽样来构建对比对,从而忽略了用户(或项目)之间的邻近关系,并且无法完全利用对比度学习的潜力进行建议。为了解决上述问题,我们提出了一种新颖的对比学习方法,名为NCL,名为NCL,该方法将潜在的邻居明确地纳入对比对比对。具体而言,我们分别从图形结构和语义空间介绍了用户(或项目)的邻居。对于相互作用图上的结构邻居,我们开发了一个新颖的结构对抗性目标,该目标将用户(或项目)及其结构邻居视为积极的对比对。在实施中,用户(或项目)和邻居的表示形式对应于不同GNN层的输出。此外,为了挖掘语义空间中的潜在邻居关系,我们假设具有相似表示的用户在语义邻域内,并将这些语义邻居纳入原型对抗性目标。提出的NCL可以通过EM算法优化,并概括地应用于图形协作过滤方法。在五个公共数据集上进行的大量实验证明了拟议的NCL的有效性,尤其是在Yelp和Amazon-Book数据集上的竞争图协作滤波基础模型中,性能增长了26%和17%。我们的代码可在以下网址提供:https://github.com/rucaibox/ncl。

Recently, graph collaborative filtering methods have been proposed as an effective recommendation approach, which can capture users' preference over items by modeling the user-item interaction graphs. In order to reduce the influence of data sparsity, contrastive learning is adopted in graph collaborative filtering for enhancing the performance. However, these methods typically construct the contrastive pairs by random sampling, which neglect the neighboring relations among users (or items) and fail to fully exploit the potential of contrastive learning for recommendation. To tackle the above issue, we propose a novel contrastive learning approach, named Neighborhood-enriched Contrastive Learning, named NCL, which explicitly incorporates the potential neighbors into contrastive pairs. Specifically, we introduce the neighbors of a user (or an item) from graph structure and semantic space respectively. For the structural neighbors on the interaction graph, we develop a novel structure-contrastive objective that regards users (or items) and their structural neighbors as positive contrastive pairs. In implementation, the representations of users (or items) and neighbors correspond to the outputs of different GNN layers. Furthermore, to excavate the potential neighbor relation in semantic space, we assume that users with similar representations are within the semantic neighborhood, and incorporate these semantic neighbors into the prototype-contrastive objective. The proposed NCL can be optimized with EM algorithm and generalized to apply to graph collaborative filtering methods. Extensive experiments on five public datasets demonstrate the effectiveness of the proposed NCL, notably with 26% and 17% performance gain over a competitive graph collaborative filtering base model on the Yelp and Amazon-book datasets respectively. Our code is available at: https://github.com/RUCAIBox/NCL.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源