RotatE: Knowledge Graph Embedding by Relational Rotation in Complex Space

by   Zhiqing Sun, et al.
HEC Montréal
Université de Montréal
Peking University

We study the problem of learning representations of entities and relations in knowledge graphs for predicting missing links. The success of such a task heavily relies on the ability of modeling and inferring the patterns of (or between) the relations. In this paper, we present a new approach for knowledge graph embedding called RotatE, which is able to model and infer various relation patterns including: symmetry/antisymmetry, inversion, and composition. Specifically, the RotatE model defines each relation as a rotation from the source entity to the target entity in the complex vector space. In addition, we propose a novel self-adversarial negative sampling technique for efficiently and effectively training the RotatE model. Experimental results on multiple benchmark knowledge graphs show that the proposed RotatE model is not only scalable, but also able to infer and model various relation patterns and significantly outperform existing state-of-the-art models for link prediction.


page 1

page 2

page 3

page 4


Rot-Pro: Modeling Transitivity by Projection in Knowledge Graph Embedding

Knowledge graph embedding models learn the representations of entities a...

TeRo: A Time-aware Knowledge Graph Embedding via Temporal Rotation

In the last few years, there has been a surge of interest in learning re...

ChronoR: Rotation Based Temporal Knowledge Graph Embedding

Despite the importance and abundance of temporal knowledge graphs, most ...

Analogical Inference Enhanced Knowledge Graph Embedding

Knowledge graph embedding (KGE), which maps entities and relations in a ...

RatE: Relation-Adaptive Translating Embedding for Knowledge Graph Completion

Many graph embedding approaches have been proposed for knowledge graph c...

QuatDE: Dynamic Quaternion Embedding for Knowledge Graph Completion

In recent years, knowledge graph completion methods have been extensivel...

TransINT: Embedding Implication Rules in Knowledge Graphs with Isomorphic Intersections of Linear Subspaces

Knowledge Graphs (KG), composed of entities and relations, provide a str...

Code Repositories


High performance, easy-to-use, and scalable package for learning large-scale knowledge graph embeddings.

view repo

Please sign up or login with your details

Forgot password? Click here to reset