Using the Center Loss Function to Improve Deep Learning Performance for EEG Signal Classification

Wenxiang Zhang,Qingshan Liu
DOI: https://doi.org/10.1109/icaci.2018.8377524
2018-01-01
Abstract:Electroencephalography (EEG) signal classification is an increasingly interesting task in Brain-computer interface (BCI) systems, but how to learn the pattern from EEG signals and design a general model for EEG classification is still a challenge. In this paper, we combine convolutional neural network and Long Short-Term Memory (CNN+LSTM) with center loss function to classify EEG signals. The 1 × 3 kernels as convolutional filters in 2D CNNs will be used, which is different from other models. Deep metric learning is usually used for image classification such as face recognition. In this paper, we use center loss function to learn more discriminative features to improve classification accuracy. Meanwhile, we experiment our methods on multiple EEG datasets. We have achieved a classification accuracy of about 94% on cognitive load recognition and 86.52% on emotion recognition. On EEG dataset for visual object analysis, our method produces a comparable result compared with who publicly release this dataset. The good performance indicates that our model with the center loss function is widely suitable on EEG signal classification.
What problem does this paper attempt to address?