HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Contrast and Generation Make BART a Good Dialogue Emotion Recognizer

Shimin Li Hang Yan Xipeng Qiu

Contrast and Generation Make BART a Good Dialogue Emotion Recognizer

Abstract

In dialogue systems, utterances with similar semantics may have distinctive emotions under different contexts. Therefore, modeling long-range contextual emotional relationships with speaker dependency plays a crucial part in dialogue emotion recognition. Meanwhile, distinguishing the different emotion categories is non-trivial since they usually have semantically similar sentiments. To this end, we adopt supervised contrastive learning to make different emotions mutually exclusive to identify similar emotions better. Meanwhile, we utilize an auxiliary response generation task to enhance the model's ability of handling context information, thereby forcing the model to recognize emotions with similar semantics in diverse contexts. To achieve these objectives, we use the pre-trained encoder-decoder model BART as our backbone model since it is very suitable for both understanding and generation tasks. The experiments on four datasets demonstrate that our proposed model obtains significantly more favorable results than the state-of-the-art model in dialogue emotion recognition. The ablation study further demonstrates the effectiveness of supervised contrastive loss and generative loss.

Code Repositories

whatissimondoing/cog-bart
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
emotion-recognition-in-conversation-on-3CoG-BART
Micro-F1: 54.71
Weighted F1: 54.71
emotion-recognition-in-conversation-on-4CoG-BART
Micro-F1: 42.58
Weighted-F1: 39.04
emotion-recognition-in-conversation-on-meldCoG-BART
Weighted-F1: 64.81

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