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 条
  • [41] EEG-based Classification of Drivers Attention using Convolutional Neural Network
    Atilla, Fred
    Alimardani, Maryam
    PROCEEDINGS OF THE 2021 IEEE INTERNATIONAL CONFERENCE ON HUMAN-MACHINE SYSTEMS (ICHMS), 2021, : 59 - 62
  • [42] Personalized multi-head self-attention network for news recommendation
    Zheng, Cong
    Song, Yixuan
    NEURAL NETWORKS, 2025, 181
  • [43] Hierarchical Convolutional Neural Networks for EEG-Based Emotion Recognition
    Jinpeng Li
    Zhaoxiang Zhang
    Huiguang He
    Cognitive Computation, 2018, 10 : 368 - 380
  • [44] EEG-based emotion recognition with deep convolutional neural networks
    Ozdemir, Mehmet Akif
    Degirmenci, Murside
    Izci, Elf
    Akan, Aydin
    BIOMEDICAL ENGINEERING-BIOMEDIZINISCHE TECHNIK, 2021, 66 (01): : 43 - 57
  • [45] Hierarchical Convolutional Neural Networks for EEG-Based Emotion Recognition
    Li, Jinpeng
    Zhang, Zhaoxiang
    He, Huiguang
    COGNITIVE COMPUTATION, 2018, 10 (02) : 368 - 380
  • [46] Text summarization based on multi-head self-attention mechanism and pointer network
    Qiu, Dong
    Yang, Bing
    COMPLEX & INTELLIGENT SYSTEMS, 2022, 8 (01) : 555 - 567
  • [47] Speech emotion recognition using recurrent neural networks with directional self-attention
    Li, Dongdong
    Liu, Jinlin
    Yang, Zhuo
    Sun, Linyu
    Wang, Zhe
    EXPERT SYSTEMS WITH APPLICATIONS, 2021, 173
  • [48] MSASGCN : Multi-Head Self-Attention Spatiotemporal Graph Convolutional Network for Traffic Flow Forecasting
    Cao, Yang
    Liu, Detian
    Yin, Qizheng
    Xue, Fei
    Tang, Hengliang
    JOURNAL OF ADVANCED TRANSPORTATION, 2022, 2022
  • [49] Text summarization based on multi-head self-attention mechanism and pointer network
    Dong Qiu
    Bing Yang
    Complex & Intelligent Systems, 2022, 8 : 555 - 567
  • [50] A malicious network traffic detection model based on bidirectional temporal convolutional network with multi-head self-attention mechanism
    Cai, Saihua
    Xu, Han
    Liu, Mingjie
    Chen, Zhilin
    Zhang, Guofeng
    COMPUTERS & SECURITY, 2024, 136