[发明专利]知识图谱补全方法、装置、电子设备及存储介质

权利要求书

1.一种知识图谱补全方法，其特征在于，包括：

对待预测三元组中的头实体和尾实体进行信息交互融合，得到所述头实体对应的代理头实体和所述尾实体对应的代理尾实体；

通过计算按照所述待预测三元组中的头尾实体关系对所述代理头实体进行平移变换得到的实体与所述代理尾实体之间的差异，确定所述待预测三元组成立的概率；

若所述待预测三元组成立的概率达到预设阈值，则利用所述待预测三元组对知识图谱进行补全。

2.根据权利要求1所述的方法，其特征在于，对所述待预测三元组中的头实体和尾实体进行信息交互融合，得到所述头实体对应的代理头实体和所述尾实体对应的代理尾实体，包括：

根据待预测三元组中的头实体和尾实体，分别生成所述头实体的辅助向量和所述尾实体的辅助向量；

将所述头实体与所述尾实体的辅助向量进行交互融合，得到所述头实体对应的代理头实体，以及，将所述尾实体与所述头实体的辅助向量进行交互融合，得到所述尾实体对应的代理尾实体；

其中，所述头实体的辅助向量包含所述头实体的信息，所述尾实体的辅助向量包含所述尾实体的信息。

3.根据权利要求2所述的方法，其特征在于，将所述头实体与所述尾实体的辅助向量进行交互融合，得到所述头实体对应的代理头实体，以及，将所述尾实体与所述头实体的辅助向量进行交互融合，得到所述尾实体对应的代理尾实体，包括：

根据所述尾实体的辅助向量对所述头实体进行投影变换，得到所述头实体对应的代理头实体，以及，根据所述头实体的辅助向量对所述尾实体进行投影变换，得到所述尾实体对应的代理尾实体。

4.根据权利要求1所述的方法，其特征在于，通过计算按照所述待预测三元组中的头尾实体关系对所述代理头实体进行平移变换得到的实体与所述代理尾实体之间的差异，确定所述待预测三元组成立的概率，包括：

根据所述待预测三元组中的头尾实体关系，对所述代理头实体进行平移变换，得到实体平移向量；

计算所述实体平移向量与所述代理尾实体之间的距离；

根据所述实体平移向量与所述代理尾实体之间的距离，计算确定所述待预测三元组成立的概率。

5.根据权利要求1所述的方法，其特征在于，所述待预测三元组通过如下处理得到：

对所述知识图谱中的实体和各实体之间的实体关系进行三元组随机组合，将组合得到的、所述知识图谱中不存在的三元组作为待预测三元组。

6.根据权利要求1所述的方法，其特征在于，对待预测三元组中的头实体和尾实体进行信息交互融合，得到所述头实体对应的代理头实体和所述尾实体对应的代理尾实体，以及，通过计算按照所述待预测三元组中的头尾实体关系对所述代理头实体进行平移变换得到的实体与所述代理尾实体之间的差异，确定所述待预测三元组成立的概率，包括：

将待预测三元组输入到预先训练的知识图谱嵌入模型，使所述知识图谱嵌入模型对所述待预测三元组中的头实体和尾实体进行信息交互融合，得到所述头实体对应的代理头实体和所述尾实体对应的代理尾实体，以及，通过计算按照所述待预测三元组中的头尾实体关系对所述代理头实体进行平移变换得到的实体与所述代理尾实体之间的差异，确定所述待预测三元组成立的概率。