A Unified Joint Approach with Topological Context Learning and Rule Augmentation for Knowledge Graph Completion

Jingtao Guo, Chunxia Zhang, Lingxi Li, Xiaojun Xue, Zhendong Niu


Abstract
Knowledge graph completion (KGC) task is to infer the missing knowledge in the knowledge graph based on known factual triples. However, present KGC approaches still face the following two challenges. Those methods perform simple linear update on relation representation, and only local neighborhood information is aggregated, which makes it difficult to capture logic semantic between relations and global topological context information. To tackle the above challenges, we propose a unified joint approach with Topological Context learning and Rule Augmentation (TCRA) for KGC. The TCRA framework consists of an entity topological context learning mechanism based on dual-branch hierarchical graph attention network, and a relation rule context learning mechanism based on Rule-Transformer and rule-to-relation aggregator. The former mechanism encodes the topological structure features of entities, aggregates the local neighborhood topological context information of entities on the three levels (entity, relation and triple), and build clusters of global head or tail entities related to the same relation. It can capture the local and global topological context information of entities related to the same relation. The latter mechanism introduces chain-like Horn rules as the context information of relations, and encodes the logical semantic of relations to enrich the relation representation. Experimental performances on three benchmark datasets FB15k-237, WN18RR and Kinship indicate the effectiveness and superiority of our proposed approach. The codes are publicly available.
Anthology ID:
2024.findings-acl.812
Volume:
Findings of the Association for Computational Linguistics: ACL 2024
Month:
August
Year:
2024
Address:
Bangkok, Thailand
Editors:
Lun-Wei Ku, Andre Martins, Vivek Srikumar
Venue:
Findings
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
13686–13696
Language:
URL:
https://aclanthology.org/2024.findings-acl.812
DOI:
10.18653/v1/2024.findings-acl.812
Bibkey:
Cite (ACL):
Jingtao Guo, Chunxia Zhang, Lingxi Li, Xiaojun Xue, and Zhendong Niu. 2024. A Unified Joint Approach with Topological Context Learning and Rule Augmentation for Knowledge Graph Completion. In Findings of the Association for Computational Linguistics: ACL 2024, pages 13686–13696, Bangkok, Thailand. Association for Computational Linguistics.
Cite (Informal):
A Unified Joint Approach with Topological Context Learning and Rule Augmentation for Knowledge Graph Completion (Guo et al., Findings 2024)
Copy Citation:
PDF:
https://aclanthology.org/2024.findings-acl.812.pdf