HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Tensor Decompositions for temporal knowledge base completion

Timothée Lacroix Guillaume Obozinski Nicolas Usunier

Tensor Decompositions for temporal knowledge base completion

Abstract

Most algorithms for representation learning and link prediction in relational data have been designed for static data. However, the data they are applied to usually evolves with time, such as friend graphs in social networks or user interactions with items in recommender systems. This is also the case for knowledge bases, which contain facts such as (US, has president, B. Obama, [2009-2017]) that are valid only at certain points in time. For the problem of link prediction under temporal constraints, i.e., answering queries such as (US, has president, ?, 2012), we propose a solution inspired by the canonical decomposition of tensors of order 4. We introduce new regularization schemes and present an extension of ComplEx (Trouillon et al., 2016) that achieves state-of-the-art performance. Additionally, we propose a new dataset for knowledge base completion constructed from Wikidata, larger than previous benchmarks by an order of magnitude, as a new reference for evaluating temporal and non-temporal link prediction methods.

Code Repositories

apoorvumang/CronKGQA
pytorch
Mentioned in GitHub
facebookresearch/tkbc
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
link-prediction-on-icews05-15-1TNTComplEx (x10)
MRR: 0.67
link-prediction-on-icews05-15-1TNTComplEx
MRR: 0.60
link-prediction-on-icews14-1TNTComplEx (x10)
MRR: 0.62
link-prediction-on-icews14-1TNTComplEx
MRR: 0.56
link-prediction-on-yago15k-1TNTComplEx
MRR: 0.35
link-prediction-on-yago15k-1TNTComplEx (x10)
MRR: 0.37

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Tensor Decompositions for temporal knowledge base completion | Papers | HyperAI