摘要
本申请公开了一种基于注意力机制和图神经网络的科研合作者推荐方法,通过提出一种基于图神经网络和注意力机制的学术合作者推荐模型AGNNRec。该方法基于图神经网络构建推荐模型,引入上下文路径捕捉节点间高阶关系,并引入多头关系注意力机制来并行学习节点高阶邻居的嵌入向量,最后通过使用余弦相似度算法完成合作者的推荐。模型分别在自建数据集和公开数据集DBLP上进行了实验,在使用DBLP数据集的实验中,模型在Precision@5和NDGG@5上的效果分别提升了19.07%和1.17%,展现了AGNNRec在科技合作者推荐任务中的有效性。
技术关键词
科研合作者推荐方法
异质信息网络
注意力机制
节点
关系
神经网络模型
邻居
高维特征向量
语义
论文
计算机程序产品
项目
会议
处理器
计算机设备
输出模块
存储模块
列表
存储器