Knowledge Representation Learning via Dynamic Relation Spaces

Knowledge graphs are an important part in AI domain, and contain large scale of structured knowledge, but they are far from completeness. Previous translation models, such as TransE, TransH, TransR/CTransR and TransD, use a relation vector to translate head entity vector, the result of translation is close to tail entity vector. Compared with other classical models, these translation models achieve state-of-the-art performance. In this paper, we propose a more flexible model named TransDR, which is an improvement of TransD. In TransDR, we use two vectors to represent each entity and three vectors to represent relation. Compared with TransD, TransDR adds another vector for each relation which could not only represent model more flexibly but also reduce the noise from other relation spaces. In experiments, we evaluate our model on two typical tasks including triplets classification and link prediction. Experiment results show significant and consistent improvements compared to previous state-of-the-art models.