Affiliation:
1. Faculty of Information Engineering and Automation, Kunming University of Science and Technology, Kunming 650500, China
Abstract
In recent years, EEG-based emotion recognition technology has made progress, but there are still problems of low model efficiency and loss of emotional information, and there is still room for improvement in recognition accuracy. To fully utilize EEG’s emotional information and improve recognition accuracy while reducing computational costs, this paper proposes a Convolutional-Recurrent Hybrid Network with a dual-stream adaptive approach and an attention mechanism (CSA-SA-CRTNN). Firstly, the model utilizes a CSAM module to assign corresponding weights to EEG channels. Then, an adaptive dual-stream convolutional-recurrent network (SA-CRNN and MHSA-CRNN) is applied to extract local spatial-temporal features. After that, the extracted local features are concatenated and fed into a temporal convolutional network with a multi-head self-attention mechanism (MHSA-TCN) to capture global information. Finally, the extracted EEG information is used for emotion classification. We conducted binary and ternary classification experiments on the DEAP dataset, achieving 99.26% and 99.15% accuracy for arousal and valence in binary classification and 97.69% and 98.05% in ternary classification, and on the SEED dataset, we achieved an accuracy of 98.63%, surpassing relevant algorithms. Additionally, the model’s efficiency is significantly higher than other models, achieving better accuracy with lower resource consumption.
Funder
National Natural Science Foundation of China
Key Research Project of Yunnan Province
Reference64 articles.
1. Semi-supervised regression with adaptive graph learning for EEG-based emotion recognition;Sha;Math. Biosci. Eng.,2023
2. A Channel-Fused Dense Convolutional Network for EEG-Based Emotion Recognition;Gao;IEEE Trans. Cogn. Dev. Syst.,2021
3. Machine-Learning-Based Approaches for Post-Traumatic Stress Disorder Diagnosis Using Video and EEG Sensors: A Review;Othmani;IEEE Sens. J.,2023
4. 3DCANN: A Spatio-Temporal Convolution Attention Neural Network for EEG Emotion Recognition;Liu;IEEE J. Biomed. Health Inform.,2022
5. Wang, S., Shuai, H., Liu, Q., and Wang, F. (2024). Cooperative Sentiment Agents for Multimodal Sentiment Analysis. arXiv.