In plain words: It turns each fact—two entities and their relation—into short text using their descriptions, then lets a pre-trained language model score it to fill in missing facts. This beat the usual approach of learning simple vector scores for entities and relations on standard tests.
Abstract · KG-BERT: BERT for Knowledge Graph Completion
Knowledge graphs are important resources for many artificial intelligence tasks but often suffer from incompleteness. In this work, we propose to use pre-trained language models for knowledge graph completion. We treat triples in knowledge graphs as textual sequences and propose a novel framework named Knowledge Graph Bidirectional Encoder Representations from Transformer (KG-BERT) to model these triples. Our method takes entity and relation descriptions of a triple as input and computes scoring function of the triple with the KG-BERT language model. Experimental results on multiple benchmark knowledge graphs show that our method can achieve state-of-the-art performance in triple classification, link prediction and relation prediction tasks.
Liang Yao, Chengsheng Mao, Yuan Luo
arXiv:1909.03193 · cs.CL, cs.AI · submitted Sep 7, 2019 · updated Sep 11, 2019
abstract · pdf · html