arXiv:2501.06985cs.IRcs.AI2025-01被引 6

用对比学习提升多标签推荐效果,解决数据量大时的性能瓶颈。

Graph Contrastive Learning on Multi-label Classification for Recommendations

  • 分主任务与子任务:整体图学习+同类型节点子图建模
  • 在Amazon Reviews数据集上多标签分类准确率优于现有方法
  • 适合大规模推荐系统优化,尤其关注用户/物品间复杂关系建模

在商业分析中,提供有效的推荐对提升公司利润至关重要。基于图结构(如二分图)的方法因其能分析复杂数据关系而受到关注,链接预测对向用户推荐特定项目尤为关键。传统方法通常依赖图结构模式识别或图神经网络等表征技术,但随着数据量增加,性能逐渐下降。为此,我们提出一种名为多标签分类图对比学习(MCGCL)的模型。MCGCL利用对比学习提升推荐效果,包含两个训练阶段:主任务为整体用户-物品图学习,以捕捉用户-物品关系;子任务构建同类型节点子图(用户-用户、物品-物品),用于捕获同类节点间关系。我们在Amazon Reviews的真实数据集上进行了多标签分类任务评估。与当前先进方法的对比实验表明,MCGCL显著有效,展现了其在改进推荐系统方面的潜力。

原文摘要 · Abstract (English)

In business analysis, providing effective recommendations is essential for enhancing company profits. The utilization of graph-based structures, such as bipartite graphs, has gained popularity for their ability to analyze complex data relationships. Link prediction is crucial for recommending specific items to users. Traditional methods in this area often involve identifying patterns in the graph structure or using representational techniques like graph neural networks (GNNs). However, these approaches encounter difficulties as the volume of data increases. To address these challenges, we propose a model called Graph Contrastive Learning for Multi-label Classification (MCGCL). MCGCL leverages contrastive learning to enhance recommendation effectiveness. The model incorporates two training stages: a main task and a subtask. The main task is holistic user-item graph learning to capture user-item relationships. The homogeneous user-user (item-item) subgraph is constructed to capture user-user and item-item relationships in the subtask. We assessed the performance using real-world datasets from Amazon Reviews in multi-label classification tasks. Comparative experiments with state-of-the-art methods confirm the effectiveness of MCGCL, highlighting its potential for improving recommendation systems.

图对比学习多标签推荐GNN优化

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