Minimum spanning tree based graph neural network for emotion classification using EEG

Neural Netw. 2022 Jan:145:308-318. doi: 10.1016/j.neunet.2021.10.023. Epub 2021 Nov 3.

Abstract

Emotion classification based on neurophysiology signals has been a challenging issue in the literature. Recent neuroscience findings suggest that brain network structure underlying the different emotions provides a window in understanding human affection. In this paper, we propose a novel method to capture the distinct minimum spanning tree (MST) topology underpinning the different emotions. Specifically, we propose a hierarchical aggregation-based graph neural network to investigate the MST structure in emotion recognition. Extensive experiments on the public available DEAP dataset demonstrate the superior performance of the model in emotion classification as compared to existing methods. In addition, the results show that the theta, lower beta and gamma frequency band network information are more sensitive to emotions, suggesting a multi-frequency interaction in emotion processing.

Keywords: DEAP; Emotion classification; Graph neural network; MST.

MeSH terms

  • Brain / diagnostic imaging
  • Electroencephalography*
  • Emotions
  • Humans
  • Neural Networks, Computer*