集成胶囊网络的脑电情绪识别

为了充分提取脑电信号多频带的时频信息和保留导联空间分布的位置信息,提出了一种基于集成胶囊网络的情绪识别模型.对预处理过的脑电信号进行小波包特征提取,并将Theta、Alpha、Beta、Gamma四个频带的小波系数能量值填充于根据导联空间分布映射的稀疏矩阵中,拼接构成多频带特征矩阵,通过胶囊网络对特征数据进行训练,对不...

Emotion Recognition of EEG Based on Ensemble CapsNet

In this paper, an emotion classification model based on ensemble CapsNet is proposed in order to obtain the characteristics of multiband information of EEG signals and the spatial information of channel distribution. Firstly, the wavelet packet transform（WPT） is employed to decompose the pre-processed EEG signals into sub-frequency bands（Theta,Alpha, Beta, and Gamma） and then wavelet coefficient energy values are computed as the features in the four frequence bands. Secondly, the features are spliced to form a multiband feature matrix which are trained through the CapsNet and then generate multiple capsule models. Finally, the ensemble CapsNet model is constructed by the ensemble method. The experimental results show that high frequency features have more obvious effect on emotion recognition than those on low frequency bands, and the feature in Gamma band has the best emotional representation among all the independent frequency bands. Furthermore, the emotional feature integrating multiband information has a higher identification accuracy than that of each single band. The proposed method can achieve an average accuracy of 95.11% and 92.43% for two-category and four-category identification tasks respectively, which presents a certain improvement compared with similar research.