《计算机应用》唯一官方网站

• •    下一篇

基于全局图数据增强的对比学习神经网络推荐系统

陈龙,侯新民1,2   

  1. 1.中国科学技术大学 人工智能与大数据学院 2.合肥国家实验室
  • 收稿日期:2025-09-24 修回日期:2025-11-16 发布日期:2025-12-22 出版日期:2025-12-22
  • 通讯作者: 陈龙
  • 作者简介:陈龙(2000—),男,安徽芜湖人,硕士研究生,主要研究方向:图神经网络、机器学习、深度学习; 侯新民(1972—),男,山东郓城人,教授,博士,主要研究方向:图论及其应用、复杂网络、图神经网络。
  • 基金资助:
    国家重点研发计划项目(2023YFA1010203);国家自然科学基金资助项目(12471336);“量子通信与量子计算机”重大项目(2021ZD0302902)

Neural network recommendation system integrating contrastive learning and global graph data enhancement

CHEN Long1, HOU Xinmin1,2   

  1. 1.School of Mathematical Sciences, University of Science and Technology of China 2.Hefei Institutes of Physical Science
  • Received:2025-09-24 Revised:2025-11-16 Online:2025-12-22 Published:2025-12-22
  • About author:CHEN Long, born in 2000, M.S. candidate. His research interests include graph neural network, machine leaning, deep learning. HOU Xinmin, born in 1972, Ph. D., professor. His research interests include graph theory and its applications, complex network, graph neural network.
  • Supported by:
    National Key Research and Development Program of China (2023YFA1010203), National Natural Science Foundation of China (12471336), Major Program of "Quantum Communication and Quantum Computer" (2021ZD0302902)

摘要: 随着信息量的指数增长,推荐系统在改善用户体验和解决信息过载方面发挥了重要作用。然而,传统推荐方法在数据稀疏性、冷启动问题和可解释性方面仍存在局限。为解决这些问题,提出了一种基于全局图数据增强的图对比学习卷积神经网络推荐系统。首先,在节点、边和特征层面进行全局增强,以提升模型在稀疏数据下的鲁棒性。其次,设计了多粒度图对比学习框架,通过节点级、子图级和全图级的对比学习任务,增强了嵌入表示的一致性与判别性。在MovieLens-1M、Amazon-Books等多个公开数据集上的实验结果表明,所提模型在推荐精度、鲁棒性和冷启动场景中优于Matrix Factorization 与 LightGCN等传统方法,尤其在稀疏数据集上的表现尤为突出。通过消融实验和参数敏感性分析,验证了全局增强和多粒度对比学习机制的有效性。可见,所提模型在解决推荐系统中的典型问题,如数据稀疏和冷启动问题,具有显著优势。

关键词: 推荐系统, 图神经网络, 数据增强, 对比学习, 冷启动问题

Abstract: With the exponential growth of information, recommendation systems have played a crucial role in improving user experience and alleviating information overload. However, traditional recommendation methods still face limitations in addressing data sparsity, cold-start issues, and model interpretability. To tackle these challenges, this study proposes a graph contrastive learning convolutional neural network recommendation system based on global graph data augmentation.First, global augmentation is performed at the node, edge, and feature levels to enhance the model’s robustness under sparse data conditions. Second, a multi-granularity graph contrastive learning framework is designed, which conducts contrastive learning tasks at the node, subgraph, and global graph levels to strengthen the consistency and discriminability of the learned embeddings.Experiments conducted on several public datasets, including MovieLens-1M and Amazon-Books, demonstrate that the proposed model outperforms traditional methods such as Matrix Factorization and LightGCN in terms of recommendation accuracy, robustness, and performance in cold-start scenarios—particularly achieving superior results on sparse datasets. Furthermore, ablation studies and parameter sensitivity analyses verify the effectiveness of the global augmentation strategy and the multi-granularity contrastive learning mechanism. These findings indicate that the proposed model shows significant advantages in addressing key challenges in recommendation systems, such as data sparsity and the cold-start problem.

Key words: recommender system, graph neural network, data augmentation, contrastive learning, cold-start problem

中图分类号: