Knowledge graph embedding based on embedding permutation and high-frequency feature fusion for link prediction

被引:0
作者
Yu, Qien [1 ]
Vargas, Danilo Vasconcellos [1 ]
机构
[1] Kyushu Univ, Dept Informat Sci & Technol, 744 Motooka,Nishi Ku, Fukuoka 8190395, Japan
关键词
Knowledge graph embedding; Link prediction; Feature fusion; Convolutional neural network; Embedding permutation; Projection attention; NETWORK;
D O I
10.1016/j.neucom.2025.129743
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Knowledge graph embedding has excellent performance in capturing intrinsic relations and semantics in a wealth of information for link prediction. Knowledge graph embedding methods have achieved impressive results in recent years, especially those using convolutional neural networks. However, many previous approaches focus on interactions between relations and entities, ignoring interactions of internal data elements and the crucial role of high-frequency features. In this paper, we propose a novel approach, a knowledge graph Embedding model using 2D convolution operations integrating Embedding permutation strategy and High-frequency features fusion mechanism, named EHE, for link prediction. First, we design the embedding permutation mechanism for the embedding vectors. This mechanism leverages internal element permutation, efficiently broadening the local interactions of internal elements, especially for far-flung data elements in the one-dimensional space. Subsequently, a high-frequency feature fusion module is proposed to capture the high-frequency feature representations by using Sobel and Laplacian operators. Additionally, the projection attention mechanism is utilized to emphasize the unique semantic regions of interest in entities and relations. We assess our approach on several benchmark link prediction datasets. Considering the important metrics, MRR and H@1, our method achieves the overall best performance compared with existing state-of-the-art methods on five public datasets, showcasing its superior capacity for link prediction.
引用
收藏
页数:21
相关论文
共 64 条
  • [1] Hypernetwork Knowledge Graph Embeddings
    Balazevic, Ivana
    Allen, Carl
    Hospedales, Timothy M.
    [J]. ARTIFICIAL NEURAL NETWORKS AND MACHINE LEARNING - ICANN 2019: WORKSHOP AND SPECIAL SESSIONS, 2019, 11731 : 553 - 565
  • [2] Bansal T, 2019, 57TH ANNUAL MEETING OF THE ASSOCIATION FOR COMPUTATIONAL LINGUISTICS (ACL 2019), P4387
  • [3] Bollacker Kurt D., 2008, P 2008 ACM SIGMOD IN, P1247
  • [4] Bordes A., 2013, Advances in neural information processing systems, P2787
  • [5] Bordes A, 2014, Arxiv, DOI arXiv:1406.3676
  • [6] Distantly supervised relation extraction with a Meta-Relation enhanced Contrastive learning framework
    Chen, Chuanshu
    Hao, Shuang
    Liu, Jian
    [J]. NEUROCOMPUTING, 2025, 617
  • [7] Dettmers T, 2018, AAAI CONF ARTIF INTE, P1811
  • [8] Ding BY, 2018, Arxiv, DOI arXiv:1805.02408
  • [9] Generalized Translation-Based Embedding of Knowledge Graph
    Ebisu, Takuma
    Ichise, Ryutaro
    [J]. IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2020, 32 (05) : 941 - 951
  • [10] Ebisu T, 2018, AAAI CONF ARTIF INTE, P1819