HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Supervised Prototypical Contrastive Learning for Emotion Recognition in Conversation

Xiaohui Song Longtao Huang Hui Xue Songlin Hu

Supervised Prototypical Contrastive Learning for Emotion Recognition in Conversation

Abstract

Capturing emotions within a conversation plays an essential role in modern dialogue systems. However, the weak correlation between emotions and semantics brings many challenges to emotion recognition in conversation (ERC). Even semantically similar utterances, the emotion may vary drastically depending on contexts or speakers. In this paper, we propose a Supervised Prototypical Contrastive Learning (SPCL) loss for the ERC task. Leveraging the Prototypical Network, the SPCL targets at solving the imbalanced classification problem through contrastive learning and does not require a large batch size. Meanwhile, we design a difficulty measure function based on the distance between classes and introduce curriculum learning to alleviate the impact of extreme samples. We achieve state-of-the-art results on three widely used benchmarks. Further, we conduct analytical experiments to demonstrate the effectiveness of our proposed SPCL and curriculum learning strategy. We release the code at https://github.com/caskcsg/SPCL.

Code Repositories

caskcsg/spcl
Official
pytorch

Benchmarks

BenchmarkMethodologyMetrics
emotion-recognition-in-conversation-onSPCL-CL-ERC
Weighted-F1: 69.74
emotion-recognition-in-conversation-on-4SPCL-CL-ERC
Weighted-F1: 40.94
emotion-recognition-in-conversation-on-meldSPCL-CL-ERC
Weighted-F1: 67.25

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
Supervised Prototypical Contrastive Learning for Emotion Recognition in Conversation | Papers | HyperAI