arrow
返回

Graph Structure Enhanced Pre-Training Language Model for Knowledge Graph Completion

delete2024-08-01
delete17
PRE
AI
H
Huashi Zhu
D
Dexuan Xu
Y
Yu Huang *
Z
Zhi Jin
丁卫平 封面图
丁卫平 (Weiping Ding)
J
Jiahui Tong
G
Guoshuang Chong
DOI:10.1109/TETCI.2024.3372442delete
delete原文链接
delete分享
delete收藏
摘要

摘要

En 中文
A vast amount of textual and structural information is required for knowledge graph construction and its downstream tasks. However, most of the current knowledge graphs are incomplete due to the difficulty of knowledge acquisition and integration. Knowledge Graph Completion (KGC) is used to predict missing connections. In previous studies, textual information and graph structural information are utilized independently, without an effective method for fusing these two types of information. In this paper, we propose a graph structure enhanced pre-training language model for knowledge graph completion. Firstly, we design a graph sampling algorithm and a Graph2Seq module for constructing sub-graphs and their corresponding contexts to support large-scale knowledge graph learning and parallel training. It is also the basis for fusing textual data and graph structure. Next, two pre-training tasks based on masked modeling are designed for capturing accurate entity-level and relation-level information. Furthermore, this paper proposes a novel asymmetric Encoder-Decoder architecture to restore masked components, where the encoder is a Pre-trained Language Model (PLM) and the decoder is a multi-relational Graph Neural Network (GNN). The purpose of the architecture is to integrate textual information effectively with graph structural information. Finally, the model is fine-tuned for KGC tasks on two widely used public datasets. The experiments show that the model achieves excellent performance and outperforms baselines in most metrics, which demonstrate the effectiveness of our approach by fusing the structure and semantic information to knowledge graph.
Keyword:
Knowledge graph completion
masked modeling
pre-trained language models
graph neural network

期刊

I
IEEE Transactions on Emerging Topics in Computational Intelligence
IF:
6.5
论文数:
1.5K
被引数:
5.0K

机构

P
Peking University
学者数:
7.0K
论文数: 2.1K
被引数: 1
N
Nantong University
学者数:
1.3K
论文数: 388
被引数: 0
引用论文

引用论文

DBpedia - A large-scale, multilingual knowledge base extracted from WikipediaDBpedia-从维基百科中提取的大规模多语言知识库
err2015-01-01
err2.0K
errOAAI
errLehmann, Jens; Isele, Robert; Jakob, Max; Jentzsch, Anja; Kontokostas, Dimitris; Mendes, Pablo N.; Hellmann, Sebastian; Morsey, Mohamed; van Kleef, Patrick; Auer, Soeren; Bizer, Christian
err分享
err收藏
Lithium Dendrites: Inside or Outside: Origin of Lithium Dendrite Formation of All Solid‐State Electrolytes (Adv. Energy Mater. 40/2019)
err2019-10-24
err0
errOAAI
errFangjie Mo; Jiafeng Ruan; Shuxian Sun; Zixuan Lian; Sangpu Yang; Xinyang Yue; Yun Song; Yong‐Ning Zhou; Fang Fang; Guangai Sun; Shuming Peng; Dalin Sun
err分享
err收藏
Management of Caspian Biodiversity Protection and Conservation
err2016-06-29
err0
PREAI
errE. Mammadov; S. Timirkhanov; T. Shiganova; D. Katunin; A. Abdoli; R. Shahifar; Y. Kim; R. Khodorevsakaya; J. Annachariyeva; V. Velikova
err分享
err收藏
Relphormer: Relational Graph Transformer for Knowledge Graph RepresentationsRelphormer: 用于知识图表示的关系图转换器
err2024-01-01
err20
PREAI
errBi, Zhen; Cheng, Siyuan; Chen, Jing; Liang, Xiaozhuan; Xiong, Feiyu; Zhang, Ningyu
err分享
err收藏
学者 查看更多内容