计算机工程与应用››2024,第60卷››问题(9): 172-180.内政部:10.3778/j.issn.1002-8331.2308-0212

理论与研发 • 上一篇   下一篇

基于邻域采样的多任务图推荐算法

张俊三,肖森,高慧,邵明文,张培颖,朱杰  

  1. 1中国石油大学华东) 计算机科学与技术学院,山东 青岛 266580
    2河北大学 数学与信息科学学院 河北省机器学习与计算智能重点实验室,河北 保定 071002
  • 出版日期:2024-05-01 发布日期:2024-04-29

基于邻域抽样的多任务图推荐算法

张俊三、肖森、高慧、邵明文、张培英、朱洁  

  1. 1.中国石油大学(华东)计算机科学与技术学院,中国山东青岛,266580
    2.河北大学数学与信息科学学院机器学习与计算智能河北省重点实验室,河北保定,071002,中国
  • 在线:2024-05-01 出版:2024-04-29

摘要:近年来,图神经网络(GNN)成为解决协同过滤的主流方法之一。它通过构建用户-物品图,模拟用户与物品的交互关系,并用GNN(全球导航号码)学习它们的特征表示。尽管现有在模型结构上的研究已取得了较大进展,但如何在图结构上更有效地进行负采样仍未有效解决。为此,提出一种基于邻域采样的多任务图推荐算法。该算法提出了一种基于GNN(全球导航号码)的邻域采样策略,该策略以每个用户为中心构建子图,将次高阶物品作为用户邻域采样的负样本,可以更有效地挖掘强负样本并提高采样质量。通过GNN(全球导航号码)对图结点进行信息聚合与特征提取,得到结点的最终嵌入表示。设计一种余弦边际损失来过滤部分冗余负样本,以有效减少采样过程中的噪声数据。同时,该算法引入了多任务策略对模型进行联合优化,以增强模型的泛化能力。在个公开数据集上进行的大量实验表明,该算法在大多数情况下明显优于其他主流算法。

关键词: 图神经网络, 协同过滤, 负采样, 在S形符中, 余弦边际损失, 多任务策略

摘要:近年来,图神经网络(GNN)已成为协同过滤的主流方法。它构建用户-项目图来模拟交互,并利用GNN学习其特征。尽管在模型结构方面取得了重大进展,但对图结构进行有效的负采样仍然具有挑战性。为了解决这个问题,提出了一种基于邻域抽样的多任务图推荐算法。首先,该算法引入了基于GNN的邻域采样策略,将每个子图围绕单个用户进行中心化,并使用高阶项作为负样本进行用户邻域采样。这种方法有效地探索了强烈的负面情况,并提高了抽样质量。其次,利用GNN对图节点进行信息聚合和特征提取,得到最终的节点嵌入。最后,设计余弦裕度损失来过滤冗余的负样本,有效地降低了采样过程中的噪声。此外,该算法采用多任务策略联合优化模型,增强了模型的泛化能力。在三个公共数据集上进行的大量实验表明,该算法在大多数情况下都优于其他主流方法。

关键词: 图形神经网络(GNN), 协同过滤, 负采样, 邻里抽样, 余弦裕度损失, 多任务策略