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

被引:24
|
作者
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 条
  • [31] A novel two-stream multi-head self-attention convolutional neural network for bearing fault diagnosis
    Ren, Hang
    Liu, Shaogang
    Wei, Fengmei
    Qiu, Bo
    Zhao, Dan
    PROCEEDINGS OF THE INSTITUTION OF MECHANICAL ENGINEERS PART C-JOURNAL OF MECHANICAL ENGINEERING SCIENCE, 2024, 238 (11) : 5393 - 5405
  • [32] EEG emotion recognition using attention-based convolutional transformer neural network
    Gong, Linlin
    Li, Mingyang
    Zhang, Tao
    Chen, Wanzhong
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2023, 84
  • [33] EEG-based emotion recognition via improved evolutionary convolutional neural network
    Guo, Lexiang
    Li, Nan
    Zhang, Tian
    INTERNATIONAL JOURNAL OF BIO-INSPIRED COMPUTATION, 2024, 23 (04) : 203 - 213
  • [34] Neural Linguistic Steganalysis via Multi-Head Self-Attention
    Jiao, Sai-Mei
    Wang, Hai-feng
    Zhang, Kun
    Hu, Ya-qi
    JOURNAL OF ELECTRICAL AND COMPUTER ENGINEERING, 2021, 2021 (2021)
  • [35] Lip Recognition Based on Bi-GRU with Multi-Head Self-Attention
    Ni, Ran
    Jiang, Haiyang
    Zhou, Lu
    Lu, Yuanyao
    ARTIFICIAL INTELLIGENCE APPLICATIONS AND INNOVATIONS, PT III, AIAI 2024, 2024, 713 : 99 - 110
  • [36] Convolutional multi-head self-attention on memory for aspect sentiment classification
    Zhang, Yaojie
    Xu, Bing
    Zhao, Tiejun
    IEEE-CAA JOURNAL OF AUTOMATICA SINICA, 2020, 7 (04) : 1038 - 1044
  • [37] Multimodal EEG Emotion Recognition Based on the Attention Recurrent Graph Convolutional Network
    Chen, Jingxia
    Liu, Yang
    Xue, Wen
    Hu, Kailei
    Lin, Wentao
    INFORMATION, 2022, 13 (11)
  • [38] Convolutional Multi-Head Self-Attention on Memory for Aspect Sentiment Classification
    Yaojie Zhang
    Bing Xu
    Tiejun Zhao
    IEEE/CAAJournalofAutomaticaSinica, 2020, 7 (04) : 1038 - 1044
  • [39] 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)
  • [40] Multi-head enhanced self-attention network for novelty detection
    Zhang, Yingying
    Gong, Yuxin
    Zhu, Haogang
    Bai, Xiao
    Tang, Wenzhong
    PATTERN RECOGNITION, 2020, 107