Task-specific Pre-training and Prompt Decomposition for Knowledge Graph Population with Language Models

08/26/2022

∙

by   Tianyi Li, et al.

∙

1

∙

share

We present a system for knowledge graph population with Language Models, evaluated on the Knowledge Base Construction from Pre-trained Language Models (LM-KBC) challenge at ISWC 2022. Our system involves task-specific pre-training to improve LM representation of the masked object tokens, prompt decomposition for progressive generation of candidate objects, among other methods for higher-quality retrieval. Our system is the winner of track 1 of the LM-KBC challenge, based on BERT LM; it achieves 55.0 of the challenge.