Multi-level Shared Knowledge Guided Learning for Knowledge Graph Completion

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Shan, Yongxue, Zhou, Jie, Peng, Jie, Zhou, Xin, Yin, Jiaqian, Wang, Xiaodong
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866913346433843200
author Shan, Yongxue
Zhou, Jie
Peng, Jie
Zhou, Xin
Yin, Jiaqian
Wang, Xiaodong
author_facet Shan, Yongxue
Zhou, Jie
Peng, Jie
Zhou, Xin
Yin, Jiaqian
Wang, Xiaodong
contents In the task of Knowledge Graph Completion (KGC), the existing datasets and their inherent subtasks carry a wealth of shared knowledge that can be utilized to enhance the representation of knowledge triplets and overall performance. However, no current studies specifically address the shared knowledge within KGC. To bridge this gap, we introduce a multi-level Shared Knowledge Guided learning method (SKG) that operates at both the dataset and task levels. On the dataset level, SKG-KGC broadens the original dataset by identifying shared features within entity sets via text summarization. On the task level, for the three typical KGC subtasks - head entity prediction, relation prediction, and tail entity prediction - we present an innovative multi-task learning architecture with dynamically adjusted loss weights. This approach allows the model to focus on more challenging and underperforming tasks, effectively mitigating the imbalance of knowledge sharing among subtasks. Experimental results demonstrate that SKG-KGC outperforms existing text-based methods significantly on three well-known datasets, with the most notable improvement on WN18RR.
format Preprint
id arxiv_https___arxiv_org_abs_2405_06696
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Multi-level Shared Knowledge Guided Learning for Knowledge Graph Completion
Shan, Yongxue
Zhou, Jie
Peng, Jie
Zhou, Xin
Yin, Jiaqian
Wang, Xiaodong
Computation and Language
Artificial Intelligence
In the task of Knowledge Graph Completion (KGC), the existing datasets and their inherent subtasks carry a wealth of shared knowledge that can be utilized to enhance the representation of knowledge triplets and overall performance. However, no current studies specifically address the shared knowledge within KGC. To bridge this gap, we introduce a multi-level Shared Knowledge Guided learning method (SKG) that operates at both the dataset and task levels. On the dataset level, SKG-KGC broadens the original dataset by identifying shared features within entity sets via text summarization. On the task level, for the three typical KGC subtasks - head entity prediction, relation prediction, and tail entity prediction - we present an innovative multi-task learning architecture with dynamically adjusted loss weights. This approach allows the model to focus on more challenging and underperforming tasks, effectively mitigating the imbalance of knowledge sharing among subtasks. Experimental results demonstrate that SKG-KGC outperforms existing text-based methods significantly on three well-known datasets, with the most notable improvement on WN18RR.
title Multi-level Shared Knowledge Guided Learning for Knowledge Graph Completion
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2405.06696