A Comparative Study of Window Size and Channel Arrangement on EEG-Emotion Recognition Using Deep CNN

Sensors (Basel). 2021 Mar 1;21(5):1678. doi: 10.3390/s21051678.

Abstract

Emotion recognition based on electroencephalograms has become an active research area. Yet, identifying emotions using only brainwaves is still very challenging, especially the subject-independent task. Numerous studies have tried to propose methods to recognize emotions, including machine learning techniques like convolutional neural network (CNN). Since CNN has shown its potential in generalization to unseen subjects, manipulating CNN hyperparameters like the window size and electrode order might be beneficial. To our knowledge, this is the first work that extensively observed the parameter selection effect on the CNN. The temporal information in distinct window sizes was found to significantly affect the recognition performance, and CNN was found to be more responsive to changing window sizes than the support vector machine. Classifying the arousal achieved the best performance with a window size of ten seconds, obtaining 56.85% accuracy and a Matthews correlation coefficient (MCC) of 0.1369. Valence recognition had the best performance with a window length of eight seconds at 73.34% accuracy and an MCC value of 0.4669. Spatial information from varying the electrode orders had a small effect on the classification. Overall, valence results had a much more superior performance than arousal results, which were, perhaps, influenced by features related to brain activity asymmetry between the left and right hemispheres.

Keywords: CNN; EEG; brainwave; electrode order; emotion recognition; machine learning; neuroscience; spatiotemporal data; window size.

MeSH terms

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