Affiliations: [a] School of Electrical and Information Engineering, Beijing University of Civil Engineering and Architecture, Beijing, China
| [b] School of Management Science and Engineering, Shandong University of Finance and Economics, Jinan, China
Correspondence:
[*]
Corresponding author. Chuyuan Wei, School of Electrical and Information Engineering, Beijing University of Civil Engineering and Architecture, Beijing, China. E-mail: [email protected].
Abstract: Knowledge graphs are knowledge bases that represent entities and relations in the objective world through graph structure, and they have a boosting effect on many artificial intelligence tasks. To facilitate the development of downstream artificial intelligence tasks, knowledge graph embedding (KGE) is proposed. It aims to express semantic information for each entity and relation in the knowledge graph within a low-dimensional space. However, when it comes to semantic hierarchy, multiple relation patterns and multi-fold relational structures in knowledge graphs, most of the existing models tend to focus on only one or two aspects, often neglecting the importance of considering all three simultaneously. Therefore, we propose a new knowledge graph embedding model, Hierarchical relation and Entity Rotation-based Multi-Feature knowledge graph Embedding (HERotMFE). Concerning hierarchical relation rotation and entity rotation, it can represent semantic hierarchy, multiple relation patterns and multi-fold relations simultaneously. Self-attention mechanism is used to learn the weights of the two-part rotation to further enhance the model’s performance. According to the findings of the experiments, HERotMFE outperforms existing models on most metrics and achieves state-of-the-art results.
