Journal
IEEE TRANSACTIONS ON BIG DATA
Volume 8, Issue 3, Pages 630-643Publisher
IEEE-INST ELECTRICAL ELECTRONICS ENGINEERS INC
DOI: 10.1109/TBDATA.2018.2867583
Keywords
Link prediction; knowledge graph embedding; hierarchy
Funding
- National Grand Fundamental Research 973 Program of China [2013CB329602, 2014CB340401]
- National Natural Science Foundation of China [61572469, 61402442, 61572473, 61303244, 61402022, 61572467]
- Beijing nova program
- Beijing Natural Science Foundation [4154086]
Ask authors/readers for more resources
This research proposes a hierarchy-constrained link prediction method, called hTransM, which is based on knowledge graph embedding methods. By detecting hierarchical structures, hTransM can adaptively determine the optimal margin and has been proven effective through theoretical analysis and experiments.
Link prediction over a knowledge graph aims to predict the missing head entities h or tail entities t and missing relations r for a triple (h, r, t). Recent years have witnessed great advance of knowledge graph embedding based link prediction methods, which represent entities and relations as elements of a continuous vector space. Most methods learn the embedding vectors by optimizing a margin-based loss function, where the margin is used to separate negative and positive triples in the loss function. The loss function utilizes the general structures of knowledge graphs, e.g., the vector of r is the translation of the vector of h and t, and the vector of t should be the nearest neighbor of the vector of h + r. However, there are many particular structures, and can be employed to promote the performance of link prediction. One typical structure in knowledge graphs is hierarchical structure, which existing methods have much unexplored. We argue that the hierarchical structures also contain rich inference patterns, and can further enhance the link prediction performance. In this paper, we propose a hierarchy-constrained link prediction method, called hTransM, on the basis of the translation-based knowledge graph embedding methods. It can adaptively determine the optimal margin by detecting the single-step and multi-step hierarchical structures. Moreover, we prove the effectiveness of hTransM theoretically, and experiments over three benchmark datasets and two sub-tasks of link prediction demonstrate the superiority of hTransM.
Authors
I am an author on this paper
Click your name to claim this paper and add it to your profile.
Reviews
Recommended
No Data Available