THT: Transfer Head and Tail for Joint Entity Relation Extraction

H.J. Zou,Jiawei Jiang,Yan Fu

DOI: https://doi.org/10.1088/1742-6596/2504/1/012015

2023-01-01

Journal of Physics Conference Series

Abstract:Extracting entities and relations from unstructured text is the basic task of building knowledge graphs. Various methods based on sequence labeling have been proposed recently, but most of them use uniform token representations to identify entities. It is clear that the first and last words of the head and tail entities should have different representations. For this reason, we propose a method based on word interaction that provides different representations for the first and last words of entities. Specifically, initial token representation is obtained by BERT. Each token interacts with the next k tokens for the entity’s first token recognition, interacts with the previous k tokens for the entity’s last token recognition, and merges them to obtain the head entity. Then recognize the tail entities based on the head entities and form entity pairs. Finally, get the triples based on a simple multi-classification for relation determination. Extensive experimental results on four datasets show that THT performs better than state-of-the-art baselines.