HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

TPLinker: Single-stage Joint Extraction of Entities and Relations Through Token Pair Linking

Yucheng Wang Bowen Yu Yueyang Zhang Tingwen Liu Hongsong Zhu Limin Sun

TPLinker: Single-stage Joint Extraction of Entities and Relations Through Token Pair Linking

Abstract

Extracting entities and relations from unstructured text has attracted increasing attention in recent years but remains challenging, due to the intrinsic difficulty in identifying overlapping relations with shared entities. Prior works show that joint learning can result in a noticeable performance gain. However, they usually involve sequential interrelated steps and suffer from the problem of exposure bias. At training time, they predict with the ground truth conditions while at inference it has to make extraction from scratch. This discrepancy leads to error accumulation. To mitigate the issue, we propose in this paper a one-stage joint extraction model, namely, TPLinker, which is capable of discovering overlapping relations sharing one or both entities while immune from the exposure bias. TPLinker formulates joint extraction as a token pair linking problem and introduces a novel handshaking tagging scheme that aligns the boundary tokens of entity pairs under each relation type. Experiment results show that TPLinker performs significantly better on overlapping and multiple relation extraction, and achieves state-of-the-art performance on two public datasets.

Code Repositories

131250208/TPlinker-joint-extraction
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
relation-extraction-on-nyt10-hrlTPLinker
F1: 71.93
relation-extraction-on-nyt10-hrlTPLinker
F1: 72.45
relation-extraction-on-nyt11-hrlTPLinker
F1: 55.67
relation-extraction-on-nyt11-hrlTPLinker
F1: 55.28
relation-extraction-on-webnlgTPLinker
F1: 91.9

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