arXiv:2411.01376cs.LGcs.IR2024-11被引 5

用多通道超图对比学习提升推荐系统矩阵补全效果

Multi-Channel Hypergraph Contrastive Learning for Matrix Completion

  • 构建多通道超图捕捉用户与物品的高阶关联
  • 通过跨评级对比学习增强不同评分间的相互作用
  • 在五个数据集上超越现有最优方法,尤其在稀疏场景下表现优异

评分是用户显式反馈的典型形式,矩阵补全本质上是评分预测过程,也是推荐系统中的关键问题。近年来,图神经网络(GNN)被广泛用于矩阵补全,通过将评分矩阵建模为二分图来捕捉用户对物品的偏好。然而,现有方法在真实场景中受数据稀疏性和长尾分布影响,且 GNN 的消息传递机制难以捕捉节点间的高阶相关性与约束,而这些在推荐任务中至关重要。为此,我们提出多通道超图对比学习框架 MHCL。MHCL 自适应学习超图结构以捕捉节点间的高阶关系,并通过基于注意力的跨视图聚合联合建模局部与全局协同关系。同时,为考虑评分的大小与顺序信息,将不同评分子图视为不同通道,鼓励相邻评分对齐,并通过多通道跨评分对比学习实现不同评分间的相互增强。在五个公开数据集上的大量实验表明,所提方法显著优于当前最先进方法。

原文摘要 · Abstract (English)

Rating is a typical user explicit feedback that visually reflects how much a user likes a related item. The (rating) matrix completion is essentially a rating prediction process, which is also a significant problem in recommender systems. Recently, graph neural networks (GNNs) have been widely used in matrix completion, which captures users' preferences over items by formulating a rating matrix as a bipartite graph. However, existing methods are susceptible due to data sparsity and long-tail distribution in real-world scenarios. Moreover, the messaging mechanism of GNNs makes it difficult to capture high-order correlations and constraints between nodes, which are essentially useful in recommendation tasks. To tackle these challenges, we propose a Multi-Channel Hypergraph Contrastive Learning framework for matrix completion, named MHCL. Specifically, MHCL adaptively learns hypergraph structures to capture high-order correlations between nodes and jointly captures local and global collaborative relationships through attention-based cross-view aggregation. Additionally, to consider the magnitude and order information of ratings, we treat different rating subgraphs as different channels, encourage alignment between adjacent ratings, and further achieve the mutual enhancement between different ratings through multi-channel cross-rating contrastive learning. Extensive experiments on five public datasets demonstrate that the proposed method significantly outperforms the current state-of-the-art approaches.

矩阵补全超图学习推荐系统对比学习

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