3 个月前

EmoBERTa:基于RoBERTa的对话中具备说话人感知能力的情感识别

EmoBERTa:基于RoBERTa的对话中具备说话人感知能力的情感识别

摘要

我们提出EmoBERTa:一种基于RoBERTa的说话人感知对话情感识别方法,这是一种简洁而富有表现力的解决对话情感识别(ERC)任务的方案。通过在对话话语前简单地添加说话人姓名,并在对话轮次之间插入分隔标记,EmoBERTa能够以端到端的方式学习说话人内部及跨说话人状态与上下文信息,从而预测当前说话人的情感。实验结果表明,我们采用一种基础且直接的方法,在两个主流的ERC数据集上均达到了新的最先进水平。相关代码与模型已开源,地址为:https://github.com/tae898/erc。

代码仓库

tae898/erc
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
emotion-recognition-in-conversation-onEmoBERTa
Weighted-F1: 68.57
emotion-recognition-in-conversation-on-cpedEmoBERTa
Accuracy of Sentiment: 48.09
Macro-F1 of Sentiment: 44.60
emotion-recognition-in-conversation-on-meldEmoBERTa
Weighted-F1: 65.61

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
EmoBERTa:基于RoBERTa的对话中具备说话人感知能力的情感识别 | 论文 | HyperAI超神经