HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

An Embarrassingly Simple Model for Dialogue Relation Extraction

Fuzhao Xue; Aixin Sun; Hao Zhang; Jinjie Ni; Eng Siong Chng

An Embarrassingly Simple Model for Dialogue Relation Extraction

Abstract

Dialogue relation extraction (RE) is to predict the relation type of two entities mentioned in a dialogue. In this paper, we propose a simple yet effective model named SimpleRE for the RE task. SimpleRE captures the interrelations among multiple relations in a dialogue through a novel input format named BERT Relation Token Sequence (BRS). In BRS, multiple [CLS] tokens are used to capture possible relations between different pairs of entities mentioned in the dialogue. A Relation Refinement Gate (RRG) is then designed to extract relation-specific semantic representation in an adaptive manner. Experiments on the DialogRE dataset show that SimpleRE achieves the best performance, with much shorter training time. Further, SimpleRE outperforms all direct baselines on sentence-level RE without using external resources.

Code Repositories

XueFuzhao/SimpleRE
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
dialog-relation-extraction-on-dialogreSimpleRE
F1 (Chinese): 65.2
F1 (v1): 66.3
F1 (v2): 66.7
F1c (v2): 63.3

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
An Embarrassingly Simple Model for Dialogue Relation Extraction | Papers | HyperAI