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

说明书

本申请提出一种知识图谱补全方法、装置、电子设备及存储介质，方法包括：对待预测三元组中的头实体和尾实体进行信息交互融合，得到头实体对应的代理头实体和尾实体对应的代理尾实体；通过计算按照待预测三元组中的头尾实体关系对代理头实体进行平移变换得到的实体与代理尾实体之间的差异，确定待预测三元组成立的概率；若待预测三元组成立的概率达到预设阈值，则利用待预测三元组对知识图谱进行补全。采用本申请的技术方案，可以实现三元组中头实体和尾实体的信息交互融合，利用实体信息交互后的代理头实体和代理尾实体判断三元组是否成立，能够提高事实三元组的判定准确性和知识图谱补全的准确度。

技术领域

本申请涉及自然语言处理技术领域，尤其涉及一种知识图谱补全方法、装置、电子设备及存储介质。

背景技术

在知识图谱中，一条事实通常表示为三元组(triple)形式，即“(头实体，关系，尾实体)”形式。通常情况下，知识图谱补全是利用知识图谱嵌入模型，通过对三元组中真实头实体和实体关系的变换，得到变换后的尾实体，知识图谱嵌入模型根据变换后的尾实体与真实尾实体之间的距离，确定该三元组是否成立，以便利用成立的三元组进行知识图谱补全。

但是，现有的知识图谱嵌入模型中的缺乏头尾实体交互，导致事实三元组的判定准确性较低，影响知识图谱补全的准确度。

发明内容

基于上述现有技术的缺陷和不足，本申请提出一种知识图谱补全方法、装置、电子设备及存储介质，能够在知识图谱嵌入中，实现头尾实体之间的信息交互，提高事实三元组的判定准确性和知识图谱补全的准确度。

本申请第一方面提供了一种知识图谱补全方法，包括：

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

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

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

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

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

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

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

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

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

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

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

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

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