HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

A Densely Connected Criss-Cross Attention Network for Document-level Relation Extraction

Liang Zhang Yidong Cheng

A Densely Connected Criss-Cross Attention Network for Document-level Relation Extraction

Abstract

Document-level relation extraction (RE) aims to identify relations between two entities in a given document. Compared with its sentence-level counterpart, document-level RE requires complex reasoning. Previous research normally completed reasoning through information propagation on the mention-level or entity-level document-graph, but rarely considered reasoning at the entity-pair-level.In this paper, we propose a novel model, called Densely Connected Criss-Cross Attention Network (Dense-CCNet), for document-level RE, which can complete logical reasoning at the entity-pair-level. Specifically, the Dense-CCNet performs entity-pair-level logical reasoning through the Criss-Cross Attention (CCA), which can collect contextual information in horizontal and vertical directions on the entity-pair matrix to enhance the corresponding entity-pair representation. In addition, we densely connect multiple layers of the CCA to simultaneously capture the features of single-hop and multi-hop logical reasoning.We evaluate our Dense-CCNet model on three public document-level RE datasets, DocRED, CDR, and GDA. Experimental results demonstrate that our model achieves state-of-the-art performance on these three datasets.

Benchmarks

BenchmarkMethodologyMetrics
relation-extraction-on-cdrDense-CCNet-SciBERTbase
F1: 77.06
relation-extraction-on-docredDense-CCNet-BERTbase
F1: 62.55
Ign F1: 60.46
relation-extraction-on-gdaDense-CCNet-SciBERTbase
F1: 86.44

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
A Densely Connected Criss-Cross Attention Network for Document-level Relation Extraction | Papers | HyperAI