KGLM: Integrating Knowledge Graph Structure in Language Models for Link Prediction

Jason Youn, Ilias Tagkopoulos


Abstract
The ability of knowledge graphs to represent complex relationships at scale has led to their adoption for various needs including knowledge representation, question-answering, and recommendation systems. Knowledge graphs are often incomplete in the information they represent, necessitating the need for knowledge graph completion tasks. Pre-trained and fine-tuned language models have shown promise in these tasks although these models ignore the intrinsic information encoded in the knowledge graph, namely the entity and relation types. In this work, we propose the Knowledge Graph Language Model (KGLM) architecture, where we introduce a new entity/relation embedding layer that learns to differentiate distinctive entity and relation types, therefore allowing the model to learn the structure of the knowledge graph. In this work, we show that further pre-training the language models with this additional embedding layer using the triples extracted from the knowledge graph, followed by the standard fine-tuning phase sets a new state-of-the-art performance for the link prediction task on the benchmark datasets.
Anthology ID:
2023.starsem-1.20
Volume:
Proceedings of the 12th Joint Conference on Lexical and Computational Semantics (*SEM 2023)
Month:
July
Year:
2023
Address:
Toronto, Canada
Editors:
Alexis Palmer, Jose Camacho-collados
Venue:
*SEM
SIG:
SIGLEX
Publisher:
Association for Computational Linguistics
Note:
Pages:
217–224
Language:
URL:
https://aclanthology.org/2023.starsem-1.20
DOI:
10.18653/v1/2023.starsem-1.20
Bibkey:
Cite (ACL):
Jason Youn and Ilias Tagkopoulos. 2023. KGLM: Integrating Knowledge Graph Structure in Language Models for Link Prediction. In Proceedings of the 12th Joint Conference on Lexical and Computational Semantics (*SEM 2023), pages 217–224, Toronto, Canada. Association for Computational Linguistics.
Cite (Informal):
KGLM: Integrating Knowledge Graph Structure in Language Models for Link Prediction (Youn & Tagkopoulos, *SEM 2023)
Copy Citation:
PDF:
https://aclanthology.org/2023.starsem-1.20.pdf