高级检索

    魏琛, 陈兰岚, 张傲. 基于集成卷积神经网络的脑电情感识别[J]. 华东理工大学学报(自然科学版), 2019, 45(4): 614-622. DOI: 10.14135/j.cnki.1006-3080.20180416004
    引用本文: 魏琛, 陈兰岚, 张傲. 基于集成卷积神经网络的脑电情感识别[J]. 华东理工大学学报(自然科学版), 2019, 45(4): 614-622. DOI: 10.14135/j.cnki.1006-3080.20180416004
    WEI Chen, CHEN Lanlan, ZHANG Ao. Emotion Recognition of EEG Based on Ensemble Convolutional Neural Networks[J]. Journal of East China University of Science and Technology, 2019, 45(4): 614-622. DOI: 10.14135/j.cnki.1006-3080.20180416004
    Citation: WEI Chen, CHEN Lanlan, ZHANG Ao. Emotion Recognition of EEG Based on Ensemble Convolutional Neural Networks[J]. Journal of East China University of Science and Technology, 2019, 45(4): 614-622. DOI: 10.14135/j.cnki.1006-3080.20180416004

    基于集成卷积神经网络的脑电情感识别

    Emotion Recognition of EEG Based on Ensemble Convolutional Neural Networks

    • 摘要: 采用脑电情感数据集SEED进行情感识别研究,旨在利用深度学习中的卷积神经网络(CNN)自动提取脑电样本的抽象特征,省去人工选择特征与降维的过程。首先,采用小波包变换(WPT)对脑电信号进行6级分解并构成二维结构样本;然后,通过改变网络深度设计了6个深度不同的CNN模型;最后,通过投票法与加权平均法建立集成模型,提高了识别精度。实验结果表明,本文方法对3种情感类别的平均分类精度达到了93.12%,能够满足情感识别的研究需求。

       

      Abstract: Emotional state has an important influence on people's work and life. Electroencephalogram (EEG) signals, as central neurophysiological signals, are closely related to human emotions. This paper uses the public EEG dataset SEED to undergo the emotion recognition research for automatically extracting the abstract features of the EEG samples and avoiding the process of artificial feature selection and dimensionality reduction. Firstly, the wavelet packet transform (WPT) is utilized to perform six-level wavelet packet decomposition on the EEG signals of the subjects and extract the features of the signals and form two-dimensional structure samples. Since the structural depth has an essential effect on the classification ability of the network, we take the depth of the convolutional neural network as the main line to establish six CNN classification models with different depths. Finally, ensemble convolutional neural networks are constructed by means of the voting method and the weighted averaging method, respectively, such that the classification accuracy can be improved. The experimental results show that the CNN with the appropriate depth has excellent classification ability, and the network structure that is too shallow or too deep is not conducive to the improvement of classification performance. The ensemble model constructed by the weighted method can obtain the best classification accuracy and minimum variance among all the classifiers, which illustrates its reliable and stable classification ability. The proposed research scheme in this paper achieves an average classification accuracy of 93.12% for the three emotional categories, which can meet the needs of emotion recognition.

       

    /

    返回文章
    返回