LineaRE: Simple but Powerful Knowledge Graph Embedding for Link Prediction

21 Apr 2020  ·  Yanhui Peng, Jing Zhang ·

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.

PDF Abstract

Results from the Paper

Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Link Prediction FB15k LineaRE Hits@10 0.906 # 1
Hits@1 0.805 # 1
Hits@3 0.867 # 1
MR 36 # 2
MRR 0.843 # 1
Link Prediction FB15k-237 LineaRE MRR 0.357 # 24
Hits@10 0.545 # 20
Hits@3 0.391 # 19
Hits@1 0.264 # 22
MR 155 # 11
Link Prediction WN18 LineaRE MRR 0.952 # 3
Hits@10 0.961 # 2
Hits@3 0.955 # 2
Hits@1 0.947 # 3
MR 170 # 4
Link Prediction WN18RR LineaRE MRR 0.495 # 15
Hits@10 0.578 # 25
Hits@3 0.509 # 17
Hits@1 0.453 # 13
MR 1644 # 13