EEG-Based Emotion Recognition Using Convolutional Recurrent Neural Network with Multi-Head Self-Attention

被引:22
|
作者
Hu, Zhangfang [1 ]
Chen, Libujie [1 ,2 ]
Luo, Yuan [1 ,2 ]
Zhou, Jingfan [1 ]
机构
[1] Chongqing Univ Posts & Telecommun, Key Lab Optoelect Informat Sensing & Technol, Chongqing 400065, Peoples R China
[2] Chongqing Univ Posts & Telecommun, Sch Adv Mfg Engn, Chongqing 400065, Peoples R China
来源
APPLIED SCIENCES-BASEL | 2022年 / 12卷 / 21期
基金
中国国家自然科学基金;
关键词
EEG; emotion recognition; CNN; BiLSTM; multi-head self-attention; time-frequency map; CLASSIFICATION; DEEP;
D O I
10.3390/app122111255
中图分类号
O6 [化学];
学科分类号
0703 ;
摘要
Featured Application The proposed method in this study can be used in EEG emotion recognition and achieve better results. In recent years, deep learning has been widely used in emotion recognition, but the models and algorithms in practical applications still have much room for improvement. With the development of graph convolutional neural networks, new ideas for emotional recognition based on EEG have arisen. In this paper, we propose a novel deep learning model-based emotion recognition method. First, the EEG signal is spatially filtered by using the common spatial pattern (CSP), and the filtered signal is converted into a time-frequency map by continuous wavelet transform (CWT). This is used as the input data of the network; then the feature extraction and classification are performed by the deep learning model. We called this model CNN-BiLSTM-MHSA, which consists of a convolutional neural network (CNN), bi-directional long and short-term memory network (BiLSTM), and multi-head self-attention (MHSA). This network is capable of learning the time series and spatial information of EEG emotion signals in depth, smoothing EEG signals and extracting deep features with CNN, learning emotion information of future and past time series with BiLSTM, and improving recognition accuracy with MHSA by reassigning weights to emotion features. Finally, we conducted experiments on the DEAP dataset for sentiment classification, and the experimental results showed that the method has better results than the existing classification. The accuracy of high and low valence, arousal, dominance, and liking state recognition is 98.10%, and the accuracy of four classifications of high and low valence-arousal recognition is 89.33%.
引用
收藏
页数:19
相关论文
共 50 条
  • [21] Continuous Convolutional Neural Network with 3D Input for EEG-Based Emotion Recognition
    Yang, Yilong
    Wu, Qingfeng
    Fu, Yazhen
    Chen, Xiaowei
    NEURAL INFORMATION PROCESSING (ICONIP 2018), PT VII, 2018, 11307 : 433 - 443
  • [22] Music emotion recognition based on temporal convolutional attention network using EEG
    Qiao, Yinghao
    Mu, Jiajia
    Xie, Jialan
    Hu, Binghui
    Liu, Guangyuan
    FRONTIERS IN HUMAN NEUROSCIENCE, 2024, 18
  • [23] Multi-Head Self-Attention Gated-Dilated Convolutional Neural Network for Word Sense Disambiguation
    Zhang, Chun-Xiang
    Zhang, Yu-Long
    Gao, Xue-Yao
    IEEE ACCESS, 2023, 11 : 14202 - 14210
  • [24] Identifying sex differences in EEG-based emotion recognition using graph convolutional network with attention mechanism
    Peng, Dan
    Zheng, Wei-Long
    Liu, Luyu
    Jiang, Wei-Bang
    Li, Ziyi
    Lu, Yong
    Lu, Bao-Liang
    JOURNAL OF NEURAL ENGINEERING, 2023, 20 (06)
  • [25] AttenEpilepsy: A 2D convolutional network model based on multi-head self-attention
    Ma, Shuang
    Wang, Haifeng
    Yu, Zhihao
    Du, Luyao
    Zhang, Ming
    Fu, Qingxi
    ENGINEERING ANALYSIS WITH BOUNDARY ELEMENTS, 2024, 169
  • [26] Hierarchical Convolutional Neural Networks for EEG-Based Emotion Recognition
    Jinpeng Li
    Zhaoxiang Zhang
    Huiguang He
    Cognitive Computation, 2018, 10 : 368 - 380
  • [27] Fast Neural Chinese Named Entity Recognition with Multi-head Self-attention
    Qi, Tao
    Wu, Chuhan
    Wu, Fangzhao
    Ge, Suyu
    Liu, Junxin
    Huang, Yongfeng
    Xie, Xing
    KNOWLEDGE GRAPH AND SEMANTIC COMPUTING: KNOWLEDGE COMPUTING AND LANGUAGE UNDERSTANDING, 2019, 1134 : 98 - 110
  • [28] EEG-based emotion recognition using simple recurrent units network and ensemble learning
    Wei, Chen
    Chen, Lan-lan
    Song, Zhen-zhen
    Lou, Xiao-guang
    Li, Dong-dong
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2020, 58
  • [29] EmotioNet: A 3-D Convolutional Neural Network for EEG-based Emotion Recognition
    Wang, Yi
    Huang, Zhiyi
    McCane, Brendan
    Neo, Phoebe
    2018 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2018,
  • [30] EEG-based mild depression recognition using convolutional neural network
    Xiaowei Li
    Rong La
    Ying Wang
    Junhong Niu
    Shuai Zeng
    Shuting Sun
    Jing Zhu
    Medical & Biological Engineering & Computing, 2019, 57 : 1341 - 1352