Complex-InversE: Improving bilinear knowledge graph embeddings by mapping into complex space

Article type: Research Article

Authors: Cao, Yukun^{; *} | Miao, Zeyu

Affiliations: Shanghai University of Electric Power, Shanghai

Correspondence: [*] Corresponding author. Yukun Cao, Shanghai University of Electric Power, 1851 Huchenghuan Road, Shanghai. E-mail: marilyn_cao@163.com.

Abstract: Knowledge graph link prediction uses known fact links to infer the missing link information in the knowledge graph, which is of great significance to the completion of the knowledge graph. Generating low-dimensional embeddings of entities and relations which are used to make inferences is a popular way for such link prediction problems. This paper proposes a knowledge graph link prediction method called Complex-InversE in the complex space, which maps entities and relations into the complex space. The composition of complex embeddings can handle a large variety of binary relations, among them symmetric and antisymmetric relations. The Complex-InversE effectively captures the antisymmetric relations and introduces Dropout and Early-Stopping technologies into deal with the problem of small numbers of relationships and entities, thus effectively alleviates the model’s overfitting. The results of comparison experiment on the public knowledge graph datasets show that the Complex-InversE achieves good results on multiple benchmark evaluation indicators and outperforms previous methods. Complex-InversE’s code is available on GitHub at https://github.com/ZeyuMiao97/Complex-InversE.

DOI: 10.3233/JIFS-212374

Journal: Journal of Intelligent & Fuzzy Systems, vol. 42, no. 6, pp. 6079-6089, 2022