UCL Discovery
UCL home » Library Services » Electronic resources » UCL Discovery

Knowledge Graph Completion via Complex Tensor Factorization

Trouillon, T; Dance, CR; Gaussier, E; Welbl, J; Riedel, S; Bouchard, G; (2017) Knowledge Graph Completion via Complex Tensor Factorization. Journal of Machine Learning Research , 18 (130) pp. 1-38. Green open access

[thumbnail of Riedel_Knowledge Graph Completion via Complex Tensor Factorization_VoR.pdf]
Preview
Text
Riedel_Knowledge Graph Completion via Complex Tensor Factorization_VoR.pdf - Published Version

Download (1MB) | Preview

Abstract

In statistical relational learning, knowledge graph completion deals with automatically understanding the structure of large knowledge graphs—labeled directed graphs—and predicting missing relationships—labeled edges. State-of-the-art embedding models propose different trade-offs between modeling expressiveness, and time and space complexity. We reconcile both expressiveness and complexity through the use of complex-valued embeddings and explore the link between such complex-valued embeddings and unitary diagonalization. We corroborate our approach theoretically and show that all real square matrices—thus all possible relation/adjacency matrices—are the real part of some unitarily diagonalizable matrix. This results opens the door to a lot of other applications of square matrices factorization. Our approach based on complex embeddings is arguably simple, as it only involves a Hermitian dot product, the complex counterpart of the standard dot product between real vectors, whereas other methods resort to more and more complicated composition functions to increase their expressiveness. The proposed complex embeddings are scalable to large data sets as it remains linear in both space and time, while consistently outperforming alternative approaches on standard link prediction benchmarks.

Type: Article
Title: Knowledge Graph Completion via Complex Tensor Factorization
Open access status: An open access version is available from UCL Discovery
Publisher version: http://jmlr.org/papers/v18/16-563.html
Language: English
Additional information: © 2017 Théo Trouillon, Christopher R. Dance, Johannes Welbl, Sebastian Riedel, Éric Gaussier and Guillaume Bouchard. This is an open access article distributed under the terms of the Creative Commons Attribution 4.0 International License (https://creativecommons.org/licenses/by/4.0/)
Keywords: complex embeddings, tensor factorization, knowledge graph, matrix completion, statistical relational learning
UCL classification: UCL
UCL > Provost and Vice Provost Offices > UCL BEAMS
UCL > Provost and Vice Provost Offices > UCL BEAMS > Faculty of Engineering Science
UCL > Provost and Vice Provost Offices > UCL BEAMS > Faculty of Engineering Science > Dept of Computer Science
URI: https://discovery.ucl.ac.uk/id/eprint/10040098
Downloads since deposit
30Downloads
Download activity - last month
Download activity - last 12 months
Downloads by country - last 12 months

Archive Staff Only

View Item View Item