Document-Level Relation Extraction with a Dependency Syntax Transformer and Supervised Contrastive Learning

Ming Yang,Yijia Zhang,Santosh Kumar Banbhrani,Hongfei Lin,Mingyu Lu

DOI: https://doi.org/10.1007/978-981-19-7596-7_4

2022-01-01

Abstract:Document-level Relation Extraction is more challenging than its sentence-level counterpart, extracting unknown relational facts from a plain text at the document level. Studies have shown that the Transformer architecture models long-distance dependencies without regard to the syntax-level dependencies between tokens in the sequence, which hinders its ability to model long-range dependencies. Furthermore, the global information among relational triples and local information around entities is critical. In this paper, we propose a Dependency Syntax Transformer and Supervised Contrastive Learning model (DSTSC) for document-level relation extraction. Specifically, dependency syntax information guides Transformer to enhance attention between tokens with dependency syntax relation in the sequence. The ability of Transformer to model document-level dependencies is improved. Supervised contrastive learning with fusion knowledge captures global information among relational triples. Gaussian probability distributions are also designed to capture local information around entities. Our experiments on two document-level relation extraction datasets, CDR and GDA, have remarkable results.