vix.ing · top · new · best · stats · spec

KG-BERT: BERT for Knowledge Graph Completion

2019/09/07 by Liang Yao, Yao, Liang, Chengsheng Mao +3 · 34 citations
Computer Science · #Advanced Graph Neural Networks #Artificial Intelligence (cs.AI) #Computation and Language (cs.CL) #FOS: Computer and information sciences #Semantic Web and Ontologies #Topic Modeling

paper · pdf · doi:10.48550/arxiv.1909.03193

openalex publication_date 2019/09/07 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/28

Abstract

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.

Citations

Cited by

Related