首页 | 本学科首页   官方微博 | 高级检索  
     

基于混合采样的图对比学习推荐算法
引用本文:袁琮淇,刘渊,刘静文.基于混合采样的图对比学习推荐算法[J].计算机应用研究,2023,40(5):1346-1351.
作者姓名:袁琮淇  刘渊  刘静文
作者单位:江南大学,人工智能与计算机学院,江南大学,人工智能与计算机学院;江苏省媒体设计与软件技术重点实验室,江南大学,人工智能与计算机学院
基金项目:国家自然科学基金资助项目(61972182)
摘    要:在推荐系统领域中,图卷积网络具有对于图结构数据更强的信息抽取能力。然而,现有的图卷积网络推荐算法主要关注改进模型结构,忽视了提高原始样本采样质量与挖掘用户—项目间隐式关系的重要性。针对上述问题,提出一种基于混合采样的图对比学习推荐算法。首先使用混合采样方法,提取出正样本中部分信息并将其注入负样本,从而生成全新的富含信息的难负样本;其次,通过轻量图卷积网络对难负样本进行特征提取,得到用户和项目的节点表征,采用邻域对比学习方法挖掘样本隐式关系;最后,利用多任务策略对推荐监督任务和对比学习任务进行联合优化。在真实数据集Yelp2018和Amazon-book上进行实验,采用recall和NDCG指标进行评估,实验结果表明,提出的模型相较其他基准模型取得了更好的效果。

关 键 词:图卷积网络  推荐系统  难负样本  图对比学习
收稿时间:2022/10/21 0:00:00
修稿时间:2023/4/12 0:00:00

Graph contrastive learning recommendation algorithm based on mixed sampling
Yuan Congqi,Liu Yuan and Liu Jingwen.Graph contrastive learning recommendation algorithm based on mixed sampling[J].Application Research of Computers,2023,40(5):1346-1351.
Authors:Yuan Congqi  Liu Yuan and Liu Jingwen
Affiliation:School of Artificial Intelligence & Computer Science, Jiangnan University,,
Abstract:In recommender systems, graph convolutional networks have stronger information extraction capabilities for graph-structured data. However, existing graph convolutional networks mainly focus on enhancing the model structure, ignoring the importance of improving the sampling quality of the original samples and mining the implicit relationship between users and items. Aiming at the above problems, this paper proposed a graph contrastive learning recommendation algorithm based on mixed sampling. Firstly, the algorithm used a mixed sampling method to extract part of the information in positive samples and injected them into negative samples, thereby generating new informative hard negative samples. Secondly, to extract features on hard negative samples, it used the light graph convolution network to obtain node representations of users and items. Finally, it carried out a multi-task strategy to jointly optimize the recommendation supervision task and the contrastive learning task. The experiments on real datasets Yelp2018 and Amazon-book demonstrate that the proposed algorithm improves performance compared with other recommendation algorithms in recall and NDCG evaluation indexes.
Keywords:graph convolutional network  recommender systems  hard negative samples  graph contrastive learning
点击此处可从《计算机应用研究》浏览原始摘要信息
点击此处可从《计算机应用研究》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号