4 个月前

DialogueRNN:一种用于对话中情感检测的注意力RNN

DialogueRNN:一种用于对话中情感检测的注意力RNN

摘要

情感检测在对话中的应用是许多场景中不可或缺的一步,包括从聊天记录、社交媒体帖子、辩论中挖掘意见,以及在实时对话中理解消费者反馈等。目前,系统尚未针对对话中的各个参与者进行个体化处理,未能根据每个发言者的特征进行适应。本文介绍了一种基于循环神经网络的新方法,该方法在整个对话过程中跟踪各个参与者的状态,并利用这些信息进行情感分类。我们的模型在两个不同的数据集上显著超越了现有最先进水平。

代码仓库

SenticNet/conv-emotion
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
emotion-recognition-in-conversation-onDialogueRNN
Accuracy: 63.5
Weighted-F1: 63.5
emotion-recognition-in-conversation-on-2DialogueRNN
MAE (Arousal): 0.165
MAE (Expectancy): 0.175
MAE (Power): 7.9
MAE (Valence): 0.168
emotion-recognition-in-conversation-on-cpedDialogueRNN
Accuracy of Sentiment: 48.57
Macro-F1 of Sentiment: 44.11
emotion-recognition-in-conversation-on-meldDialogueRNN
Accuracy: 59.54
Weighted-F1: 57.03

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供