Knowledge graph embedding model with attention-based high-low level features interaction convolutional network

作者：Wang, Jingxiong; Zhang, Qi; Shi, Fobo^*; Li, Duantengchuan^*; Cai, Yuefeng; Wang, Jian; Li, Bing; Wang, Xiaoguang; Zhang, Zhen; Zheng, Chao

来源：INFORMATION PROCESSING & MANAGEMENT, 2023, 60(4): 103350.

DOI：10.1016/j.ipm.2023.103350

• Link prediction
• Knowledge graph embedding
• Convolutional neural network
• Criss-cross attention mechanism
• High-low level features interaction

Knowledge graphs are sizeable graph-structured knowledge with both abstract and concrete concepts in the form of entities and relations. Recently, convolutional neural networks have achieved outstanding results for more expressive representations of knowledge graphs. However, existing deep learning-based models exploit semantic information from single-level feature interaction, potentially limiting expressiveness. We propose a knowledge graph embedding model with an attention-based high-low level features interaction convolutional network called ConvHLE to alleviate this issue. This model effectively harvests richer semantic information and generates more expressive representations. Concretely, the multilayer convolutional neural network is utilized to fuse high-low level features. Then, features in fused feature maps interact with other informative neighbors through the criss-cross attention mechanism, which expands the receptive fields and boosts the quality of interactions. Finally, a plausibility score function is proposed for the evaluation of our model. The performance of ConvHLE is experimentally investigated on six benchmark datasets with individual characteristics. Extensive experimental results prove that ConvHLE learns more expressive and discriminative feature representations and has outperformed other state-of-the-art baselines over most metrics when addressing link prediction tasks. Comparing MRR and Hits@1 on FB15K-237, our model outperforms the baseline ConvE by 13.5% and 16.0%, respectively.

• 单位
  武汉大学