Electroencephalogram Emotion Recognition Using Combined Features in Variational Mode Decomposition Domain

Abstract

Using Electroencephalogram (EEG) to recognize human emotion has attracted increasing attention. However, features extraction from EEG is a challenging work because it is a non-stationary continuous sequential signal. To obtain more pattern information, a combined features extraction method in Variational Mode Decomposition (VMD) domain is proposed, which can extract local features of EEG signals to overcome the effects caused by non-stationarity. This method first decomposes EEG into several components using VMD, and then combined features of Differential Entropy (DE) and Short-Time Energy (STE) are extracted from each component. To optimize combined features, the important features are selected by tree modes, and the feature set is dimensionally reduced by further using Linear Discriminant Analysis (LDA). Moreover, an XGBoost classifier with Bayesian optimization is presented to classify different emotional states. Binary-class and multi-class EEG emotion recognition are conducted on the DEAP dataset, from which the experimental results show that accuracy of binary-class classification is 81.77% for High/Low Valence and 80.47% for High/Low Arousal, and accuracy of 91.41%, 94.27%, 94.27% and 93.49% are obtained for HVHA, LVHA, LVLA and HVLA, respectively, which demonstrate its effectiveness.