返回
Knowledge graph relation prediction based on graph transformation
DOI:10.1007/s10489-024-06080-y.png)
摘要
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.
Keyword:
Knowledge graph relation prediction
Graph transformation
Graph neural network
Relation correlation
期刊
IF:
3.5
论文数:
7.6K
被引数:
1.7W
机构
引用论文
Hierarchical-aware relation rotational knowledge graph embedding for link prediction层次感知关系旋转知识图嵌入的链路预测
NEUROCOMPUTING
IF6.5
Path-based reasoning approach for knowledge graph completion using CNN-BiLSTM with attention mechanism基于注意机制的cnn-bilstm路径推理知识图谱补全方法
PRGNN: Modeling high-order proximity with relational graph neural network for knowledge graph completionPRGNN: 用关系图神经网络对知识图进行高阶逼近建模
NEUROCOMPUTING
IF6.5

