HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Representation Iterative Fusion based on Heterogeneous Graph Neural Network for Joint Entity and Relation Extraction

{Kai Gao Xiaoteng Li Yue Cheng Hua Xu Kang Zhao}

Abstract

Joint entity and relation extraction is an essential task in information extraction, which aims to extract all relational triples from unstructured text. However, few existing works consider possible relations information between entities before extracting them, which may lead to the fact that most of the extracted entities cannot constitute valid triples. In this paper, we propose a representation iterative fusion based on heterogeneous graph neural networks for relation extraction (RIFRE). We model relations and words as nodes on the graph and fuse the two types of semantic nodes by the message passing mechanism iteratively to obtain nodes representation that is more suitable for relation extraction tasks. The model performs relation extraction after nodes representation is updated. We evaluate RIFRE on two public relation extraction datasets: NYT and WebNLG. The results show that RIFRE can effectively extract triples and achieve state-of-the-art performance. Moreover, RIFRE is also suitable for the relation classification task, and significantly outperforms the previous methods on SemEval 2010 Task 8 datasets.

Benchmarks

BenchmarkMethodologyMetrics
relation-extraction-on-semeval-2010-task-8RIFRE
F1: 91.3
relation-extraction-on-webnlgRIFRE
F1: 92.6

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
Representation Iterative Fusion based on Heterogeneous Graph Neural Network for Joint Entity and Relation Extraction | Papers | HyperAI