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

Convolutional 2D knowledge graph embeddings

Dettmers, T; Minervini, P; Stenetorp, P; Riedel, S; (2018) Convolutional 2D knowledge graph embeddings. In: Zilberstein, Shlomo and McIlraith, Sheila and Weinberger, Kilian, (eds.) Proceedings of the 32nd AAAI Conference on Artificial Intelligence (AAAI-18). (pp. pp. 1811-1818). AAI Publications: Palo Alto, CA, USA. Green open access

[thumbnail of 1707.01476v6.pdf]
1707.01476v6.pdf - Accepted version

Download (372kB) | Preview


Link prediction for knowledge graphs is the task of predicting missing relationships between entities. Previous work on link prediction has focused on shallow, fast models which can scale to large knowledge graphs. However, these models learn less expressive features than deep, multi-layer models - which potentially limits performance. In this work we introduce ConvE, a multi-layer convolutional network model for link prediction, and report state-of-the-art results for several established datasets. We also show that the model is highly parameter efficient, yielding the same performance as DistMult and R-GCN with 8x and 17x fewer parameters. Analysis of our model suggests that it is particularly effective at modelling nodes with high indegree - which are common in highly-connected, complex knowledge graphs such as Freebase and YAGO3. In addition, it has been noted that the WN18 and FB15k datasets suffer from test set leakage, due to inverse relations from the training set being present in the test set - however, the extent of this issue has so far not been quantified. We find this problem to be severe: a simple rule-based model can achieve state-of-the-art results on both WN18 and FB15k. To ensure that models are evaluated on datasets where simply exploiting inverse relations cannot yield competitive results, we investigate and validate several commonly used datasets - deriving robust variants where necessary. We then perform experiments on these robust datasets for our own and several previously proposed models, and find that ConvE achieves state-of-the-art Mean Reciprocal Rank across all datasets.

Type: Proceedings paper
Title: Convolutional 2D knowledge graph embeddings
Event: 32nd AAAI Conference on Artificial Intelligence (AAAI-18), 2-7 February 2018, New Orleans, LA, USA
ISBN-13: 9781577358008
Open access status: An open access version is available from UCL Discovery
Publisher version: https://aaai.org/Conferences/AAAI-18/
Language: English
Additional information: This version is the author accepted manuscript. For information on re-use, please refer to the publisher’s terms and conditions.
Keywords: knowledge graphs; representation learning; convolution
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/1569849
Downloads since deposit
Download activity - last month
Download activity - last 12 months
Downloads by country - last 12 months

Archive Staff Only

View Item View Item