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

Researchers

Journal

Modalities

Models

Abstract

The 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.
Deep 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.
The 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.
Our 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.
Copyright © 2020 Elsevier B.V. All rights reserved.