A Pruned Deep Learning Approach for Classification of Motor Imagery Electroencephalography Signals

The Deep Learning (DL) approach has been gaining much popularity in recent years in the development of electroencephalogram (EEG) based Motor Imagery (MI) Brain-Computer Interface (BCI) systems, aiming to improve the performance of existing stroke rehabilitation strategies. A complex deep neural network structure has lots of neurons with thousands of parameters to optimize, and a great deal of data is often required to train the network and the training process can take an extremely long time. High training costs and high model complexity not only have negative impacts on the performance of the BCI system but also on its applicability to meet the real-time requirement to support the rehabilitation exercises of patients. To tackle the challenge, a contribution-based neuron selection method is proposed in this paper. A Convolutional Neural Network (CNN) based motor imagery classification framework is implemented, and a neuron pruning approach is developed and applied. The temporal and spatial features of EEG signals are captured by the CNN layers, and then the fast recursive algorithm (FRA) is applied to prune redundant parameters in the fully connected layers which reduces the computation cost of the CNN model without affecting its performance. The experimental results show that the proposed method can achieve up to 50% model size reduction and 67.09% computation savings.