arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2501.06985cs.IRcs.AI

面向推荐系统多标签分类的图对比学习

Graph Contrastive Learning on Multi-label Classification for Recommendations

Jinan University(暨南大学);University of Illinois Chicago(伊利诺伊大学芝加哥分校)

Jiayang Wu, Wensheng Gan, Huashen Lu, Philip S. Yu

AI总结:

针对传统图推荐方法随数据量增长性能下降的问题,提出含主、子双训练阶段的多标签分类图对比学习模型MCGCL,在真实数据集上优于现有最优方法,可提升推荐系统效果。

摘要:

在商业分析中,提供有效推荐对提升企业利润至关重要。基于图的结构(如二分图)因具备分析复杂数据关系的能力而得到广泛应用,链路预测则是为用户推荐特定物品的核心环节。该领域的传统方法通常依赖识别图结构中的模式,或使用图神经网络(GNNs)等表征技术,但随着数据量增长,这些方法会面临性能瓶颈。为应对上述挑战,本文提出名为Graph Contrastive Learning for Multi-label Classification(多标签分类图对比学习,MCGCL)的模型,通过对比学习提升推荐效果。该模型包含两个训练阶段:主任务与子任务。主任务执行全局用户-物品图学习,以捕捉用户与物品间的关联;子任务则构建同构的用户-用户(物品-物品)子图,挖掘用户间及物品间的关系。本文采用Amazon Reviews的真实世界数据集,在多标签分类任务上评估模型性能,与当前最优方法的对比实验验证了MCGCL的有效性,凸显其在改进推荐系统方面的潜力。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑