1. 3DCANN: A Spatio-Temporal Convolution Attention Neural Network for EEG Emotion Recognition
- Author
-
Ling Zhao, Bing Li, Weiming Hu, Shuaiqi Liu, Xu Wang, Jie Yu, and Yudong Zhang
- Subjects
Computer science ,Emotion classification ,Emotions ,Feature extraction ,ComputerApplications_COMPUTERSINOTHERSYSTEMS ,Health Informatics ,Electroencephalography ,Convolution ,Health Information Management ,Artificial Intelligence ,Classifier (linguistics) ,medicine ,Humans ,Attention ,Electrical and Electronic Engineering ,Artificial neural network ,medicine.diagnostic_test ,business.industry ,Deep learning ,Pattern recognition ,Computer Science Applications ,ComputingMethodologies_PATTERNRECOGNITION ,Softmax function ,Neural Networks, Computer ,Artificial intelligence ,business - Abstract
Since electroencephalogram (EEG) signals can truly reflect human emotional state, emotion recognition based on EEG has turned into a critical branch in the field of artificial intelligence. Aiming at the disparity of EEG signals in various emotional states, we propose a new deep learning model named three-dimension convolution attention neural network (3DCANN) for EEG emotion recognition in this paper. The 3DCANN model is composed of spatio-temporal feature extraction module and EEG channel attention weight learning module, which can extract the dynamic relation well among multi-channel EEG signals and the internal spatial relation of multi-channel EEG signals during continuous period time. In this model, the spatio-temporal features are fused with the weights of dual attention learning, and the fused features are input into the softmax classifier for emotion classification. In addition, we utilize SJTU Emotion EEG Dataset (SEED) to appraise the feasibility and effectiveness of the proposed algorithm. Finally, experimental results display that the 3DCANN method has superior performance over the state-of-the-art models in EEG emotion recognition.
- Published
- 2022
- Full Text
- View/download PDF