Knowledge Graph Representation Learning Based on Automatic Network Search for Link Prediction

Abstract

Link prediction, also known as Knowledge Graph Completion (KGC), is the common task in Knowledge Graphs (KGs) to predict missing connections between entities.Most existing methods focus on designing shallow, scalable models, which have less expressive than deep, multi-layer models. Furthermore, most operations like addition, matrix multiplications or factorization are handcrafted based on a few known relation patterns in several wellknown datasets, such as FB15k, WN18, etc. However, due to the diversity and complex nature of real-world data distribution, it is inherently difficult to preset all latent patterns. To address this issue,we proposeKGE-ANS, a novel knowledge graph embedding framework for general link prediction tasks using automatic network search. KGEANS can learn a deep, multi-layer effective architecture to adapt to different datasets through neural architecture search. In addition, the general search spacewe designed is tailored forKGtasks.We performextensive experiments on benchmark datasets and the dataset constructed in this paper. The results show that our KGE-ANS outperforms several state-of-the-art methods, especially on these datasets with complex relation patterns.