[发明专利]基于注意力机制的脑电信号情感判别方法及装置

摘要

本发明公开了一种基于注意力机制的脑电信号情感判别方法及装置，该方法包括：(1)获取通过脑电采集设备的多个信号电极采集的脑电信号；(2)提取每个频段信号上的过零计数特征作为脑电特征；(3)构建全局特征提取模型，所述模型包括特征提取分支和注意力向量生成分支和全局特征生成单元；(4)构建带有全连接层的长短时记忆网络LSTM，采用全局特征作为网络输入进行训练，(5)按照步骤(1)‑(3)提取待识别脑电信号的全局特征，并输入训练好的带有全连接层的长短时记忆网络，输出即为识别的情感类别。本发明识别结果更精确。

主权项

1.一种基于注意力机制的脑电信号情感判别方法，其特征在于该方法包括：/n(1)获取用户处于不同情感时的原始脑电信号，所述原始脑电信号通过脑电采集设备的多个信号采集电极分布于用户的脑部不同位置进行采集得到；/n(2)将原始脑电信号去噪后通过带通滤波获取多个频段信号，并提取每个频段信号上的过零计数特征作为脑电特征；/n(3)构建全局特征提取模型，所述模型包括特征提取分支和注意力向量生成分支和全局特征生成单元，所述特征提取分支用于采用两层图卷积对所述脑电特征进行图滤波，得到初级特征，所述注意力向量生成分支用于采用一层图卷积对所述脑电特征处理后再转换为一维注意力列向量，所述全局特征生成单元用于将所述注意力列向量扩充为以向量中元素为对角元素的对角矩阵，并与初级特征点乘得到全局特征；其中，所述图卷积根据脑电采集设备的信号采集电极的邻接关联图生成；/n(4)构建带有全连接层的长短时记忆网络LSTM，采用全局特征作为网络输入进行训练，训练时采用的损失函数为：全局特征对应的实际情感类别和网络输出的情感类别的交叉熵，加上注意力向量正则项的函数；/n(5)按照步骤(1)-(3)提取待识别脑电信号的全局特征，并输入到训练好的带有全连接层的长短时记忆网络，输出为每个情感类别对应的概率值，选取概率最大的情感类别作为预测的情感类型。/n