Emotion recognition with convolutional neural network and EEG-based EFDMs

Neuropsychologia. 2020 Sep:146:107506. doi: 10.1016/j.neuropsychologia.2020.107506. Epub 2020 Jun 1.

Abstract

Electroencephalogram (EEG), as a direct response to brain activity, can be used to detect mental states and physical conditions. Among various EEG-based emotion recognition studies, due to the non-linear, non-stationary and the individual difference of EEG signals, traditional recognition methods still have the disadvantages of complicated feature extraction and low recognition rates. Thus, this paper first proposes a novel concept of electrode-frequency distribution maps (EFDMs) with short-time Fourier transform (STFT). Residual block based deep convolutional neural network (CNN) is proposed for automatic feature extraction and emotion classification with EFDMs. Aim at the shortcomings of the small amount of EEG samples and the challenge of differences in individual emotions, which makes it difficult to construct a universal model, this paper proposes a cross-datasets emotion recognition method of deep model transfer learning. Experiments carried out on two publicly available datasets. The proposed method achieved an average classification score of 90.59% based on a short length of EEG data on SEED, which is 4.51% higher than the baseline method. Then, the pre-trained model was applied to DEAP through deep model transfer learning with a few samples, resulted an average accuracy of 82.84%. Finally, this paper adopts the gradient weighted class activation mapping (Grad-CAM) to get a glimpse of what features the CNN has learned during training from EFDMs and concludes that the high frequency bands are more favorable for emotion recognition.

Keywords: Convolutional neural network; Electrode-frequency distribution maps; Electroencephalogram; Emotion recognition; Gradient-weighted class activation mapping.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Electrodes
  • Electroencephalography*
  • Emotions
  • Humans
  • Machine Learning
  • Neural Networks, Computer*