公开(公告)号：CN118396091A

公开(公告)日：2024-07-26

申请号：CN202410495042.0

申请日：2024-04-23

Applicant: 中国电子产品可靠性与环境试验研究所((工业和信息化部电子第五研究所)(中国赛宝实验室))

Inventor： 章婷华 ,  乐琦雯 ,  汤锦依 ,  马燕娇 ,  云雷

IPC: G06N5/022 ,  G06N5/025 ,  G06N3/0464 ,  G06N3/084

Abstract: 本发明公开了一种基于多元方向注意力机制的知识图谱补全方法，包括：对每个三元组进行初始化得到实体和关系的初始化向量，对初始化向量中的每个元素计算注意力分数，得到多元注意力机制的得分矩阵；在得分矩阵中加入方向位置掩码矩阵得到多元方向注意力机制的注意力评分函数，计算注意力评分函数的方向概率，通过将方向概率与初始化向量进行融合，得到三元组的特征矩阵；将得到的特征矩阵输入到卷积神经网络中进行深层特征提取，得到最终得分函数；根据得分函数构建损失函数，通过迭代更新卷积神经网络的参数来最小化损失函数。本发明通过挖掘知识图谱中三元组的方向信息以及细粒度特征，能够得到具有辨识度的实体和关系表示向量来补全知识图谱。