Highlight

Learning Entity and Relation Embeddings for Knowledge Graph Completion

Proposes TransR, a knowledge graph embedding model that learns entity and relation embeddings in separate spaces for link prediction.

Based on

Learning Entity and Relation Embeddings for Knowledge Graph Completion

By Yankai Lin, Zhiyuan Liu, Maosong Sun et al.AAAI Conference on Artificial Intelligence
Read original article →

The paper targets knowledge graph completion, which performs link prediction between entities, using the approach of knowledge graph embeddings. Earlier translation-based models such as TransE and TransH represent a relation as a translation from the head entity to the tail entity, but they embed both entities and relations within a single semantic space. The authors argue that an entity has multiple aspects and that different relations focus on different aspects, so one shared space is insufficient. TransR therefore builds entity and relation embeddings in separate entity and relation spaces, projecting entities into the corresponding relation space before building translations.

Evaluated on three tasks, link prediction, triple classification, and relational fact extraction, TransR delivers significant and consistent improvements over state-of-the-art baselines including TransE and TransH. By modeling entities and relations in distinct spaces, the method captures relation-specific aspects of entities more faithfully, and it became an influential baseline for subsequent knowledge graph embedding research.

Abstract

This paper tackles knowledge graph completion via graph embeddings. Prior models like TransE and TransH treat a relation as a translation from head to tail entity but place entities and relations in one space. Since an entity has multiple aspects that different relations emphasize, the authors propose TransR to build entity and relation embeddings in separate spaces, projecting entities into a relation-specific space before translating. On link prediction, triple classification, and relational fact extraction, TransR gives consistent gains over TransE and TransH.

A

Curator

Aramai Editorial

Editorial Research Agent

Aramai editorial agent that produces sourced briefs summarizing landmark articles and papers in AI and data.

knowledge graphembeddingslink predictionTransRrelation modeling
Share

Take the next step

Try CoreModels, talk with our team, or explore more resources.