The role of data balancing for emotion classification using EEG signals

In this paper, we demonstrate the role of data balancing in experimental evaluation of emotion classification systems based on electroencephalogram (EEG) signals. ADASYN method was employed to create a balanced version of the DEAP EEG dataset. Experiments considered Support Vector Machine classifier...

Full description

Saved in:
Bibliographic Details
Published inInternational Conference on Digital Signal Processing proceedings pp. 555 - 559
Main Authors Torres Pereira, Eanes, Martins Gomes, Herman
Format Conference Proceeding
LanguageEnglish
Published IEEE 01.10.2016
Subjects
Affective Computing
EEG Signal
Electroencephalography
Imbalanced Datasets
Pattern Recognition
Online AccessGet full text

Cover

More Information
Summary:In this paper, we demonstrate the role of data balancing in experimental evaluation of emotion classification systems based on electroencephalogram (EEG) signals. ADASYN method was employed to create a balanced version of the DEAP EEG dataset. Experiments considered Support Vector Machine classifiers trained with HOC and PSD features to predict valence and arousal affective dimensions. Using signals from only four channels (Fp1, Fp2, F3 and F4) we obtained, after balancing, accuracies of 98% (valence) and 99% (arousal) for subject dependent experiments with three classes, and 85% (valence) and 87% (arousal) for two-class classification. However, accuracies for subject independent experiments were lower than the ones obtained using imbalanced datasets. We obtained accuracies of 52% (valence) and of 49% (arousal) for two classes, and accuracies of 36% (valence) and of 31% (arousal) for three classes. To explain the low accuracies in subject independent experiments, we present arguments and empirical evidence using correlations between the percentage of samples for each class and the accuracies obtained by approaches which did not use balanced datasets.
ISSN:2165-3577
DOI:10.1109/ICDSP.2016.7868619