EEG-Based Emotion Recognition Using Trainable Adjacency Relation Driven Graph Convolutional Network
Wei Li, Mingming Wang, Junyi Zhu, Aiguo Song
Abstract
In recent years, there has been a growing research interest in using deep learning to resolve the issue of electroencephalogram (EEG)-based emotion recognition. Current research emphasizes exploiting the useful information from each single EEG channel or each individual set of multichannel EEG, but overlooks the correlation information among different multichannel EEG sets. To explore such discriminative correlation information, we propose a novel and effective method, “trainable adjacency relation driven graph convolutional network (TARDGCN),” which contains two complementary modules: 1) trainable adjacency relation (TAR) and 2) graph convolutional network (GCN). TAR optimizes the local pairwise positions of multichannel EEG sets, which helps form an improved graphic representation for GCN to learn the global correlation among these sets for classification. The proposed method is capable of dealing with the problem of small sample size but large data variation in this issue. Our experimental results conducted on the databases DREAMER and DEAP in the subject-dependent and subject-independent modes show that TARDGCN outperforms the state-of-the-art approaches in classifying all of valence, arousal, and dominance.