3 个月前

对比与生成使BART成为优秀的对话情感识别器

对比与生成使BART成为优秀的对话情感识别器

摘要

在对话系统中,语句在不同语境下可能表现出相似的语义但截然不同的情感。因此,建模具有说话人依赖性的长距离上下文情感关系,在对话情感识别中具有至关重要的作用。与此同时,区分不同情感类别也颇具挑战,因为它们通常具有语义相近的情感倾向。为此,我们采用监督对比学习(supervised contrastive learning),使各类情感之间相互排斥,从而更有效地识别语义相近的情感。此外,我们引入辅助的响应生成任务,以增强模型对上下文信息的处理能力,进而促使模型在不同语境下准确识别语义相似但情感各异的表达。为实现上述目标,我们选用预训练的编码器-解码器模型 BART 作为骨干网络,因其在理解与生成任务中均表现出色。在四个数据集上的实验结果表明,所提出的模型在对话情感识别任务中显著优于当前最先进的模型。消融实验进一步验证了监督对比损失与生成损失的有效性。

代码仓库

whatissimondoing/cog-bart
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
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

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
对比与生成使BART成为优秀的对话情感识别器 | 论文 | HyperAI超神经