EEG-Based Emotion Recognition Using Convolutional Recurrent Neural Network with Multi-Head Self-Attention
Source abstract
Abstract
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%.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: EEG and Brain-Computer Interfaces · Emotion and Mood Recognition · Advanced Computing and Algorithms
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
Key Laboratory of Optoelectronic Information Sensing and Technology, Chongqing University of Posts and Telecommunications, Chongqing 400065, China
School of Advanced Manufacturing Engineering, Chongqing University of Posts and Telecommunications, Chongqing 400065, China
Funding
National Natural Science Foundation of China
References (34)
Show source-provided references
1. Chao, H., Dong, L., Liu, Y., and Lu, B. (2019). Emotion recognition from multiband EEG signals using CapsNet. Sensors, 19.10.3390/s19092212 ↗
2. The genetic basis of the normal human electroencephalogram (EEG)10.1007/BF00295509 ↗
3. EEG-based classification of fast and slow hand movements using wavelet-CSP algorithm10.1109/TBME.2013.2248153 ↗
4. Automatic epileptic EEG detection using DT-CWT-based non-linear features10.1016/j.bspc.2017.01.010 ↗
5. Discriminative common spatial pattern sub-bands weighting based on distinction sensitive learning vector quantization method in motor imagery based brain-computer interface10.4103/2228-7477.161482 ↗
6. Pandey, P., and Seeja, K. (2019, January 21–22). Subject-independent emotion detection from EEG signals using deep neural network. Proceedings of the International Conference on Innovative Computing and Communications, Ostrava, Czech Republic.10.1007/978-981-13-2354-6_5 ↗
7. A novel deep learning approach for classification of EEG motor imagery signals10.1088/1741-2560/14/1/016003 ↗
8. Fourati, R., Ammar, B., Aouiti, C., Sanchez-Medina, J., and Alimi, A.M. (2017, January 14–18). Optimized Echo State Network with Intrinsic Plasticity for Eeg-Based Emotion Recognition. Proceedings of the International Conference on Neural Information Processing, Guangzhou, China.10.1007/978-3-319-70096-0_73 ↗
9. EEG-based emotion recognition with deep convolutional neural networks10.1515/bmt-2019-0306 ↗
10. Fares, A., Zhong, S., and Jiang, J. (2018, January 3–6). Region level bi-directional deep learning framework for eeg-based image classification. Proceedings of the 2018 IEEE International Conference on Bioinformatics and Biomedicine (BIBM), Madrid, Spain.10.1109/BIBM.2018.8621152 ↗
11. Speech emotion classification using attention-based LSTM10.1109/TASLP.2019.2925934 ↗
12. Cui, F., Wang, R., Ding, W., Chen, Y., and Huang, L. (2022). A Novel DE-CNN-BiLSTM Multi-Fusion Model for EEG Emotion Recognition. Mathematics, 10.10.3390/math10040582 ↗
13. Galvão, F., Alarcão, S.M., and Fonseca, M.J. (2021). Predicting Exact Valence and Arousal Values from EEG. Sensors, 21.10.3390/s21103414 ↗
14. Kim, Y., and Choi, A. (2020). EEG-based emotion classification using long short-term memory network with attention mechanism. Sensors, 20.10.3390/s20236727 ↗
15. Convolutional neural networks on graphs with fast localized spectral filtering
16. Tripathi, S., Acharya, S., Sharma, R.D., Mittal, S., and Bhattacharya, S. (2017, January 4–9). Using deep and convolutional neural networks for accurate emotion classification on deap dataset. Proceedings of the Twenty-ninth IAAI conference, San Francisco, CA, USA.
17. Hierarchical convolutional neural networks for EEG-based emotion recognition10.1007/s12559-017-9533-x ↗
18. Li, Y., Huang, J., Zhou, H., and Zhong, N. (2017). Human Emotion Recognition with Electroencephalographic Multidimensional Features by Hybrid Deep Neural Networks. Appl. Sci., 7.10.3390/app7101060 ↗
19. Automated emotion recognition based on higher order statistics and deep learning algorithm10.1016/j.bspc.2020.101867 ↗
20. Khan, N., Haq, I.U., Ullah, F.U.M., Khan, S.U., and Lee, M.Y. (2021). CL-Net: ConvLSTM-Based Hybrid Architecture for Batteries’ State of Health and Power Consumption Forecasting. Mathematics, 9.10.3390/math9243326 ↗
21. Attention is all you need
22. Cross-Subject Emotion Recognition Using Flexible Analytic Wavelet Transform From EEG Signals10.1109/JSEN.2018.2883497 ↗
23. Emotional states recognition, implementing a low computational complexity strategy
24. Synchrosqueezing transform based feature extraction from EEG signals for emotional state prediction10.1016/j.bspc.2019.04.023 ↗
25. Xu, H., Wang, X., Li, W., Wang, H., and Bi, Q. (2019, January 6–8). Research on EEG Channel Selection Method for Emotion Recognition. Proceedings of the 2019 IEEE International Conference on Robotics and Biomimetics (ROBIO), Dali, China.10.1109/ROBIO49542.2019.8961740 ↗
26. mOptimizing spatial filters for robust EEG single-trial analysis10.1109/MSP.2008.4408441 ↗
27. Transfer learning with deep convolutional neural network for automated detection of schizophrenia from EEG signals10.1007/s13246-020-00925-9 ↗
28. Sleep staging by bidirectional long short-term memory convolution neural network10.1016/j.future.2020.03.019 ↗
29. Bidirectional LSTM with attention mechanism and convolutional layer for text classification10.1016/j.neucom.2019.01.078 ↗
30. An attention-based bi-LSTM method for visual object classification via EEG10.1016/j.bspc.2020.102174 ↗
31. Deap: A database for emotion analysis; using physiological signals10.1109/T-AFFC.2011.15 ↗
32. Gaertner, M., Sauter, D., Baumgartl, H., Rieg, T., and Buettner, R. (2021, January 9–13). Multi-Class Emotion Recognition within the Valence-Arousal-Dominance Space Using EEG. Proceedings of the AMCIS, Online.
33. Kwon, Y.-H., Shin, S.-B., and Kim, S.-D. (2018). Electroencephalography based fusion two-dimensional (2D)-convolution neural networks (CNN) model for emotion recognition system. Sensors, 18.10.3390/s18051383 ↗
34. Classification of human emotions from electroencephalogram (EEG) signal using deep neural network