公开(公告)号：CN114679372A

公开(公告)日：2022-06-28

申请号：CN202210262626.4

申请日：2022-03-17

Applicant: 扬州大学

Inventor： 刘渊 ,  赵紫娟 ,  杨凯

IPC: H04L41/04 ,  H04L41/12 ,  H04L41/14 ,  H04L41/147 ,  G06K9/62 ,  G06N3/04 ,  G06N3/08 ,  G06N5/00 ,  G06N20/20

Abstract: 本发明公开了一种基于节点相似性的图注意力网络的链路预测方法，首先构建无向异质网络模型，对节点数据进行预处理，计算网络的相似度矩阵；其次，构建基于节点相似性指标的图注意力网络模型，求出网络中所有节点的嵌入表示；再次，对已存在的边的信息进行负采样，将所有的“真边”和“假边”划分为训练集和测试集，在模型训练中使用；最后，将通过基于节点相似性的图注意力方法学习到的节点的嵌入表示以串联的方式输入到机器学习分类器中，进行链路预测。本发明通过节点相似性指标，将更多相关的节点联系再一起，使得每一个节点学习的嵌入表示更加充分地表现网络自身的结构信息，进而使得节点的嵌入表示在下游链路预测任务中取得更好的结果。