基于 Swin Transformer 的四维脑电情绪识别

来源：计算机技术与发展杂志2023年第12期北京时间：

作者：陈宗楠;金家瑞;潘家辉*

摘要：近年来,基于脑电图( Electroencephalogram,EEG) 的情绪识别研究主要使用卷积神经网络、循环神经网络和深度信念网络模型。 这些方法能利用全局差异来区分不同情绪状态,但忽视了局部脑电的变化对情绪状态的影响。 针对上述问题,使用了一种基于 Swin Transformer 的 EEG 四维脑电情绪识别模型,能够更好地捕捉到细小的局部空间特征和复杂的时间序列特征。 相较于其它情绪识别方法,该模型通过基于滑动窗口的自注意力机制提高了不同块之间的特征连接,使得模型的建模能力更强,也降低了计算的复杂度。 此外,利用情绪脑电公开数据集 SEED 来评估该模型的可行性与有效性,在单被试情绪三分类的准确率为 94. 73% ±1. 72% ,跨被试情绪三分类的准确度为 89. 63% ±3. 42% ,并且测试速度能达到实时处理的水平。 实验结果表明,基于 Swin Transformer 的 EEG 四维脑电情绪识别模型通过局部特征的学习在小样本训练上也能达到较高的情绪分类准确率和较快的测试速度。

关键词：深度学习;情绪识别;脑电图;特征融合;Swin Transformer