Combining Two and Three-Way Embedding Models for Link Prediction in Knowledge Bases - Archive ouverte HAL Accéder directement au contenu
Article Dans Une Revue Journal of Artificial Intelligence Research Année : 2016

Combining Two and Three-Way Embedding Models for Link Prediction in Knowledge Bases

Résumé

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.

Dates et versions

hal-01313319 , version 1 (09-05-2016)

Identifiants

Citer

Alberto García-Durán, Antoine Bordes, Nicolas Usunier, Yves Grandvalet. Combining Two and Three-Way Embedding Models for Link Prediction in Knowledge Bases. Journal of Artificial Intelligence Research, 2016, 55, pp.715--742. ⟨10.1613/jair.5013⟩. ⟨hal-01313319⟩
122 Consultations
0 Téléchargements

Altmetric

Partager

Gmail Facebook X LinkedIn More