论文标题

线性:简单但功能强大的知识图嵌入链接预测

LineaRE: Simple but Powerful Knowledge Graph Embedding for Link Prediction

论文作者

Peng, Yanhui, Zhang, Jing

论文摘要

知识图的链接预测的任务是预测实体之间的缺失关系。知识图嵌入旨在将知识图表示的实体和关系作为连续矢量空间中低维向量的实体和关系,已经实现了有希望的预测性能。如果嵌入模型可以涵盖不同类型的连接模式和尽可能多的关系的映射属性,则可能会为链接预测任务带来更多好处。在本文中,我们提出了一种新型的嵌入模型,即线性,该模型能够建模四种连接模式(即对称,反对称,反向,反转和组成)和四个映射属性(即一对一,一对一,一对一,多对一对,多个,多个对象)。具体而言,我们将嵌入知识图作为一个简单的线性回归任务,其中关系被建模为两个具有两个权重矢量和一个偏置向量的两个低维矢量呈现的实体的线性函数。由于向量是在实际数字空间中定义的,并且模型的评分函数是线性的,因此我们的模型简单可扩展到大知识图。多种使用现实世界中数据集的实验结果表明,提出的线性模型显着优于链接预测任务的现有最新模型。

The task of link prediction for knowledge graphs is to predict missing relationships between entities. Knowledge graph embedding, which aims to represent entities and relations of a knowledge graph as low dimensional vectors in a continuous vector space, has achieved promising predictive performance. If an embedding model can cover different types of connectivity patterns and mapping properties of relations as many as possible, it will potentially bring more benefits for link prediction tasks. In this paper, we propose a novel embedding model, namely LineaRE, which is capable of modeling four connectivity patterns (i.e., symmetry, antisymmetry, inversion, and composition) and four mapping properties (i.e., one-to-one, one-to-many, many-to-one, and many-to-many) of relations. Specifically, we regard knowledge graph embedding as a simple linear regression task, where a relation is modeled as a linear function of two low-dimensional vector-presented entities with two weight vectors and a bias vector. Since the vectors are defined in a real number space and the scoring function of the model is linear, our model is simple and scalable to large knowledge graphs. Experimental results on multiple widely used real-world datasets show that the proposed LineaRE model significantly outperforms existing state-of-the-art models for link prediction tasks.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源