Title
Combining Two and Three-Way Embedding Models for Link Prediction in Knowledge Bases.
Abstract
This paper tackles the problem of endogenous link prediction for knowledge base completion. Knowledge bases can be represented as directed graphs whose nodes correspond to entities and edges to relationships. Previous attempts either consist of powerful systems with high capacity to model complex connectivity patterns, which unfortunately usually end up overfitting on rare relationships, or in approaches that trade capacity for simplicity in order to fairly model all relationships, frequent or not. In this paper, we propose TATEC, a happy medium obtained by complementing a high-capacity model with a simpler one, both pre-trained separately and then combined. We present several variants of this model with different kinds of regularization and combination strategies and show that this approach outperforms existing methods on different types of relationships by achieving state-of-the-art results on four benchmarks of the literature.
Year
DOI
Venue
2016
10.1613/jair.5013
JOURNAL OF ARTIFICIAL INTELLIGENCE RESEARCH
Field
DocType
Volume
Graph,Embedding,Regularization (mathematics),Artificial intelligence,Knowledge base,Overfitting,Machine learning,Mathematics
Journal
55
Issue
ISSN
Citations 
1
1076-9757
19
PageRank 
References 
Authors
0.64
26
4
Name
Order
Citations
PageRank
Alberto García-Durán1531.70
Antoine Bordes23289157.12
Nicolas Usunier3197497.52
Yves Grandvalet499593.81