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

Multi-Task Graph Recommendation Algorithm Based on Neighborhood Sampling

作　　者：张俊三[1] 肖森 高慧[1] 邵明文[1] 张培颖[1] 朱杰[2] ZHANG Junsan;XIAO Sen;GAO Hui;SHAO Mingwen;ZHANG Peiying;ZHU Jie(College of Computer Science and Technology,China University of Petroleum(East China),Qingdao,Shandong 266580,China;Hebei Key Laboratory of Machine Learning and Computational Intelligence,College of Mathematics and Information Science,Hebei University,Baoding,Hebei 071002,China)

机构地区：[1]中国石油大学(华东)计算机科学与技术学院,山东青岛266580 [2]河北大学数学与信息科学学院河北省机器学习与计算智能重点实验室,河北保定071002

出　　处：《计算机工程与应用》2024年第9期172-180,共9页Computer Engineering and Applications

基　　金：山东省自然科学基金(ZR2022LZH015,ZR2022MF260);河北省自然科学基金(F2022511001)。

摘　　要：近年来,图神经网络(GNN)成为解决协同过滤的主流方法之一。它通过构建用户-物品图,模拟用户与物品的交互关系,并用GNN学习它们的特征表示。尽管现有在模型结构上的研究已取得了较大进展,但如何在图结构上更有效地进行负采样仍未有效解决。为此,提出一种基于邻域采样的多任务图推荐算法。该算法提出了一种基于GNN的邻域采样策略,该策略以每个用户为中心构建子图,将次高阶物品作为用户邻域采样的负样本,可以更有效地挖掘强负样本并提高采样质量。通过GNN对图结点进行信息聚合与特征提取,得到结点的最终嵌入表示。设计一种余弦边际损失来过滤部分冗余负样本,以有效减少采样过程中的噪声数据。同时,该算法引入了多任务策略对模型进行联合优化,以增强模型的泛化能力。在3个公开数据集上进行的大量实验表明,该算法在大多数情况下明显优于其他主流算法。In recent years,graph neural network(GNN)has become a mainstream method for collaborative filtering.It constructs user-item graphs to simulate interactions and utilizes GNN to learn their features.Although there have been sig�nificant advancements in model structures,effective negative sampling on graph structures remains challenging.To address this issue,a multi-task graph recommendation algorithm based on neighborhood sampling is proposed.Firstly,the algo�rithm introduces a neighborhood sampling strategy based on GNN,centering each subgraph around individual users and using higher-order items as negative samples for user neighborhood sampling.This approach effectively explores strong negative instances and enhances sampling quality.Secondly,GNN is employed to aggregate information and extract fea�tures from graph nodes,obtaining the final node embeddings.Finally,a margin loss of cosine is designed to filter redun�dant negative samples,effectively reducing noise in the sampling process.Additionally,the algorithm incorporates a multi�task strategy to jointly optimize the model,enhancing its generalization ability.Extensive experiments conducted on three public datasets demonstrate that this algorithm outperforms other mainstream methods in the majority of cases.

关 键 词：图神经网络 协同过滤 负采样 邻域采样 余弦边际损失 多任务策略 

分 类 号：TP391[自动化与计算机技术—计算机应用技术]