Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Contrast and Generation Make BART a Good Dialogue Emotion Recognizer

About

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.

Shimin Li, Hang Yan, Xipeng Qiu• 2021

Related benchmarks

TaskDatasetResultRank
Emotion Recognition in ConversationIEMOCAP (test)
Weighted Average F1 Score66.18
154
Emotion Recognition in ConversationMELD
Weighted Avg F164.81
137
Conversational Emotion RecognitionIEMOCAP
Weighted Average F1 Score66.18
129
Emotion Recognition in ConversationMELD (test)
Weighted F164.81
118
Emotion DetectionEmoryNLP (test)
Weighted-F10.3904
96
Dialogue Emotion DetectionEmoryNLP
Weighted Avg F139.04
80
Emotion RecognitionIEMOCAP
Accuracy66.71
71
Dialogue Emotion DetectionDailyDialog
Micro F1 (- neutral)0.5552
27
Emotion Recognition in ConversationDailyDialog (test)--
16
Emotion Recognition in ConversationsIEMOCAP (standard split)
Micro F164.1
11
Showing 10 of 11 rows

Other info

Code

Follow for update