arrow
Return

Knowledge graph relation prediction based on graph transformation

delete2024-12-30
delete0
PRE
AI
L
Linlan Liu
W
Weide Huang *
J
Jian Shu
H
H. Zhao
DOI:10.1007/s10489-024-06080-ydelete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Knowledge graph relation prediction aims to predict the missing relation between entities. Many existing graph neural network (GNN)-based relation prediction models suffer from over-parameterization, and some models cannot effectively learn the correlation between relations for the relation prediction task. In order to solve the above problems, we propose a knowledge graph relation prediction model based on graph transformation. We use two kinds of graph transformation and a parallel fusion model to learn the semantic information, which effectively reduces the number of parameters and reduces the loss of semantic information compared to the Levi graph. Then, we utilize the self-attention mechanism to learn the correlation between relations, and combine it with the DistMult scoring function to complete the relation prediction task. Experiments on four real-world datasets WN18RR, CoDEx-S, Kinship, and FB15K-237 show that our model achieved a better balance between the number of parameters and prediction performance compared to existing GNN-based models on most datasets.
Keywords:
Knowledge graph relation prediction
Graph transformation
Graph neural network
Relation correlation

Journal

Applied Intelligence cover
Applied Intelligence
IF:
3.5
Papers:
7.6K
Citations:
1.7W

Organization

N
Nanchang Hangkong University
Scholars:
7.4K
Papers: 4.0K
Citations: 81
Cited Papers

Cited Papers

err
IF0
err
err0
PREAI
err
errShare
errSave
Hierarchical-aware relation rotational knowledge graph embedding for link prediction
err2021-10-01
err23
PREAI
errWang, Shensi; Fu, Kun; Sun, Xian; Zhang, Zequn; Li, Shuchao; Jin, Li
errShare
errSave
Path-based reasoning approach for knowledge graph completion using CNN-BiLSTM with attention mechanism
err2020-03-01
err53
PREAI
errJagvaral, Batselem; Lee, Wan-Kon; Roh, Jae-Seung; Kim, Min-Sung; Park, Young Tack
errShare
errSave
Surveillance following radical or partial nephrectomy for renal cell carcinoma
err2005-01-01
err0
PREAI
errJohn S. Lam; John T. Leppert; Robert A. Figlin; Arie S. Belldegrun
errShare
errSave
errShare
errSave
researcher View more