CWT Based Transfer Learning for Motor Imagery Classification for Brain computer Interfaces

Abstract

BackgroundThe processing of brain signals for Motor imagery (MI) classification to have better accuracy is a key issue in the Brain-Computer Interface (BCI). While conventional methods like Artificial neural network (ANN), Linear discernment analysis (LDA), K-Nearest Neighbor (KNN), Support vector machine (SVM), etc. have made significant progress in terms of classification accuracy, deep transfer learning-based systems have shown the potential to outperform them. BCI can play a vital role in enabling communication with the external world for persons with motor disabilities. New MethodsDeep learning has been a success in many fields. However, for Electroencephalogram (EEG) signals, relatively minimal work has been carried out using deep learning. This paper proposes a combination of Continuous Wavelet Transform (CWT) along with deep learning-based transfer learning to solve the problem. CWT transforms one dimensional EEG signals into two-dimensional time-frequency-amplitude representation enabling us to exploit available deep networks through transfer learning. ResultsThe effectiveness of the proposed approach is evaluated in this study using an openly available BCI competition data-set. The results of the approach have been compared to earlier works on the same dataset, and a promising validation accuracy of 95.71% is achieved in our investigation. Comparison with existing methods and ConclusionOur approach has shown significant improvement over other studies, which is 5.71% improvement over earlier reported algorithm (Tabar and Halici, 2017) using the same dataset. Results show the validity of the proposed Deep Transfer-Learning based technique as a state of the art technique for MI classification in BCI.