arXiv:2409.05633cs.IR2024-09KDD被引 14

用离散码增强推荐图对比学习,提升协同信息保留能力。

Enhancing Graph Contrastive Learning with Reliable and Informative Augmentation for Recommendation

  • 将用户/物品映射为携带协同信息的离散码,生成更可靠的对比视图。
  • 在四个公开数据集上显著优于基线方法,提升推荐效果。
  • 适合做推荐系统中图对比学习的研究者和工程师参考。

图神经网络(GNN)因其建模高阶用户-物品关系的能力,在协同过滤(CF)中表现强劲。近期,为缓解数据稀疏性并增强表示学习,大量工作尝试将对比学习(CL)与GNN结合。然而,现有方法基于结构和表示扰动生成对比视图,可能破坏协同信息,导致正样本对齐效果有限。为此,我们提出CoGCL框架,通过离散码构建具有更强协同信息的对比视图以增强图对比学习。核心思想是将用户和物品映射到富含协同信息的离散码,实现可靠且信息丰富的对比视图生成。为此,我们首次以端到端方式引入多层级向量量化器,将用户和物品表示量化为离散码。基于这些离散码,我们分别通过考虑邻域结构和语义相关性来增强对比视图的协同信息。对于邻域结构,提出虚拟邻居增强:将离散码视为虚拟邻居,将观察到的用户-物品交互扩展为包含离散码的多条边;针对语义相关性,基于共享离散码和交互目标识别相似用户/物品,生成语义相关视图。通过这些策略,构建出协同信息更强的对比视图,并提出三视图图对比学习方法。在四个公开数据集上的广泛实验验证了所提方法的有效性。

原文摘要 · Abstract (English)

Graph neural network(GNN) has been a powerful approach in collaborative filtering(CF) due to its ability to model high-order user-item relationships. Recently, to alleviate the data sparsity and enhance representation learning, many efforts have been conducted to integrate contrastive learning(CL) with GNNs. Despite the promising improvements, the contrastive view generation based on structure and representation perturbations in existing methods potentially disrupts the collaborative information in contrastive views, resulting in limited effectiveness of positive alignment. To overcome this issue, we propose CoGCL, a novel framework that aims to enhance graph contrastive learning by constructing contrastive views with stronger collaborative information via discrete codes. The core idea is to map users and items into discrete codes rich in collaborative information for reliable and informative contrastive view generation. To this end, we initially introduce a multi-level vector quantizer in an end-to-end manner to quantize user and item representations into discrete codes. Based on these discrete codes, we enhance the collaborative information of contrastive views by considering neighborhood structure and semantic relevance respectively. For neighborhood structure, we propose virtual neighbor augmentation by treating discrete codes as virtual neighbors, which expands an observed user-item interaction into multiple edges involving discrete codes. Regarding semantic relevance, we identify similar users/items based on shared discrete codes and interaction targets to generate the semantically relevant view. Through these strategies, we construct contrastive views with stronger collaborative information and develop a triple-view graph contrastive learning approach. Extensive experiments on four public datasets demonstrate the effectiveness of our proposed approach.

图对比学习推荐系统离散编码协同过滤

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。