暂无图片
暂无图片
暂无图片
暂无图片
暂无图片

记忆注意力用于多模态情感计算(论文复现)

wei_shuo 2024-07-31
106

记忆注意力用于多模态情感计算(论文复现)

本文所涉及所有资源均在传知代码平台可获取

Table of Contents

情感计算

近年来,社交媒体的快速扩张推动了用户生成内容的大幅增加,特别是视频。跨不同模态的自动情感分析已成为旨在增强人机交互的关键研究领域。情感分析最初专注于提取和分析通过文本传达的情感。然而,随着计算能力的提高和视听通信的兴起,多模态情感分析(MSA)已经出现。这种方法集成了文本,音频和视觉输入,以加深对人类情感的理解。MSA在教育、客户反馈分析、心理健康监测和个性化广告等多个领域都有应用,显著增强了我们在各种沟通形式中解释和应对情感线索的能力。多模态情感分析(MSA)试图整合来自视觉,听觉和文本来源的情感线索来预测说话者的情感状态或倾向。然而,这些模式的情绪信号可能会有微妙的变化。下图示出了MSA的好处,展示了添加视觉和音频信息可以提高在歧义句和不同上下文中情感极性预测的准确性。

在这里插入图片描述

此外,由于现如今大多数的情感计算数据集都是基于英文语言开发的,所以我加入了中文数据集(SIMS, SIMSv2)应用在模型中,以开发适用于国人的情感计算分析模型,在未来,我也计划加入更多小众数据集,以便检测更隐匿的情感,如嫉妒、嘲讽等,使得AI可以更好的服务于社会。

一、概述

本文在提出了两个新的模块,文本增强和记忆模块,应用在多模态情感计算中;这篇文章对两个模块进行讲解和手把手复现教学,解决当下热门的多模态情感计算问题,并展示在MOSI和MOSEI两个数据集上的效果

二、研究背景

随着多模态数据量的迅速增加,多模态情感分析(MSA)越来越受到关注。该方法通过整合不同数据模态间的信息,提高了情感极性提取的准确性,从而实现了信息的全面融合,提高了情感分析的精度。然而,当前的深度模型往往忽视了全局记忆信息,无法有效地辅助情感分析,因此我将介绍一种新的记忆注意力模块;此外现如今的研究表明,文本模态相对于其他模态拥有更丰富和准确的情感信息,因此,我提出了一个新的文本增强模块。

三、模型结构和代码

多模态情感分析是指从视频片段中的多个人体信号中判断人物的情感状态。对于视频片段XX,它通常由三个部分组成,包括 tt(文本)、aa(声学)和 vv(视觉),它们是由二维张量 Xt={Xm1,Xm2,…,XmLm},m∈{t,a,v}X**t={X**m1,X**m2,…,XmL**m},m∈{t,a,v} 表示。𝐿𝑚L**m表示模态m的序列长度,xti,xai,xvixti,xai,xvi 表示每一帧视频、音频和第ii 个字。对于视觉序列,我们使用Facet或Pyfeat获取其原始特征。对于文本序列,我们使用来自预先训练的语言模型的标记器来获得其对应的标记序列 It∈RLtI**tRLt。另一方面,声学序列通过COVAREP处理以获得原始特征。其中,Sa∈RLa×daS**aRLa×d**a 和 Sv∈RLv×dvS**vRLv×d**v表示原始的声学和视觉特征序列。在这里,dvd**v 和 dad**a分别是视觉和声学向量维度。具体来说,受CENet的启发,我们采用kmeans算法将视觉和声学数据的真实的向量序列转换为索引,建立不同的“声学词汇”和“视觉词汇”,以减轻异构模态之间的初始分布差异。随后,我们使用Im={lm0,lm1,…,lmLm−1}I**m={l**m0,l**m1,…,lmL**m−1}表示模态m∈{a,v}m∈{a,v}.

1. 记忆注意力模块(EM)

记忆模块在多模态情感计算中起着关键作用,它允许模型通过注意机制根据查询动态地访问和整合记忆中的信息,从而增强特征的表达能力,使模型能够更准确地理解和捕捉复杂的情感内容。此外,EM可以存储和检索多模态数据中的关键信息,帮助模型减少噪声或不完整数据的不确定性,并提高情感分析任务的准确性和效率,并且通过动态更新和学习如何整合不同模态数据之间的关系,还使模型能够在训练后通过推广到以前看不见的数据来表现良好,保持高效和准确。同时,EM模块应用于所有三种模式(文本、视觉和声学)。

在这里插入图片描述

首先初始化一个可学习的记忆矩阵Et∈Rde×dvE**tRde×d**v,其中 ded**e 表示记忆矩阵的维数,它可以根据模型的训练过程动态地更新参数,用于存储和检索模型的关键信息。之后,我们将对输入的视觉特征的查询 QvQ**v、键 KvK**v 和值 VvV**v进行线性更改,以保留输入维度;

在这里插入图片描述

然后,在我们转移特征维度之后,我们可以使用Q和K计算注意力得分。

在这里插入图片描述

其中,ATT_SvATT_S**v表示注意力得分,我们使用 SoftmaxSoftmax函数来获得注意力权重,并结合 VV进行加权求和,得到最终结果;

在这里插入图片描述

在这个阶段,我们拥有包含关键记忆信息的特征,用于后续的特征交互。
下面是该模块的代码实现

class ExternalMemory(nn.Module): def __init__(self, text_dim, mem_size): super(ExternalMemory, self).__init__() self.text_dim = text_dim self.mem_size = mem_size # 外部记忆矩阵 self.memory = nn.Parameter(torch.randn(mem_size, text_dim)) # Linear transformations self.query_transform = nn.Linear(text_dim, text_dim) self.key_transform = nn.Linear(text_dim, text_dim) self.value_transform = nn.Linear(text_dim, text_dim) def forward(self, query): # 将输入的查询向量query进行线性变换,得到变换后的查询向量。这个变换的目的是将查询向量调整到与记忆矩阵相同的维度,以便进行后续的注意力计算。 query = self.query_transform(query) # (batch_size, text_dim) # 使用self.key_transform(self.memory)和self.value_transform(self.memory)分别对外部记忆矩阵进行键和值的线性变换, # 得到变换后的键矩阵和值矩阵。这些变换将外部记忆中的每个条目映射到与查询向量相同的空间。 key = self.key_transform(self.memory) # (mem_size, text_dim) value = self.value_transform(self.memory) # (mem_size, text_dim) # 通过矩阵乘法torch.matmul(query, key.t())计算查询向量与每个键之间的点积注意力分数。 # 这里使用了缩放点积注意力机制,通过除以math.sqrt(self.text_dim)来缩放注意力分数,其中self.text_dim是查询和键的维度大小 scores = torch.matmul(query, key.t()) / math.sqrt(self.text_dim) # (batch_size, mem_size) # 对注意力分数应用softmax函数F.softmax(scores, dim=-1),得到注意力权重,表示每个外部记忆条目对查询的重要性。 attention_weights = F.softmax(scores, dim=-1) # (batch_size, mem_size) # 使用注意力权重对值矩阵进行加权求和,计算加权平均值。weighted_sum表示基于注意力机制加权后的记忆信息. weighted_sum = torch.matmul(attention_weights.unsqueeze(1), value.unsqueeze(0)) # (batch_size, 1, text_dim) weighted_sum = weighted_sum.squeeze(1) # (batch_size, text_dim) return weighted_sum

2. 文本增强模块

模块的核心组件是跨模态嵌入单元。该单元利用跨模态注意力机制捕捉长程非文本情感信息,并生成基于文本的非语言嵌入。嵌入层的参数是可学习的,用于将经过特征转换策略处理后的非文本索引向量映射到高维空间,进而生成文本模态对非文本模态的注意力权重矩阵。换句话说,这一单元通过复杂的神经网络结构,有效地整合了不同模态之间的信息交互,从而提升了情感分析任务的多模态处理能力。

训练开始阶段,由于语言表征和非语言表征处于不同的特征空间,它们之间的相关性通常较低。因此,注意力权重矩阵中的元素可能较小。为了更有效地学习模型参数,研究者在应用 softmax 函数之前使用超参数来缩放这些注意力权重矩阵。

基于注意力权重矩阵,可以生成基于文本的非语言向量。将基于文本的声学嵌入和基于文本的视觉嵌入结合起来,形成非语言增强嵌入。最后,通过整合非语言增强嵌入来更新文本的表示。因此,该模块的提出旨在为文本提供非语言上下文信息,通过增加非语言增强嵌入来调整文本表示,使其在语义上更加准确和丰富。

六、数据集介绍

1. CMU-MOSI: 它是一个多模态数据集,包括文本、视觉和声学模态。它来自Youtube上的93个电影评论视频。这些视频被剪辑成2199个片段。每个片段都标注了[-3,3]范围内的情感强度。该数据集分为三个部分,训练集(1,284段)、验证集(229段)和测试集(686段)。
2. CMU-MOSEI: 它类似于CMU-MOSI,但规模更大。它包含了来自在线视频网站的23,453个注释视频片段,涵盖了250个不同的主题和1000个不同的演讲者。CMU-MOSEI中的样本被标记为[-3,3]范围内的情感强度和6种基本情绪。因此,CMU-MOSEI可用于情感分析和情感识别任务。

3. SIMS/SIMSV2: CH-SIMS数据集是一个中文多模态情感分析数据集,为每种模态提供了详细的标注。该数据集包括2281个精选视频片段,这些片段来自各种电影、电视剧和综艺节目,每个样本都被赋予了情感分数,范围从-1(极度负面)到1(极度正面)

七、性能展示

在情感计算任务中,通过对比实验和消融实验,可以看到我们的模块使得模型性能明显提高,证明了其有效性;

对比实验:

MOSI数据集:

在这里插入图片描述

MOSEI数据集:

在这里插入图片描述

Case Study

在这里插入图片描述

八、复现过程

在准备好数据集并调试代码后,进行下面的步骤,附件已经调通并修改,可直接正常运行;

  • 数据集准备
    下载MOSI和MOSEI数据集已提取好的特征文件(.pkl)。把它放在"./dataset”目录。
  • 下载预训练语言模型
    下载SentiLARE语言模型文件,然后将它们放入"/pretrained-model / sentilare_model”目录。
  • 下载需要的包
pip install -r requirements.txt

九、运行过程

训练过程

在这里插入图片描述

最终结果

在这里插入图片描述

模型总结

我们的模型适用于多模态情感分析场景,特别是在需要同时处理文本、声音和视觉信息的任务中表现突出。以下是模型的项目特点和适用场景总结:

适用场景

  • 社交媒体监控与分析: 模型能够有效地从社交媒体上的多模态数据中提取情感信息,帮助分析用户在言语、音频和视频内容中的情感表达,用于品牌监控、舆情分析等。
  • 视频内容理解: 在视频内容分析中,可以同时处理视频中的语言文本、音频和视觉信息,提升对视频内容情感和情绪的理解能力,适用于视频推荐系统、内容审核等应用。
  • 电影和电视节目评价: 通过结合电影或电视节目中的对话文本、声音和场景图像,可以更全面地分析观众对作品的情感反应,用于市场调研、口碑分析等。
  • 心理健康监测与干预: 模型的能力使其在心理健康领域具有潜在应用,例如通过分析患者的语音、文本和面部表情数据来识别抑郁症或其他情感健康问题的迹象,为个性化的干预和治疗提供支持。

项目特点

  • 多模态信息融合: 采用记忆注意力机制,能够有效地整合文本、声音和视觉信息,提升情感分析的准确性和丰富性。
  • 自适应学习策略: 模型内部使用超参数来调节注意力权重矩阵,从而在初始训练阶段优化不同模态信息之间的关联性,提升模型性能和参数学习效率。
  • 非语言信息增强: 模块引入了非语言增强嵌入和记忆模块,通过结合基于文本的声学和视觉嵌入来丰富和调整文本的语义表示,提高文本表达的准确性和情感丰富度。
  • 应用领域广泛: 不仅适用于情感分析,CENet还可扩展到其他需要综合多模态信息的任务,如智能会话系统、娱乐内容推荐和情感驱动的虚拟助理等领域。

综上所述,我们的模型以其能够处理多种模态信息并提升情感分析精度的特性,在多个现实世界的应用场景中展现了广泛的潜力和价值。

文章代码资源点击附件获取

「喜欢这篇文章,您的关注和赞赏是给作者最好的鼓励」
关注作者
【版权声明】本文为墨天轮用户原创内容,转载时必须标注文章的来源(墨天轮),文章链接,文章作者等基本信息,否则作者和墨天轮有权追究责任。如果您发现墨天轮中有涉嫌抄袭或者侵权的内容,欢迎发送邮件至:contact@modb.pro进行举报,并提供相关证据,一经查实,墨天轮将立刻删除相关内容。

评论