论文标题
Stiln:一个新颖的空间信息学习网络,用于基于脑电图的情绪识别
STILN: A Novel Spatial-Temporal Information Learning Network for EEG-based Emotion Recognition
论文作者
论文摘要
在脑电图(EEG)的情绪识别中,空间相关性和时间上下文是必不可少的。但是,学习几个渠道之间复杂的空间相关性是一个具有挑战性的问题。此外,时间上下文学习对强调关键的脑电图框架是有益的,因为受试者只会在刺激的一部分期间达到潜在的情绪。因此,我们提出了一个新型的时空信息学习网络(Stiln),以通过捕获空间相关性和时间环境来提取歧视性特征。具体而言,生成的2D功率地形图捕获了电极之间的依赖项,并将其馈送到基于CNN的空间特征提取网络中。此外,卷积阻滞注意模块(CBAM)重新校准了功率地图的重量,以强调至关重要的大脑区域和频带。同时,适当组合了批准(BNS)和实例归一化(INS)以缓解个体差异。在时间上下文学习中,我们采用双向长短期内存网络(BI-LSTM)网络来捕获脑电图之间的依赖关系。为了验证所提出的方法的有效性,在公共DEAP数据集上进行了独立的实验。所提出的方法达到了未来的表现,唤醒和价分类的准确性分别达到0.6831和0.6752。
The spatial correlations and the temporal contexts are indispensable in Electroencephalogram (EEG)-based emotion recognition. However, the learning of complex spatial correlations among several channels is a challenging problem. Besides, the temporal contexts learning is beneficial to emphasize the critical EEG frames because the subjects only reach the prospective emotion during part of stimuli. Hence, we propose a novel Spatial-Temporal Information Learning Network (STILN) to extract the discriminative features by capturing the spatial correlations and temporal contexts. Specifically, the generated 2D power topographic maps capture the dependencies among electrodes, and they are fed to the CNN-based spatial feature extraction network. Furthermore, Convolutional Block Attention Module (CBAM) recalibrates the weights of power topographic maps to emphasize the crucial brain regions and frequency bands. Meanwhile, Batch Normalizations (BNs) and Instance Normalizations (INs) are appropriately combined to relieve the individual differences. In the temporal contexts learning, we adopt the Bidirectional Long Short-Term Memory Network (Bi-LSTM) network to capture the dependencies among the EEG frames. To validate the effectiveness of the proposed method, subject-independent experiments are conducted on the public DEAP dataset. The proposed method has achieved the outstanding performance, and the accuracies of arousal and valence classification have reached 0.6831 and 0.6752 respectively.