科研之友
微信
新浪微博
Facebook
分享链接
摘要
本发明公开了一种基于多粒度注意力网络的图立方链接预测方法,包括：对待补全的图立方进行加载,获得图立方中的实体和关系；对加载的实体和关系进行初始化,得到初始实体嵌入向量和初始关系嵌入向量；将初始实体嵌入向量和初始关系嵌入向量输入多头自注意力神经网络中,更新各多元关系的实体嵌入向量和关系嵌入向量；将更新后的实体嵌入向量和关系嵌入向量输入到HPMG模型中进行训练；通过评分模块对每个多元关系评分,根据评分指标来判断多元关系的评分结果是否正确：若正确,则将正确的多元关系添加到图立方中,补全图立方,若错误,则将错误的多元关系舍弃。本发明从多粒度的角度对多元关系进行评分,提升了链接预测的准确率。
- 单位
