Simple Convolutional Neural Network for Left-Right Hands Motor Imagery EEG Signals Classification

Abstract

This article proposes a classification method of two-class motor imagery electroencephalogram (EEG) signals based on convolutional neural network (CNN), in which EEG signals from C3, C4 and Cz electrodes of publicly available BCI competition IV dataset 2b were used to test the performance of the CNN. The authors investigate two similar CNNs: a single-input CNN with a form of 2-dimensional input from short time Fourier transform (STFT) combining time, frequency and location information, and a multiple-input CNN with 3-dimensional input which processes the electrodes as an independent dimension. Fisher discriminant analysis-type F-score based on band pass (BP) feature and power spectra density (PSD) feature are employed respectively to select the subject-optimal frequency bands. In the experiments, typical frequency bands related to motor imagery EEG signals, subject-optimal frequency bands and extension frequency bands are employed respectively as the frequency range of the input image of CNN. The better classification performance of extension frequency bands show that CNN can extract optimal feature from frequency information automatically. The classification result also demonstrates that the proposed approach is more competitive in prediction of left/right hand motor imagery task compared with other state-of-art approaches.