联合注意力机制与卷积神经网络的知识超图链接预测方法

作者:庞俊; 徐浩; 任亮; 林晓丽; 张鸿; 徐新; 张晓龙; 李波
来源:2022-04-29, 中国, ZL202210475730.1.

摘要

本发明公开了一种联合注意力机制与卷积神经网络的知识超图链接预测方法,包括:S1、对待补全的知识超图进行加载,获得实体和关系;S2、对加载的实体和关系进行初始化,得到初始实体嵌入向量和初始关系嵌入向量;S3、将初始实体嵌入向量和初始关系嵌入向量以元组的形式输入到ACLP模型中进行训练;S4、对初始关系嵌入向量进行处理得到处理后的关系注意力向量;S5、对初始实体嵌入向量进行处理得到处理后的实体投影嵌入向量;S6、通过预设的评分模块对处理后的元组打分,得出预测结果,并判断元组的打分结果是否正确,将正确的元组添加到知识超图中,补全知识超图。本发明使得处理后的元组包含更多的信息,提升链接预测准确率。