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
相关论文
共 50 条
  • [31] Knowledge graph embedding based on dynamic adaptive atrous convolution and attention mechanism for link prediction
    Deng, Weibin
    Zhang, Yiteng
    Yu, Hong
    Li, Hongxing
    INFORMATION PROCESSING & MANAGEMENT, 2024, 61 (03)
  • [32] A lightweight CNN-based knowledge graph embedding model with channel attention for link prediction
    Zhou, Xin
    Guo, Jingnan
    Jiang, Liling
    Ning, Bo
    Wang, Yanhao
    MATHEMATICAL BIOSCIENCES AND ENGINEERING, 2023, 20 (06) : 9607 - 9624
  • [33] Beyond Triplets: Hyper-Relational Knowledge Graph Embedding for Link Prediction
    Rosso, Paolo
    Yang, Dingqi
    Cudre-Mauroux, Philippe
    WEB CONFERENCE 2020: PROCEEDINGS OF THE WORLD WIDE WEB CONFERENCE (WWW 2020), 2020, : 1885 - 1896
  • [34] A Federated Multi-Server Knowledge Graph Embedding Framework For Link Prediction
    Hu, Ce
    Liu, Baisong
    Zhang, Xueyuan
    Wang, Zhiye
    Lin, Chennan
    Luo, Linze
    2022 IEEE 34TH INTERNATIONAL CONFERENCE ON TOOLS WITH ARTIFICIAL INTELLIGENCE, ICTAI, 2022, : 366 - 371
  • [35] RotatHS: Rotation Embedding on the Hyperplane with Soft Constraints for Link Prediction on Knowledge Graph
    Le, Thanh
    Huynh, Ngoc
    Le, Bac
    COMPUTATIONAL COLLECTIVE INTELLIGENCE (ICCCI 2021), 2021, 12876 : 29 - 41
  • [36] Link Prediction on Knowledge Graph by Rotation Embedding on the Hyperplane in the Complex Vector Space
    Thanh Le
    Ngoc Huynh
    Bac Le
    ARTIFICIAL NEURAL NETWORKS AND MACHINE LEARNING - ICANN 2021, PT III, 2021, 12893 : 164 - 175
  • [37] Hierarchical-aware relation rotational knowledge graph embedding for link prediction
    Wang, Shensi
    Fu, Kun
    Sun, Xian
    Zhang, Zequn
    Li, Shuchao
    Jin, Li
    NEUROCOMPUTING, 2021, 458 (458) : 259 - 270
  • [38] Knowledge graph embedding with the special orthogonal group in quaternion space for link prediction
    Le, Thanh
    Tran, Huy
    Le, Bac
    KNOWLEDGE-BASED SYSTEMS, 2023, 266
  • [39] Prediction of adverse drug reactions based on knowledge graph embedding
    Zhang, Fei
    Sun, Bo
    Diao, Xiaolin
    Zhao, Wei
    Shu, Ting
    BMC MEDICAL INFORMATICS AND DECISION MAKING, 2021, 21 (01)
  • [40] Graph embedding based feature selection
    Wei, Dan
    Li, Shutao
    Tan, Mingkui
    NEUROCOMPUTING, 2012, 93 : 115 - 125