Instance Transfer Subject-Dependent Strategy for Motor Imagery Signal Classification Using Deep Convolutional Neural Networks
Kai Zhang, Guanghua Xu, Longting Chen, Peiyuan Tian, Chengcheng Han, Sicong Zhang, Nan Duan
Abstract
In the process of brain-computer interface (BCI), variations across sessions/subjects result in differences in the properties of potential of the brain. This issue may lead to variations in feature distribution of electroencephalogram (EEG) across subjects, which greatly reduces the generalization ability of a classifier. Although subject-dependent (SD) strategy provides a promising way to solve the problem of personalized classification, it cannot achieve expected performance due to the limitation of the amount of data especially for a deep neural network (DNN) classification model. Herein, we propose an instance transfer subject-independent (ITSD) framework combined with a convolutional neural network (CNN) to improve the classification accuracy of the model during motor imagery (MI) task. The proposed framework consists of the following steps. Firstly, an instance transfer learning based on the perceptive Hash algorithm is proposed to measure similarity of spectrogram EEG signals between different subjects. Then, we develop a CNN to decode these signals after instance transfer learning. Next, the performance of classifications by different training strategies (subject-independent- (SI-) CNN, SD-CNN, and ITSD-CNN) are compared. To verify the effectiveness of the algorithm, we evaluate it on the dataset of BCI competition IV-2b. Experiments show that the instance transfer learning can achieve positive instance transfer using a CNN classification model. Among the three different training strategies, the average classification accuracy of ITSD-CNN can achieve <mml:math xmlns:mml="http://www.w3.org/1998/Math/MathML" id="M1"><mml:mn>94.7</mml:mn><mml:mo>±</mml:mo><mml:mn>2.6</mml:mn></mml:math> and obtain obvious improvement compared with a contrast model <mml:math xmlns:mml="http://www.w3.org/1998/Math/MathML" id="M2"><mml:mfenced open="(" close=")"><mml:mrow><mml:mi>p</mml:mi><mml:mo><</mml:mo><mml:mn>0.01</mml:mn></mml:mrow></mml:mfenced></mml:math>. Compared with other methods proposed in previous research, the framework of ITSD-CNN outperforms the state-of-the-art classification methods with a mean kappa value of 0.664.