情感预测模型的预训练优化方法、装置、设备及介质与流程



1.本技术适用于人工智能技术领域,尤其涉及一种情感预测模型的预训练优化方法、装置、设备及介质。


背景技术:



2.目前,语音情感识别(speech emotion recognition,ser)是数字语音信号处理领域的一个新兴的研究方向,为人机交互开辟了全新的道路,在很多场景中扮演了重要的角。呼叫中心使用ser技术跟踪客户情绪,为其提供更好的服务;在医学领域,基于ser技术的诊断系统能够分析患者的抑郁和痛苦的程度;还有许多其他应用也利用高效的ser系统提高他们的工作效率。
3.人类声音中的情感受到多方面因素,如性别、年龄、说话人、方言和文化的影响,因此如何更好的对情感进行建模一直是研究人员的重点研究方向。如今,基于深度学习(deep leraning)的方法成为了主流。其中,自注意力(self-supervised)预训练模型提供了一种高性能的解决方案。虽然预训练可以使用大规模异构数据集,获得性能强大且泛用性很好的模型,但是由于预训练任务与目标任务并不完全一致,即预训练域和目标域存在差异,导致预训练模型应用于特定任务的效果不够理想。在ser任务中,通常使用大量未标注的数据进行预训练,需要预训练的模型具备对未标注数据进行较为准确的情感信息提取。因此,如何对预训练模型进行预训练优化,以提高预训练模型对未标注数据的情感信息提取的准确性成为亟待解决的问题。


技术实现要素:



4.有鉴于此,本技术实施例提供了一种情感预测模型的预训练优化方法、装置、设备及介质,以解决如何对预训练模型进行预训练优化,以提高预训练模型对未标注数据的情感信息提取的准确性的问题。
5.第一方面,本技术实施例提供一种情感预测模型的预训练优化方法,所述预训练优化方法包括:
6.针对训练集中任一句级语音,使用无监督情感预测模型提取所述句级语音中每一帧数据对应的帧级特征,将所述句级语音的句级情感标签作为所述帧级特征的情感类别,得到所述训练集中所有帧数据的帧级特征对应的情感类别,所述训练集中句级语音均标注有句级情感标签;
7.对所有情感类别的帧级特征进行聚类,得到n个聚类中心点,n为大于零的整数;
8.将属于同一情感类别的所有帧级特征的均值作为锚点,得到m个锚点,计算所有聚类中心点分别与每个锚点的距离,m为大于零的整数;
9.针对任一聚类中心点,确定与所述聚类中心点的距离最近的锚点为目标锚点,在所述目标锚点满足预设条件时,将所述目标锚点对应的句级情感标签作为所述聚类中心点内所有帧级特征的伪标签,得到所述训练集中所有帧数据的伪标签;
10.以所述训练集中所有帧数据的伪标签为依据,使用所述训练集对自监督情感预测模型进行训练,得到预训练好的自监督情感预测模型,所述无监督情感预测模型和所述自监督情感预测模型均具有时间步长对齐的特征编码器。
11.在一实施方式中,在计算所有聚类中心点分别与每个锚点的距离之后,还包括:
12.针对任一聚类中心点,确定所有与所述聚类中心点的距离小于第一距离阈值的锚点;
13.确定与所述聚类中心点的距离最近的锚点为目标锚点包括:
14.从所有与所述聚类中心点的距离小于所述第一距离阈值的锚点中确定与所述聚类中心点的距离最近的锚点为目标锚点。
15.在一实施方式中,针对任一聚类中心点,确定与所述聚类中心点的距离最近的锚点为目标锚点,在所述目标锚点满足预设条件时,将所述目标锚点对应的句级情感标签作为所述聚类中心点内所有帧级特征的伪标签包括:
16.针对任一聚类中心点,确定与所述聚类中心点的距离最近的锚点为目标锚点;
17.检测所述目标锚点与所述聚类中心点的距离是否小于第二距离阈值;
18.若检测到所述目标锚点与所述聚类中心点的距离小于所述第二距离阈值,则确定所述目标锚点满足预设条件,将所述目标锚点对应的句级情感标签作为所述聚类中心点内所有帧级特征的伪标签。
19.在一实施方式中,在检测所述目标锚点与所述聚类中心点的距离是否小于第二距离阈值之后,还包括:
20.若检测到所述目标锚点与所述聚类中心点的距离不小于所述第二距离阈值,则确定所述目标锚点不满足预设条件,创建其他类锚点,将所述其他类锚点所定义的句级情感标签作为所述聚类中心点内所有帧级特征的伪标签。
21.在一实施方式中,在以所述训练集中所有帧数据的伪标签为依据,使用所述训练集对自监督情感预测模型进行训练,得到预训练好的自监督情感预测模型之后,还包括:
22.针对训练集中任一句级语音,使用所述预训练好的自监督情感预测模型提取所述句级语音中每一帧数据对应的更新帧级特征,将所述句级语音的句级情感标签作为所述更新帧级特征的情感类别,得到所述训练集中所有帧数据的更新帧级特征对应的情感类别;
23.对所有情感类别的更新帧级特征进行聚类,得到n个更新聚类中心点,n为大于零的整数;
24.将属于同一情感类别的所有更新帧级特征的均值作为锚点,得到m个锚点,计算所有更新聚类中心点分别与每个锚点的距离,m为大于零的整数;
25.针对任一更新聚类中心点,确定与所述更新聚类中心点的距离最近的锚点为目标锚点,在所述目标锚点满足预设条件时,将所述目标锚点对应的句级情感标签作为所述更新聚类中心点内所有更新帧级特征的伪标签,得到所述训练集中所有帧数据的更新伪标签;
26.以所述训练集中所有帧数据的更新伪标签为依据,使用所述训练集对自监督情感预测模型进行训练,得到训练好的自监督情感预测模型。
27.在一实施方式中,所述无监督情感预测模型包括第一特征编码器、双向lstm层和第一全连接层,所述第一特征编码器由cnn层构成,所述自监督情感预测模型包括第二特征
编码器、transformer层和第二全连接层,所述第二特征编码器由cnn层构成,所述第一特征编码器中cnn层的个数与所述第二特征编码器中cnn层的个数为相同;在针对训练集中任一句级语音,使用无监督情感预测模型提取所述句级语音中每一帧数据对应的帧级特征之前,还包括:
28.使用特征集对无监督情感预测模型进行训练,所述特征集包括句级语音样本及其对应的情感标签和所属情感类别标签;
29.对所述第一全连接层输出量进行softmax函数归一化,得到预测值;
30.通过第一交叉熵函数进行损失度量,重复使用特征集对无监督情感预测模型进行训练的步骤,直至损失度量的结果满足预设条件,得到训练好的无监督情感预测模型,其中,第一交叉熵函数lg包括:
[0031][0032]
式中,z为样本总数,c为总的情感类别个数,yi表示对应样本i所属的情感类别标签,cj表示情感标签,p(cj|xi)表示对应输入特征xi的cj类的后验概率预测值。
[0033]
在一实施方式中,所述第二全连接层包括两层的全连接层,所述训练集中所有帧数据的伪标签为依据,使用所述训练集对自监督情感预测模型进行训练,得到预训练好的自监督情感预测模型包括:
[0034]
使用所述训练集中所有帧数据及其对应的伪标签对自监督情感预测模型进行训练;
[0035]
通过第二交叉熵函数进行损失度量,重复使用所述训练集中所有帧数据及其对应的伪标签对自监督情感预测模型进行训练的步骤,直至损失度量的结果满足预设条件,得到预训练好的自监督情感预测模型,其中,所述第二交叉熵函数lv包括:
[0036][0037]
式中,表示使用所述特征编码器对帧数据进行编码并输出的低维特征,t表示通过所述第二全连接层遮蔽的遮蔽部分,z
t
表示使用transformer层提取的所述遮蔽部分的上下文表示,表示所述遮蔽部分上下文的后验概率预测值。
[0038]
第二方面,本技术实施例提供一种情感预测模型的预训练优化装置,所述预训练优化装置包括:
[0039]
无监督训练模块,用于针对训练集中任一句级语音,使用无监督情感预测模型提取所述句级语音中每一帧数据对应的帧级特征,将所述句级语音的句级情感标签作为所述帧级特征的情感类别,得到所述训练集中所有帧数据的帧级特征对应的情感类别,所述训练集中句级语音均标注有句级情感标签;
[0040]
特征聚类模块,用于对所有情感类别的帧级特征进行聚类,得到n个聚类中心点,n为大于零的整数;
[0041]
特征计算模块,用于将属于同一情感类别的所有帧级特征的均值作为锚点,得到m个锚点,计算所有聚类中心点分别与每个锚点的距离,m为大于零的整数;
[0042]
伪标签确定模块,用于针对任一聚类中心点,确定与所述聚类中心点的距离最近
的锚点为目标锚点,在所述目标锚点满足预设条件时,将所述目标锚点对应的句级情感标签作为所述聚类中心点内所有帧级特征的伪标签,得到所述训练集中所有帧数据的伪标签;
[0043]
自监督训练模块,用于以所述训练集中所有帧数据的伪标签为依据,使用所述训练集对自监督情感预测模型进行训练,得到预训练好的自监督情感预测模型,所述无监督情感预测模型和所述自监督情感预测模型均具有时间步长对齐的特征编码器。
[0044]
在一实施方式中,所述预训练优化装置还包括:
[0045]
锚点筛选单元,用于在计算所有聚类中心点分别与每个锚点的距离之后,针对任一聚类中心点,确定所有与所述聚类中心点的距离小于第一距离阈值的锚点;
[0046]
所述伪标签确定模块包括:
[0047]
第一目标锚点确定单元,用于从所有与所述聚类中心点的距离小于所述第一距离阈值的锚点中确定与所述聚类中心点的距离最近的锚点为目标锚点。
[0048]
在一实施方式中,所述伪标签确定模块包括:
[0049]
第二目标锚点确定单元,用于针对任一聚类中心点,确定与所述聚类中心点的距离最近的锚点为目标锚点;
[0050]
距离检测单元,用于检测所述目标锚点与所述聚类中心点的距离是否小于第二距离阈值;
[0051]
第一伪标签确定单元,用于若检测到所述目标锚点与所述聚类中心点的距离小于所述第二距离阈值,则确定所述目标锚点满足预设条件,将所述目标锚点对应的句级情感标签作为所述聚类中心点内所有帧级特征的伪标签。
[0052]
在一实施方式中,所述预训练优化装置还包括:
[0053]
第二伪标签确定单元,用于在检测所述目标锚点与所述聚类中心点的距离是否小于第二距离阈值之后,若检测到所述目标锚点与所述聚类中心点的距离不小于所述第二距离阈值,则确定所述目标锚点不满足预设条件,创建其他类锚点,将所述其他类锚点所定义的句级情感标签作为所述聚类中心点内所有帧级特征的伪标签。
[0054]
在一实施方式中,所述预训练优化装置还包括:
[0055]
微调模块,所述微调模块具体用于:
[0056]
在以所述训练集中所有帧数据的伪标签为依据,使用所述训练集对自监督情感预测模型进行训练,得到预训练好的自监督情感预测模型之后,针对训练集中任一句级语音,使用所述预训练好的自监督情感预测模型提取所述句级语音中每一帧数据对应的更新帧级特征,将所述句级语音的句级情感标签作为所述更新帧级特征的情感类别,得到所述训练集中所有帧数据的更新帧级特征对应的情感类别;
[0057]
对所有情感类别的更新帧级特征进行聚类,得到n个更新聚类中心点;
[0058]
将属于同一情感类别的所有更新帧级特征的均值作为锚点,得到m个锚点,计算所有更新聚类中心点分别与每个锚点的距离;
[0059]
针对任一更新聚类中心点,确定与所述更新聚类中心点的距离最近的锚点为目标锚点,在所述目标锚点满足预设条件时,将所述目标锚点对应的句级情感标签作为所述更新聚类中心点内所有更新帧级特征的伪标签,得到所述训练集中所有帧数据的更新伪标签;
[0060]
以所述训练集中所有帧数据的更新伪标签为依据,使用所述训练集对自监督情感预测模型进行训练,得到训练好的自监督情感预测模型。
[0061]
在一实施方式中,所述无监督情感预测模型包括第一特征编码器、双向lstm层和第一全连接层,所述第一特征编码器由cnn层构成,所述自监督情感预测模型包括第二特征编码器、transformer层和第二全连接层,所述第二特征编码器由cnn层构成,所述第一特征编码器中cnn层的个数与所述第二特征编码器中cnn层的个数为相同;所述预训练优化装置还包括:
[0062]
训练模块,用于在针对训练集中任一句级语音,使用无监督情感预测模型提取所述句级语音中每一帧数据对应的帧级特征之前,使用特征集对无监督情感预测模型进行训练,所述特征集包括句级语音样本及其对应的情感标签和所属情感类别标签;
[0063]
归一化模块,用于对所述第一全连接层输出量进行softmax函数归一化,得到预测值;
[0064]
返回执行模块,用于通过第一交叉熵函数进行损失度量,重复使用特征集对无监督情感预测模型进行训练的步骤,直至损失度量的结果满足预设条件,得到训练好的无监督情感预测模型,其中,第一交叉熵函数lg包括:
[0065][0066]
式中,z为样本总数,c为总的情感类别个数,yi表示对应样本i所属的情感类别标签,cj表示情感标签,p(cj|xi)表示对应输入特征xi的cj类的后验概率预测值。
[0067]
在一实施方式中,所述第二全连接层包括两层的全连接层,所述自监督训练模块具体用于:
[0068]
使用所述训练集中所有帧数据及其对应的伪标签对自监督情感预测模型进行训练;
[0069]
通过第二交叉熵函数进行损失度量,重复使用所述训练集中所有帧数据及其对应的伪标签对自监督情感预测模型进行训练的步骤,直至损失度量的结果满足预设条件,得到预训练好的自监督情感预测模型,其中,所述第二交叉熵函数lv包括:
[0070][0071]
式中,表示使用所述特征编码器对帧数据进行编码并输出的低维特征,t表示通过所述第二全连接层遮蔽的遮蔽部分,z
t
表示使用transformer层提取的所述遮蔽部分的上下文表示,表示所述遮蔽部分上下文的后验概率预测值。
[0072]
第三方面,本技术实施例提供一种计算机设备,所述计算机设备包括处理器、存储器以及存储在所述存储器中并可在所述处理器上运行的计算机程序,所述处理器执行所述计算机程序时实现如第一方面所述的预训练优化方法。
[0073]
第四方面,本技术实施例提供一种计算机可读存储介质,所述计算机可读存储介质存储有计算机程序,所述计算机程序被处理器执行时实现如第一方面所述的预训练优化方法。
[0074]
本技术实施例与现有技术相比存在的有益效果是:本技术针对训练集中任一句级
等仅用于区分描述,而不能理解为指示或暗示相对重要性。
[0085]
在本技术说明书中描述的参考“一个实施例”或“一些实施例”等意味着在本技术的一个或多个实施例中包括结合该实施例描述的特定特征、结构或特点。由此,在本说明书中的不同之处出现的语句“在一个实施例中”、“在一些实施例中”、“在其他一些实施例中”、“在另外一些实施例中”等不是必然都参考相同的实施例,而是意味着“一个或多个但不是所有的实施例”,除非是以其他方式另外特别强调。术语“包括”、“包含”、“具有”及它们的变形都意味着“包括但不限于”,除非是以其他方式另外特别强调。
[0086]
本技术实施例可以基于人工智能技术对相关的数据进行获取和处理。其中,人工智能(artificial intelligence,ai)是利用数字计算机或者数字计算机控制的机器模拟、延伸和扩展人的智能,感知环境、获取知识并使用知识获得最佳结果的理论、方法、技术及应用系统。
[0087]
人工智能基础技术一般包括如传感器、专用人工智能芯片、云计算、分布式存储、大数据处理技术、操作/交互系统、机电一体化等技术。人工智能软件技术主要包括计算机视觉技术、机器人技术、生物识别技术、语音处理技术、自然语言处理技术以及机器学习/深度学习等几大方向。
[0088]
应理解,以下实施例中各步骤的序号的大小并不意味着执行顺序的先后,各过程的执行顺序应以其功能和内在逻辑确定,而不应对本技术实施例的实施过程构成任何限定。
[0089]
为了说明本技术的技术方案,下面通过具体实施例来进行说明。
[0090]
本技术实施例一提供的一种情感预测模型的预训练优化方法,可应用在如图1的应用环境中,其中,客户端与服务端进行通信。其中,客户端包括但不限于掌上电脑、桌上型计算机、笔记本电脑、超级移动个人计算机(ultra-mobile personal computer,umpc)、上网本、云端计算机设备、个人数字助理(personal digital assistant,pda)等计算机设备。服务端可以用独立的服务器或者是多个服务器组成的服务器集来实现。
[0091]
参见图2,是本技术实施例二提供的一种情感预测模型的预训练优化方法的流程示意图,上述情感预测模型的预训练优化方法应用于图1中的服务端,服务端对应的计算机设备连接相应的数据库,以获取数据库中相应的训练用数据。上述计算机设备还可以连接相应的客户端,客户端为用户所操作,用户可以通过客户端向服务端提供相应的训练集。如图2所示,该情感预测模型的预训练优化方法可以包括以下步骤:
[0092]
步骤s201,针对训练集中任一句级语音,使用无监督情感预测模型提取句级语音中每一帧数据对应的帧级特征,将句级语音的句级情感标签作为帧级特征的情感类别,得到训练集中所有帧数据的帧级特征对应的情感类别。
[0093]
本技术中,训练集中包括至少一个句级语音,每个句级语音均标注有相应的句级情感标签,句级语音可以是指以句子为单位的一组语音数据,其中的句子可以为一组词语、一段话等,句级情感标签即为一个句子的标签,也即是一组语音数据中每一帧数据均对应一个句级情感标签。
[0094]
无监督情感预测模型可以为wav2vec模型,该wav2vec模型包括特征编码器、一个双向lstm层和一个全连接层,其中,全连接层中可以使用softmax函数进行归一化。
[0095]
其中的特征编码器是用于与自监督情感预测模型进行时间步长的对齐而设定,如
果自监督情感预测模型采用wav2vec2.0模型,该wav2vec2.0模型包含一个多层的cnn网络构成的特征编码器,相应地,在wav2vec模型中的特征编码器也需要具备与其相同层数的cnn网络。
[0096]
帧级特征为每一帧数据的特征,也即是将句级语音分割为一帧帧的语音,对该一帧语音进行特征提取,对应的一个句级语音对应一个情感标签,该句级语音中所有帧语音的特征对应的情感类别为该句级语音的情感标签。
[0097]
步骤s202,对所有情感类别的帧级特征进行聚类,得到n个聚类中心点。
[0098]
本技术中,n为大于零的整数,使用改进的k-means聚类算法对帧级特征进行聚类,将帧级特征聚类为n簇,每一簇中所有帧级特征的均值对应的聚类中心点。
[0099]
具体的可以是,从所有的帧级特征中任意选取n个帧级特征作为初始的聚类中心点,计算每个帧级特征与各个初始聚类中心之间的距离,把每个帧级特征分配给距离它最近的聚类中心点,重新计算每个聚类中心点对应的所有帧级特征的质心,如果新计算出来的质心和原来的聚类中心点之间的距离小于某一个设置的阈值,则表示重新计算的质心的位置变化不大,趋于稳定收敛,如果新的质心和原来的聚类中心点距离变化很大,则需要再次迭代,最终得到n个聚类中心点,也即是n簇。
[0100]
步骤s203,将属于同一情感类别的所有帧级特征的均值作为锚点,得到m个锚点,计算所有聚类中心点分别与每个锚点的距离。
[0101]
本技术中,m为大于零的整数。由于提取到的帧级特征已经能够表示一定的情感信息,所以提取同一情感类别的所有帧级特征的均值作为锚点,可以得到m个锚点,m也表示情感类别的总个数。使用上述的聚类结果,计算每个簇的聚类中心点与各个锚点之间的距离。
[0102]
步骤s204,针对任一聚类中心点,确定与聚类中心点的距离最近的锚点为目标锚点,在目标锚点满足预设条件时,将目标锚点对应的句级情感标签作为聚类中心点内所有帧级特征的伪标签,得到训练集中所有帧数据的伪标签。
[0103]
其中,在确定目标锚点之后,如果将预设条件定义为目标锚点和对应聚类中心点的距离小于特定值,则满足预设条件即为目标锚点与对应聚类中心点的距离小于该特定值。
[0104]
与任一个聚类中心点的距离最近的锚点可能为一个或者多个,当为多个时,预设条件可定义为被选中,即从多个锚点中随机选取一个锚点作为被选中的锚点,也即是满足预设条件的锚点。
[0105]
可选的是,在计算所有聚类中心点分别与每个锚点的距离之后,还包括:
[0106]
针对任一聚类中心点,确定所有与聚类中心点的距离小于第一距离阈值的锚点;
[0107]
确定与聚类中心点的距离最近的锚点为目标锚点包括:
[0108]
从所有与聚类中心点的距离小于第一距离阈值的锚点中确定与聚类中心点的距离最近的锚点为目标锚点。
[0109]
可选的是,针对任一聚类中心点,确定与聚类中心点的距离最近的锚点为目标锚点,在目标锚点满足预设条件时,将目标锚点对应的句级情感标签作为聚类中心点内所有帧级特征的伪标签包括:
[0110]
针对任一聚类中心点,确定与聚类中心点的距离最近的锚点为目标锚点;
[0111]
检测目标锚点与聚类中心点的距离是否小于第二距离阈值;
[0112]
若检测到目标锚点与聚类中心点的距离小于第二距离阈值,则确定目标锚点满足预设条件,将目标锚点对应的句级情感标签作为聚类中心点内所有帧级特征的伪标签。
[0113]
可选的是,在检测目标锚点与聚类中心点的距离是否小于第二距离阈值之后,还包括:
[0114]
若检测到目标锚点与聚类中心点的距离不小于第二距离阈值,则确定目标锚点不满足预设条件,创建其他类锚点,将其他类锚点所定义的句级情感标签作为聚类中心点内所有帧级特征的伪标签。
[0115]
其中,若聚类中心点与锚点之间的距离d
ij
≤γ,则将这个聚类中心点的伪标签映射为该锚点对应的情感类别,γ表示预设的阈值,i∈(0,m),j∈(0,n);若距离大于γ,则创建新的other类锚点,作为非情感类别的统一类别表示,则最后共有m+1个伪标签。
[0116]
步骤s205,以训练集中所有帧数据的伪标签为依据,使用训练集对自监督情感预测模型进行训练,得到预训练好的自监督情感预测模型。
[0117]
本技术中,无监督情感预测模型和自监督情感预测模型均具有时间步长对齐的特征编码器。
[0118]
可选的是,无监督情感预测模型包括第一特征编码器、双向lstm层和第一全连接层,第一特征编码器由cnn层构成,自监督情感预测模型包括第二特征编码器、transformer层和第二全连接层,第二特征编码器由cnn层构成,第一特征编码器中cnn层的个数与第二特征编码器中cnn层的个数为相同;在针对训练集中任一句级语音,使用无监督情感预测模型提取句级语音中每一帧数据对应的帧级特征之前,还包括:
[0119]
使用特征集对无监督情感预测模型进行训练,特征集包括句级语音样本及其对应的情感标签和所属情感类别标签;
[0120]
对第一全连接层输出量进行softmax函数归一化,得到预测值;
[0121]
通过第一交叉熵函数进行损失度量,重复使用特征集对无监督情感预测模型进行训练的步骤,直至损失度量的结果满足预设条件,得到训练好的无监督情感预测模型,其中,第一交叉熵函数lg包括:
[0122][0123]
式中,z为样本总数,c为总的情感类别个数,yi表示对应样本i所属的情感类别标签,cj表示情感标签,p(cj|xi)表示对应输入特征xi的cj类的后验概率预测值。
[0124]
其中,第二特征编码器由7层cnn组成,然后通过transformer层获得上下文表示,再通过由两层全连接层组成的线性多头预测遮蔽部分的伪标签类别。
[0125]
可选的是,第二全连接层包括两层的全连接层,训练集中所有帧数据的伪标签为依据,使用训练集对自监督情感预测模型进行训练,得到预训练好的自监督情感预测模型包括:
[0126]
使用训练集中所有帧数据及其对应的伪标签对自监督情感预测模型进行训练;
[0127]
通过第二交叉熵函数进行损失度量,重复使用训练集中所有帧数据及其对应的伪标签对自监督情感预测模型进行训练的步骤,直至损失度量的结果满足预设条件,得到预训练好的自监督情感预测模型,其中,第二交叉熵函数lv包括:
[0128][0129]
式中,表示使用特征编码器对帧数据进行编码并输出的低维特征,t表示通过第二全连接层遮蔽的遮蔽部分,z
t
表示使用transformer层提取的遮蔽部分的上下文表示,表示遮蔽部分上下文的后验概率预测值。
[0130]
可选的是,在以训练集中所有帧数据的伪标签为依据,使用训练集对自监督情感预测模型进行训练,得到预训练好的自监督情感预测模型之后,还包括:
[0131]
针对训练集中任一句级语音,使用预训练好的自监督情感预测模型提取句级语音中每一帧数据对应的更新帧级特征,将句级语音的句级情感标签作为更新帧级特征的情感类别,得到训练集中所有帧数据的更新帧级特征对应的情感类别;
[0132]
对所有情感类别的更新帧级特征进行聚类,得到n个更新聚类中心点,n为大于零的整数;
[0133]
将属于同一情感类别的所有更新帧级特征的均值作为锚点,得到m个锚点,计算所有更新聚类中心点分别与每个锚点的距离,m为大于零的整数;
[0134]
针对任一更新聚类中心点,确定与更新聚类中心点的距离最近的锚点为目标锚点,在目标锚点满足预设条件时,将目标锚点对应的句级情感标签作为更新聚类中心点内所有更新帧级特征的伪标签,得到训练集中所有帧数据的更新伪标签;
[0135]
以训练集中所有帧数据的更新伪标签为依据,使用训练集对自监督情感预测模型进行训练,得到训练好的自监督情感预测模型。
[0136]
其中,由于伪标签类别与情感类是相关的,因此这种方法可以专注于预测被遮蔽序列的情感信息。预训练结束后,可以直接使用wav2vec2.0替换wav2vec,按照步骤s201至步骤s205对预训练的自监督情感预测模型进行微调,进而得到训练好的自监督情感预测模型。
[0137]
本技术实施例针对训练集中任一句级语音,使用无监督情感预测模型提取句级语音中每一帧数据对应的帧级特征,将句级语音的句级情感标签作为帧级特征的情感类别,得到训练集中所有帧数据的帧级特征对应的情感类别,对所有情感类别的帧级特征进行聚类,得到n个聚类中心点,将属于同一情感类别的所有帧级特征的均值作为锚点,得到m个锚点,计算所有聚类中心点分别与每个锚点的距离,针对任一聚类中心点,确定与聚类中心点的距离最近的锚点为目标锚点,在目标锚点满足预设条件时,将目标锚点对应的句级情感标签作为聚类中心点内所有帧级特征的伪标签,得到训练集中所有帧数据的伪标签,以训练集中所有帧数据的伪标签为依据,使用训练集对自监督情感预测模型进行训练,得到预训练好的自监督情感预测模型,通过聚类进一步加强低维特征与情感信息的相关性,再将聚类分布作为情感伪标签,并以此进行训练,从而提高模型对情感信息的预测准确性。
[0138]
对应于上文实施例的情感预测模型的预训练优化方法,图3示出了本技术实施例三提供的情感预测模型的预训练优化装置的结构框图,上述预训练优化装置应用于图1中的服务端,服务端对应的计算机设备连接相应的数据库,以获取数据库中相应的训练用数据。上述计算机设备还可以连接相应的客户端,客户端为用户所操作,用户可以通过客户端向服务端提供相应的训练集。为了便于说明,仅示出了与本技术实施例相关的部分。
[0139]
参见图3,该预训练优化装置包括:
[0140]
无监督训练模块31,用于针对训练集中任一句级语音,使用无监督情感预测模型提取句级语音中每一帧数据对应的帧级特征,将句级语音的句级情感标签作为帧级特征的情感类别,得到训练集中所有帧数据的帧级特征对应的情感类别;
[0141]
特征聚类模块32,用于对所有情感类别的帧级特征进行聚类,得到n个聚类中心点,n为大于零的整数;
[0142]
特征计算模块33,用于将属于同一情感类别的所有帧级特征的均值作为锚点,得到m个锚点,计算所有聚类中心点分别与每个锚点的距离,m为大于零的整数;
[0143]
伪标签确定模块34,用于针对任一聚类中心点,确定与聚类中心点的距离最近的锚点为目标锚点,在目标锚点满足预设条件时,将目标锚点对应的句级情感标签作为聚类中心点内所有帧级特征的伪标签,得到训练集中所有帧数据的伪标签;
[0144]
自监督训练模块35,用于以训练集中所有帧数据的伪标签为依据,使用训练集对自监督情感预测模型进行训练,得到预训练好的自监督情感预测模型,无监督情感预测模型和自监督情感预测模型均具有时间步长对齐的特征编码器。
[0145]
可选的是,预训练优化装置还包括:
[0146]
锚点筛选单元,用于在计算所有聚类中心点分别与每个锚点的距离之后,针对任一聚类中心点,确定所有与聚类中心点的距离小于第一距离阈值的锚点;
[0147]
伪标签确定模块34包括:
[0148]
第一目标锚点确定单元,用于从所有与聚类中心点的距离小于第一距离阈值的锚点中确定与聚类中心点的距离最近的锚点为目标锚点。
[0149]
可选的是,伪标签确定模块34包括:
[0150]
第二目标锚点确定单元,用于针对任一聚类中心点,确定与聚类中心点的距离最近的锚点为目标锚点;
[0151]
距离检测单元,用于检测目标锚点与聚类中心点的距离是否小于第二距离阈值;
[0152]
第一伪标签确定单元,用于若检测到目标锚点与聚类中心点的距离小于第二距离阈值,则确定目标锚点满足预设条件,将目标锚点对应的句级情感标签作为聚类中心点内所有帧级特征的伪标签。
[0153]
可选的是,预训练优化装置还包括:
[0154]
第二伪标签确定单元,用于在检测目标锚点与聚类中心点的距离是否小于第二距离阈值之后,若检测到目标锚点与聚类中心点的距离不小于第二距离阈值,则确定目标锚点不满足预设条件,创建其他类锚点,将其他类锚点所定义的句级情感标签作为聚类中心点内所有帧级特征的伪标签。
[0155]
可选的是,预训练优化装置还包括:
[0156]
微调模块,微调模块具体用于:
[0157]
在以训练集中所有帧数据的伪标签为依据,使用训练集对自监督情感预测模型进行训练,得到预训练好的自监督情感预测模型之后,针对训练集中任一句级语音,使用预训练好的自监督情感预测模型提取句级语音中每一帧数据对应的更新帧级特征,将句级语音的句级情感标签作为更新帧级特征的情感类别,得到训练集中所有帧数据的更新帧级特征对应的情感类别;
[0158]
对所有情感类别的更新帧级特征进行聚类,得到n个更新聚类中心点;
[0159]
将属于同一情感类别的所有更新帧级特征的均值作为锚点,得到m个锚点,计算所有更新聚类中心点分别与每个锚点的距离;
[0160]
针对任一更新聚类中心点,确定与更新聚类中心点的距离最近的锚点为目标锚点,在目标锚点满足预设条件时,将目标锚点对应的句级情感标签作为更新聚类中心点内所有更新帧级特征的伪标签,得到训练集中所有帧数据的更新伪标签;
[0161]
以训练集中所有帧数据的更新伪标签为依据,使用训练集对自监督情感预测模型进行训练,得到训练好的自监督情感预测模型。
[0162]
可选的是,无监督情感预测模型包括第一特征编码器、双向lstm层和第一全连接层,第一特征编码器由cnn层构成,自监督情感预测模型包括第二特征编码器、transformer层和第二全连接层,第二特征编码器由cnn层构成,第一特征编码器中cnn层的个数与第二特征编码器中cnn层的个数为相同;预训练优化装置还包括:
[0163]
训练模块,用于在针对训练集中任一句级语音,使用无监督情感预测模型提取句级语音中每一帧数据对应的帧级特征之前,使用特征集对无监督情感预测模型进行训练,特征集包括句级语音样本及其对应的情感标签和所属情感类别标签;
[0164]
归一化模块,用于对第一全连接层输出量进行softmax函数归一化,得到预测值;
[0165]
返回执行模块,用于通过第一交叉熵函数进行损失度量,重复使用特征集对无监督情感预测模型进行训练的步骤,直至损失度量的结果满足预设条件,得到训练好的无监督情感预测模型,其中,第一交叉熵函数lg包括:
[0166][0167]
式中,z为样本总数,c为总的情感类别个数,yi表示对应样本i所属的情感类别标签,cj表示情感标签,p(cj|xi)表示对应输入特征xi的cj类的后验概率预测值。
[0168]
可选的是,第二全连接层包括两层的全连接层,自监督训练模块35具体用于:
[0169]
使用训练集中所有帧数据及其对应的伪标签对自监督情感预测模型进行训练;
[0170]
通过第二交叉熵函数进行损失度量,重复使用训练集中所有帧数据及其对应的伪标签对自监督情感预测模型进行训练的步骤,直至损失度量的结果满足预设条件,得到预训练好的自监督情感预测模型,其中,第二交叉熵函数lv包括:
[0171][0172]
式中,表示使用特征编码器对帧数据进行编码并输出的低维特征,t表示通过第二全连接层遮蔽的遮蔽部分,z
t
表示使用transformer层提取的遮蔽部分的上下文表示,表示遮蔽部分上下文的后验概率预测值。
[0173]
需要说明的是,上述模块之间的信息交互、执行过程等内容,由于与本技术方法实施例基于同一构思,其具体功能及带来的技术效果,具体可参见方法实施例部分,此处不再赘述。
[0174]
图4为本技术实施例四提供的一种计算机设备的结构示意图。如图4所示,该实施例的计算机设备包括:至少一个处理器(图4中仅示出一个)、存储器以及存储在存储器中并可在至少一个处理器上运行的计算机程序,处理器执行计算机程序时实现上述任意各个情
感预测模型的预训练优化方法实施例中的步骤。
[0175]
该计算机设备可包括,但不仅限于,处理器、存储器。本领域技术人员可以理解,图4仅仅是计算机设备的举例,并不构成对计算机设备的限定,计算机设备可以包括比图示更多或更少的部件,或者组合某些部件,或者不同的部件,例如还可以包括网络接口、显示屏和输入装置等。
[0176]
所称处理器可以是cpu,该处理器还可以是其他通用处理器、数字信号处理器(digital signal processor,dsp)、专用集成电路(application specific integrated circuit,asic)、现成可编程门阵列(field-programmable gate array,fpga)或者其他可编程逻辑器件、分立门或者晶体管逻辑器件、分立硬件组件等。通用处理器可以是微处理器或者该处理器也可以是任何常规的处理器等。
[0177]
存储器包括可读存储介质、内存储器等,其中,内存储器可以是计算机设备的内存,内存储器为可读存储介质中的操作系统和计算机可读指令的运行提供环境。可读存储介质可以是计算机设备的硬盘,在另一些实施例中也可以是计算机设备的外部存储设备,例如,计算机设备上配备的插接式硬盘、智能存储卡(smart media card,smc)、安全数字(secure digital,sd)卡、闪存卡(flash card)等。进一步地,存储器还可以既包括计算机设备的内部存储单元也包括外部存储设备。存储器用于存储操作系统、应用程序、引导装载程序(bootloader)、数据以及其他程序等,该其他程序如计算机程序的程序代码等。存储器还可以用于暂时地存储已经输出或者将要输出的数据。
[0178]
所属领域的技术人员可以清楚地了解到,为了描述的方便和简洁,仅以上述各功能单元、模块的划分进行举例说明,实际应用中,可以根据需要而将上述功能分配由不同的功能单元、模块完成,即将装置的内部结构划分成不同的功能单元或模块,以完成以上描述的全部或者部分功能。实施例中的各功能单元、模块可以集成在一个处理单元中,也可以是各个单元单独物理存在,也可以两个或两个以上单元集成在一个单元中,上述集成的单元既可以采用硬件的形式实现,也可以采用软件功能单元的形式实现。另外,各功能单元、模块的具体名称也只是为了便于相互区分,并不用于限制本技术的保护范围。上述装置中单元、模块的具体工作过程,可以参考前述方法实施例中的对应过程,在此不再赘述。集成的单元如果以软件功能单元的形式实现并作为独立的产品销售或使用时,可以存储在一个计算机可读取存储介质中。基于这样的理解,本技术实现上述实施例方法中的全部或部分流程,可以通过计算机程序来指令相关的硬件来完成,计算机程序可存储于一计算机可读存储介质中,该计算机程序在被处理器执行时,可实现上述方法实施例的步骤。其中,计算机程序包括计算机程序代码,计算机程序代码可以为源代码形式、对象代码形式、可执行文件或某些中间形式等。计算机可读介质至少可以包括:能够携带计算机程序代码的任何实体或装置、记录介质、计算机存储器、只读存储器(read-only memory,rom)、随机存取存储器(random access memory,ram)、电载波信号、电信信号以及软件分发介质。例如u盘、移动硬盘、磁碟或者光盘等。在某些司法管辖区,根据立法和专利实践,计算机可读介质不可以是电载波信号和电信信号。
[0179]
本技术实现上述实施例方法中的全部或部分流程,也可以通过一种计算机程序产品来完成,当计算机程序产品在计算机设备上运行时,使得计算机设备执行时实现可实现上述方法实施例中的步骤。
[0180]
在上述实施例中,对各个实施例的描述都各有侧重,某个实施例中没有详述或记载的部分,可以参见其它实施例的相关描述。
[0181]
本领域普通技术人员可以意识到,结合本文中所公开的实施例描述的各示例的单元及算法步骤,能够以电子硬件、或者计算机软件和电子硬件的结合来实现。这些功能究竟以硬件还是软件方式来执行,取决于技术方案的特定应用和设计约束条件。专业技术人员可以对每个特定的应用来使用不同方法来实现所描述的功能,但是这种实现不应认为超出本技术的范围。
[0182]
在本技术所提供的实施例中,应该理解到,所揭露的装置/计算机设备和方法,可以通过其它的方式实现。例如,以上所描述的装置/计算机设备实施例仅仅是示意性的,例如,模块或单元的划分,仅仅为一种逻辑功能划分,实际实现时可以有另外的划分方式,例如多个单元或组件可以结合或者可以集成到另一个系统,或一些特征可以忽略,或不执行。另一点,所显示或讨论的相互之间的耦合或直接耦合或通讯连接可以是通过一些接口,装置或单元的间接耦合或通讯连接,可以是电性,机械或其它的形式。
[0183]
作为分离部件说明的单元可以是或者也可以不是物理上分开的,作为单元显示的部件可以是或者也可以不是物理单元,即可以位于一个地方,或者也可以分布到多个网络单元上。可以根据实际的需要选择其中的部分或者全部单元来实现本实施例方案的目的。
[0184]
以上实施例仅用以说明本技术的技术方案,而非对其限制;尽管参照前述实施例对本技术进行了详细的说明,本领域的普通技术人员应当理解:其依然可以对前述各实施例所记载的技术方案进行修改,或者对其中部分技术特征进行等同替换;而这些修改或者替换,并不使相应技术方案的本质脱离本技术各实施例技术方案的精神和范围,均应包含在本技术的保护范围之内。

技术特征:


1.一种情感预测模型的预训练优化方法,其特征在于,所述预训练优化方法包括:针对训练集中任一句级语音,使用无监督情感预测模型提取所述句级语音中每一帧数据对应的帧级特征,将所述句级语音的句级情感标签作为所述帧级特征的情感类别,得到所述训练集中所有帧数据的帧级特征对应的情感类别,所述训练集中句级语音均标注有句级情感标签;对所有情感类别的帧级特征进行聚类,得到n个聚类中心点,n为大于零的整数;将属于同一情感类别的所有帧级特征的均值作为锚点,得到m个锚点,计算所有聚类中心点分别与每个锚点的距离,m为大于零的整数;针对任一聚类中心点,确定与所述聚类中心点的距离最近的锚点为目标锚点,在所述目标锚点满足预设条件时,将所述目标锚点对应的句级情感标签作为所述聚类中心点内所有帧级特征的伪标签,得到所述训练集中所有帧数据的伪标签;以所述训练集中所有帧数据的伪标签为依据,使用所述训练集对自监督情感预测模型进行训练,得到预训练好的自监督情感预测模型,所述无监督情感预测模型和所述自监督情感预测模型均具有时间步长对齐的特征编码器。2.根据权利要求1所述的预训练优化方法,其特征在于,在计算所有聚类中心点分别与每个锚点的距离之后,还包括:针对任一聚类中心点,确定所有与所述聚类中心点的距离小于第一距离阈值的锚点;确定与所述聚类中心点的距离最近的锚点为目标锚点包括:从所有与所述聚类中心点的距离小于所述第一距离阈值的锚点中确定与所述聚类中心点的距离最近的锚点为目标锚点。3.根据权利要求1所述的预训练优化方法,其特征在于,针对任一聚类中心点,确定与所述聚类中心点的距离最近的锚点为目标锚点,在所述目标锚点满足预设条件时,将所述目标锚点对应的句级情感标签作为所述聚类中心点内所有帧级特征的伪标签包括:针对任一聚类中心点,确定与所述聚类中心点的距离最近的锚点为目标锚点;检测所述目标锚点与所述聚类中心点的距离是否小于第二距离阈值;若检测到所述目标锚点与所述聚类中心点的距离小于所述第二距离阈值,则确定所述目标锚点满足预设条件,将所述目标锚点对应的句级情感标签作为所述聚类中心点内所有帧级特征的伪标签。4.根据权利要求3所述的预训练优化方法,其特征在于,在检测所述目标锚点与所述聚类中心点的距离是否小于第二距离阈值之后,还包括:若检测到所述目标锚点与所述聚类中心点的距离不小于所述第二距离阈值,则确定所述目标锚点不满足预设条件,创建其他类锚点,将所述其他类锚点所定义的句级情感标签作为所述聚类中心点内所有帧级特征的伪标签。5.根据权利要求1所述的预训练优化方法,其特征在于,在以所述训练集中所有帧数据的伪标签为依据,使用所述训练集对自监督情感预测模型进行训练,得到预训练好的自监督情感预测模型之后,还包括:针对训练集中任一句级语音,使用所述预训练好的自监督情感预测模型提取所述句级语音中每一帧数据对应的更新帧级特征,将所述句级语音的句级情感标签作为所述更新帧级特征的情感类别,得到所述训练集中所有帧数据的更新帧级特征对应的情感类别;
对所有情感类别的更新帧级特征进行聚类,得到n个更新聚类中心点,n为大于零的整数;将属于同一情感类别的所有更新帧级特征的均值作为锚点,得到m个锚点,计算所有更新聚类中心点分别与每个锚点的距离,m为大于零的整数;针对任一更新聚类中心点,确定与所述更新聚类中心点的距离最近的锚点为目标锚点,在所述目标锚点满足预设条件时,将所述目标锚点对应的句级情感标签作为所述更新聚类中心点内所有更新帧级特征的伪标签,得到所述训练集中所有帧数据的更新伪标签;以所述训练集中所有帧数据的更新伪标签为依据,使用所述训练集对自监督情感预测模型进行训练,得到训练好的自监督情感预测模型。6.根据权利要求1至5任一项所述的预训练优化方法,其特征在于,所述无监督情感预测模型包括第一特征编码器、双向lstm层和第一全连接层,所述第一特征编码器由cnn层构成,所述自监督情感预测模型包括第二特征编码器、transformer层和第二全连接层,所述第二特征编码器由cnn层构成,所述第一特征编码器中cnn层的个数与所述第二特征编码器中cnn层的个数为相同;在针对训练集中任一句级语音,使用无监督情感预测模型提取所述句级语音中每一帧数据对应的帧级特征之前,还包括:使用特征集对无监督情感预测模型进行训练,所述特征集包括句级语音样本及其对应的情感标签和所属情感类别标签;对所述第一全连接层输出量进行softmax函数归一化,得到预测值;通过第一交叉熵函数进行损失度量,重复使用特征集对无监督情感预测模型进行训练的步骤,直至损失度量的结果满足预设条件,得到训练好的无监督情感预测模型,其中,第一交叉熵函数l
g
包括:式中,z为样本总数,c为总的情感类别个数,y
i
表示对应样本i所属的情感类别标签,c
j
表示情感标签,p(c
j
|x
i
)表示对应输入特征x
i
的c
j
类的后验概率预测值。7.根据权利要求6所述的预训练优化方法,其特征在于,所述第二全连接层包括两层的全连接层,所述训练集中所有帧数据的伪标签为依据,使用所述训练集对自监督情感预测模型进行训练,得到预训练好的自监督情感预测模型包括:使用所述训练集中所有帧数据及其对应的伪标签对自监督情感预测模型进行训练;通过第二交叉熵函数进行损失度量,重复使用所述训练集中所有帧数据及其对应的伪标签对自监督情感预测模型进行训练的步骤,直至损失度量的结果满足预设条件,得到预训练好的自监督情感预测模型,其中,所述第二交叉熵函数l
v
包括:式中,表示使用所述特征编码器对帧数据进行编码并输出的低维特征,t表示通过所述第二全连接层遮蔽的遮蔽部分,z
t
表示使用transformer层提取的所述遮蔽部分的上下文表示,表示所述遮蔽部分上下文的后验概率预测值。8.一种情感预测模型的预训练优化装置,其特征在于,所述预训练优化装置包括:
无监督训练模块,用于针对训练集中任一句级语音,使用无监督情感预测模型提取所述句级语音中每一帧数据对应的帧级特征,将所述句级语音的句级情感标签作为所述帧级特征的情感类别,得到所述训练集中所有帧数据的帧级特征对应的情感类别,所述训练集中句级语音均标注有句级情感标签;特征聚类模块,用于对所有情感类别的帧级特征进行聚类,得到n个聚类中心点,n为大于零的整数;特征计算模块,用于将属于同一情感类别的所有帧级特征的均值作为锚点,得到m个锚点,计算所有聚类中心点分别与每个锚点的距离,m为大于零的整数;伪标签确定模块,用于针对任一聚类中心点,确定与所述聚类中心点的距离最近的锚点为目标锚点,在所述目标锚点满足预设条件时,将所述目标锚点对应的句级情感标签作为所述聚类中心点内所有帧级特征的伪标签,得到所述训练集中所有帧数据的伪标签;自监督训练模块,用于以所述训练集中所有帧数据的伪标签为依据,使用所述训练集对自监督情感预测模型进行训练,得到预训练好的自监督情感预测模型,所述无监督情感预测模型和所述自监督情感预测模型均具有时间步长对齐的特征编码器。9.一种计算机设备,其特征在于,所述计算机设备包括处理器、存储器以及存储在所述存储器中并可在所述处理器上运行的计算机程序,所述处理器执行所述计算机程序时实现如权利要求1至7任一项所述的预训练优化方法。10.一种计算机可读存储介质,所述计算机可读存储介质存储有计算机程序,其特征在于,所述计算机程序被处理器执行时实现如权利要求1至7任一项所述的预训练优化方法。

技术总结


本申请适用于人工智能技术领域,尤其涉及一种情感预测模型的预训练优化方法、装置、设备及介质。该方法使用无监督情感预测模型提取句级语音的帧级特征,将句级语音的句级情感标签作为帧级特征的情感类别,对所有情感类别的帧级特征进行聚类,得到N个聚类中心点,将属于同一情感类别的所有帧级特征的均值作为锚点,得到M个锚点,计算所有聚类中心点分别与每个锚点的距离,根据聚类中心点和锚点对帧级特征赋予伪标签,以训练集中所有帧数据的伪标签为依据,对自监督情感预测模型进行训练,通过聚类进一步加强低维特征与情感信息的相关性,再将聚类分布作为情感伪标签,并以此进行训练,从而提高模型对情感信息的预测准确性。从而提高模型对情感信息的预测准确性。从而提高模型对情感信息的预测准确性。


技术研发人员:

张之勇 王健宗

受保护的技术使用者:

平安科技(深圳)有限公司

技术研发日:

2022.09.06

技术公布日:

2022/12/8

本文发布于:2024-09-22 13:33:24,感谢您对本站的认可!

本文链接:https://www.17tex.com/tex/2/37948.html

版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。

标签:所述   情感   特征   类中
留言与评论(共有 0 条评论)
   
验证码:
Copyright ©2019-2024 Comsenz Inc.Powered by © 易纺专利技术学习网 豫ICP备2022007602号 豫公网安备41160202000603 站长QQ:729038198 关于我们 投诉建议