Improving Relation Extraction by Knowledge Representation Learning
Wenxing Hong, Shuyan Li, Zhiqiang Hu, Abdur Rasool, Qingshan Jiang, Yang Weng · 2021 IEEE 33rd International Conference on Tools with Artificial Intelligence (ICTAI) · 2021
Relation extraction is an important NLP task to extract the semantic relationship between two entities. Recently, large-scale pre-training language models have achieved excellent performance in many NLP applications. Most of the existing relation extraction models mainly rely on context information, but entity information is also very important for relation extraction, especially domain knowledge of entity and the direction between entity pairs. In this paper, based on the pre-trained BERT model, we propose a multi-task joint relation extraction model incorporating knowledge representation learning(KRL). The experimental results on the SemEval 2010 task 8 dataset and the KBP37 dataset show that our proposed model outperforms most of state-of-the-art methods. The results on the larger dataset FewRel80 refined from FewRel also indicate that increasing the knowledge representation learning as an auxiliary objective is helpful for the relation extraction task.