Optimal Classification of Emotions from Electroencephalography (EEG) Signals

Abstract

It is incredibly challenging to build an intelligent algorithm for emotion recognition that can deliver high accuracy because electroencephalography (EEG) signals are not stationary, nonlinear, and noisy. First, decomposing the preprocessed EEG signals of the SEED dataset into five frequency bands: delta, theta, alpha, beta, and gamma, and then calculated their energy and entropy from the extracted features. Then Principal Component Analysis (PCA) method for feature reduction was performed. It is important to note that different types of wavelets transform (db6, db5, etc.) were tested, and hyperparameter tuning of classification models was done to obtain optimal accuracy. The next step is classifying the emotions into three states: -1(negative), 0(neutral), and +1(positive), and tested the dataset on two types of classification models, namely Random Forest and Support vector machine (SVM). SVM gives better performance compared to Random Forest with an accuracy of 80.74%.