arXiv:2505.19020cs.IRcs.AI2025-05被引 2

通过构建层次化图结构提升推荐模型准确率

HGCL: Hierarchical Graph Contrastive Learning for User-Item Recommendation

  • 设计跨层对比学习预训练模块提取用户物品表征
  • 在70K至382K节点数据集上超越基线模型
  • 适合关注物品层级关系的推荐系统研究者

图对比学习(GCL)融合图神经网络与对比学习,已成为用户-物品推荐的关键技术。然而现有方法通常缺乏对物品层次结构的显式建模,而这类结构在在线商品和本地商户等场景中普遍存在,反映了物品固有的组织特性,是提升推荐精度的重要信号。本文提出层次图对比学习(HGCL),一种引入物品层次结构的新型GCL方法。首先,通过跨层对比学习预训练图神经网络模块,获得用户和物品的初始表征;其次,采用表征压缩与聚类方法构建双层用户-物品二部图;最后,在层次图上微调用户和物品表征,并基于交互得分生成推荐。在三个涵盖70K至382K节点的基准数据集上的实验表明,HGCL显著优于现有基线模型,验证了层次物品结构在增强推荐任务中的贡献。

原文摘要 · Abstract (English)

Graph Contrastive Learning (GCL), which fuses graph neural networks with contrastive learning, has evolved as a pivotal tool in user-item recommendations. While promising, existing GCL methods often lack explicit modeling of hierarchical item structures, which represent item similarities across varying resolutions. Such hierarchical item structures are ubiquitous in various items (e.g., online products and local businesses), and reflect their inherent organizational properties that serve as critical signals for enhancing recommendation accuracy. In this paper, we propose Hierarchical Graph Contrastive Learning (HGCL), a novel GCL method that incorporates hierarchical item structures for user-item recommendations. First, HGCL pre-trains a GCL module using cross-layer contrastive learning to obtain user and item representations. Second, HGCL employs a representation compression and clustering method to construct a two-hierarchy user-item bipartite graph. Ultimately, HGCL fine-tunes user and item representations by learning on the hierarchical graph, and then provides recommendations based on user-item interaction scores. Experiments on three widely adopted benchmark datasets ranging from 70K to 382K nodes confirm the superior performance of HGCL over existing baseline models, highlighting the contribution of hierarchical item structures in enhancing GCL methods for recommendation tasks.

图对比学习推荐系统层次结构

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