HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

A Hierarchical Framework for Relation Extraction with Reinforcement Learning

Ryuichi Takanobu; Tianyang Zhang; Jiexi Liu; Minlie Huang

A Hierarchical Framework for Relation Extraction with Reinforcement Learning

Abstract

Most existing methods determine relation types only after all the entities have been recognized, thus the interaction between relation types and entity mentions is not fully modeled. This paper presents a novel paradigm to deal with relation extraction by regarding the related entities as the arguments of a relation. We apply a hierarchical reinforcement learning (HRL) framework in this paradigm to enhance the interaction between entity mentions and relation types. The whole extraction process is decomposed into a hierarchy of two-level RL policies for relation detection and entity extraction respectively, so that it is more feasible and natural to deal with overlapping relations. Our model was evaluated on public datasets collected via distant supervision, and results show that it gains better performance than existing methods and is more powerful for extracting overlapping relations.

Code Repositories

truthless11/HRL-RE
Official
pytorch
Mentioned in GitHub
AndrewSukhobok95/DL_GraphEntity_project
pytorch
Mentioned in GitHub

Benchmarks

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 Hierarchical Framework for Relation Extraction with Reinforcement Learning | Papers | HyperAI