《计算机应用》唯一官方网站

• •    下一篇

基于不确定知识图谱嵌入的多关系近似推理模型

李健京1,李贯峰1,秦飞舟1,李卫军2   

  1. 1. 宁夏大学
    2. 北方民族大学
  • 收稿日期:2023-06-14 修回日期:2023-08-17 发布日期:2023-08-30 出版日期:2023-08-30
  • 通讯作者: 李健京
  • 基金资助:
    国家自然科学基金;宁夏自然科学基金;宁夏自然科学基金

Multi-relation approximate reasoning model based on uncertain knowledge graph embedding

  • Received:2023-06-14 Revised:2023-08-17 Online:2023-08-30 Published:2023-08-30

摘要: 针对大规模知识图谱(KG)的不确定性嵌入模型中无法对多种逻辑关系进行近似推理的问题,提出了一种基于不确定知识图谱嵌入的多关系近似推理模型UDConEx(Uncertainty DistMult and Complex Convolution Embedding)。首先,UDConEx模型结合了DistMult(Distance Multiplicative)模型和ComplEx(Complex Embedding)模型的特点,使得UDConEx模型具有推理对称与非对称关系的能力。其次,模型采用卷积神经网络(CNN)捕获不确定性知识图谱中的交互信息,使其具有推理逆关系和传递关系的能力。最后,UDConEx 模型利用神经网络对知识图谱的不确定信息进行置信度学习,在不确定性知识图谱嵌入空间中可以进行近似推理。通过对CN15k、NL27k、PPI5k三个公开数据集的实验表明,UDConEx模型与最新提出的MUKGE模型相比,在置信度预测任务中平均绝对误差(MAE)中分别降低了1.25,1.46,1.62;在关系事实排名任务中,基于线性的归一化折损累计增益(NDCG)在CN15k和NL27k数据集中分别提升了 5.8% 和2.6%;在多关系近似推理任务中验证了模型具有多种逻辑关系的近似推理能力。UDConEx模型弥补了传统嵌入模型无法进行置信度预测的不足,实现了对多种逻辑关系的近似推理,提供了更精确、具有可解释性的不确定性知识图谱推理能力。

关键词: 知识图谱, 多关系推理, 近似推理, 不确定性, 卷积神经网络

Abstract: In response to the issue that approximate reasoning on multiple logical relations could not be performed by the uncertain embedding model of a large-scale Knowledge Graph(KG), UDConEx(Uncertainty DistMult and Complex Convolution Embedding) was proposed, which was a multi-relation approximate reasoning model based on uncertain knowledge graph embedding. Firstly, the characteristics of the DistMult(Distance Multiplicative) model and the ComplEx(Complex Embedding) model were combined within the UDConEx model, bestowing upon it the capability to infer symmetric and asymmetric relationships. Subsequently, Convolutional Neural Network(CNN) was employed by the model to capture the interactive information in the uncertain KG, thereby enabling it to possess the ability to reason about inverse and transitive relationships. Lastly, the neural network was employed to carry out confidence learning of uncertain KG information, enabling the UDConEx model to perform approximate reasoning within the uncertain KG embedding space. The experimental results on three standard data sets of CN15k, NL27k, and PPI5k show that compared with the newly proposed MUKGE model, the Mean Absolute Error(MAE) of confidence prediction is reduced by 1.25, 1.46, and 1.62 respectively; in the task of relation fact ranking, the linear-based Normalized Discounted Cumulative Gain(NDCG) improved by 5.8% and 2.6% in the CN15k and NL27k datasets, respectively; in the multi-relation approximate reasoning task, it is verified that the model has the approximate reasoning ability of multiple logical relationships. The inability of traditional embedding models to predict confidence is compensated for by the UDConEx model, which achieves approximate reasoning for multiple logical relationships and offers enhanced accuracy and interpretability in uncertainty KG reasoning capabilities.

Key words: Knowledge Graph(KG), multi-relation reasoning, approximate reasoning, uncertainty, Convolutional Neural Network(CNN)

中图分类号: