Few-Shot Knowledge Graph Completion Model Based on Relation Learning

Considering the complexity of entity pair relations and the information contained in the target neighborhood in few-shot knowledge graphs (KG), existing few-shot KG completion methods generally suffer from insufficient relation representation learning capabilities and neglecting the contextual seman...

Full description

Bibliographic Details
Main Authors: Weijun Li, Jianlai Gu, Ang Li, Yuxiao Gao, Xinyong Zhang
Format: Article
Language:English
Published: MDPI AG 2023-08-01
Series:Applied Sciences
Subjects:
Online Access:https://www.mdpi.com/2076-3417/13/17/9513
_version_ 1797582865530093568
author Weijun Li
Jianlai Gu
Ang Li
Yuxiao Gao
Xinyong Zhang
author_facet Weijun Li
Jianlai Gu
Ang Li
Yuxiao Gao
Xinyong Zhang
author_sort Weijun Li
collection DOAJ
description Considering the complexity of entity pair relations and the information contained in the target neighborhood in few-shot knowledge graphs (KG), existing few-shot KG completion methods generally suffer from insufficient relation representation learning capabilities and neglecting the contextual semantics of entities. To tackle the above problems, we propose a Few-shot Relation Learning-based Knowledge Graph Completion model (FRL-KGC). First, a gating mechanism is introduced during the aggregation of higher-order neighborhoods of entities in formation, enriching the central entity representation while reducing the adverse effects of noisy neighbors. Second, during the relation representation learning stage, a more accurate relation representation is learned by using the correlation between entity pairs in the reference set. Finally, an LSTM structure is incorporated into the Transformer learner to enhance its ability to learn the contextual semantics of entities and relations and predict new factual knowledge. We conducted comparative experiments on the publicly available NELL-One and Wiki-One datasets, comparing FRL-KGC with six few-shot knowledge graph completion models and five traditional knowledge graph completion models for five-shot link prediction. The results showed that FRL-KGC outperformed all comparison models in terms of MRR, Hits@10, Hits@5, and Hits@1 metrics.
first_indexed 2024-03-10T23:28:35Z
format Article
id doaj.art-597be201bf2142408ab0298b85852be3
institution Directory Open Access Journal
issn 2076-3417
language English
last_indexed 2024-03-10T23:28:35Z
publishDate 2023-08-01
publisher MDPI AG
record_format Article
series Applied Sciences
spelling doaj.art-597be201bf2142408ab0298b85852be32023-11-19T07:48:06ZengMDPI AGApplied Sciences2076-34172023-08-011317951310.3390/app13179513Few-Shot Knowledge Graph Completion Model Based on Relation LearningWeijun Li0Jianlai Gu1Ang Li2Yuxiao Gao3Xinyong Zhang4The Key Laboratory of Images and Graphics Intelligent Processing of State Ethnic Affairs Commission, North Minzu University, Yinchuan 750021, ChinaSchool of Computer Science and Engineering, North Minzu University, Yinchuan 750021, ChinaSchool of Computer Science and Engineering, North Minzu University, Yinchuan 750021, ChinaSchool of Computer Science and Engineering, North Minzu University, Yinchuan 750021, ChinaSchool of Computer Science and Engineering, North Minzu University, Yinchuan 750021, ChinaConsidering the complexity of entity pair relations and the information contained in the target neighborhood in few-shot knowledge graphs (KG), existing few-shot KG completion methods generally suffer from insufficient relation representation learning capabilities and neglecting the contextual semantics of entities. To tackle the above problems, we propose a Few-shot Relation Learning-based Knowledge Graph Completion model (FRL-KGC). First, a gating mechanism is introduced during the aggregation of higher-order neighborhoods of entities in formation, enriching the central entity representation while reducing the adverse effects of noisy neighbors. Second, during the relation representation learning stage, a more accurate relation representation is learned by using the correlation between entity pairs in the reference set. Finally, an LSTM structure is incorporated into the Transformer learner to enhance its ability to learn the contextual semantics of entities and relations and predict new factual knowledge. We conducted comparative experiments on the publicly available NELL-One and Wiki-One datasets, comparing FRL-KGC with six few-shot knowledge graph completion models and five traditional knowledge graph completion models for five-shot link prediction. The results showed that FRL-KGC outperformed all comparison models in terms of MRR, Hits@10, Hits@5, and Hits@1 metrics.https://www.mdpi.com/2076-3417/13/17/9513knowledge graphcomplete the knowledge graphfew-shot relationneighborhood aggregationlink prediction
spellingShingle Weijun Li
Jianlai Gu
Ang Li
Yuxiao Gao
Xinyong Zhang
Few-Shot Knowledge Graph Completion Model Based on Relation Learning
Applied Sciences
knowledge graph
complete the knowledge graph
few-shot relation
neighborhood aggregation
link prediction
title Few-Shot Knowledge Graph Completion Model Based on Relation Learning
title_full Few-Shot Knowledge Graph Completion Model Based on Relation Learning
title_fullStr Few-Shot Knowledge Graph Completion Model Based on Relation Learning
title_full_unstemmed Few-Shot Knowledge Graph Completion Model Based on Relation Learning
title_short Few-Shot Knowledge Graph Completion Model Based on Relation Learning
title_sort few shot knowledge graph completion model based on relation learning
topic knowledge graph
complete the knowledge graph
few-shot relation
neighborhood aggregation
link prediction
url https://www.mdpi.com/2076-3417/13/17/9513
work_keys_str_mv AT weijunli fewshotknowledgegraphcompletionmodelbasedonrelationlearning
AT jianlaigu fewshotknowledgegraphcompletionmodelbasedonrelationlearning
AT angli fewshotknowledgegraphcompletionmodelbasedonrelationlearning
AT yuxiaogao fewshotknowledgegraphcompletionmodelbasedonrelationlearning
AT xinyongzhang fewshotknowledgegraphcompletionmodelbasedonrelationlearning