Knowledge base completion by learning pairwise-interaction differentiated embeddings

• DOI: 10.1007/s10618-015-0430-1
• Publication Date: 2015-07-18T06:39:41Z
• Authors: Yu Zhao; Sheng Gao; Patrick Gallinari; Jun Guo
• Keywords: [INFO.INFO-AI] Computer Science [cs]/Artificial Intelligence [cs.AI]; 0202 electrical engineering, electronic engineering, information engineering; 02 engineering and technology; [INFO.INFO-AI]Computer Science [cs]/Artificial Intelligence [cs.AI]

A knowledge base of triples like (subject entity, predicate relation,object entity) is a very important resource for knowledge management. It is very useful for human-like reasoning, query expansion, question answering (Siri) and other related AI tasks. However, such a knowledge base often suffers from incompleteness due to a large volume of increasing knowledge in the real world and a lack of reasoning capability. In this paper, we propose a Pairwise-interaction Differentiated Embeddings model to embed entities and relations in the knowledge base to low dimensional vector representations and then predict the possible truth of additional facts to extend the knowledge base. In addition, we present a probability-based objective function to improve the model optimization. Finally, we evaluate the model by considering the problem of computing how likely the additional triple is true for the task of knowledge base completion. Experiments on WordNet and Freebase show the excellent performance of our model and algorithm.