A deep neural network with subdomain adaptation for motor imagery brain-computer interface

Abstract

BackgroundThe nonstationarity problem of EEG is very serious, especially for spontaneous signals, which leads to the poor effect of machine learning related to spontaneous signals, especially in related tasks across time, which correspondingly limits the practical use of brain-computer interface (BCI). ObjectiveIn this paper, we proposed a new transfer learning algorithm, which can utilize the labeled motor imagery (MI) EEG data at the previous time to achieve better classification accuracies for a small number of labeled EEG signals at the current time. MethodsWe introduced an adaptive layer into the full connection layer of a deep convolution neural network. The objective function of the adaptive layer was designed to minimize the Local Maximum Mean Discrepancy (LMMD) and the prediction error while minimizing the distance within each class (DWC) and maximizing the distance between classes within each domain (DBCWD). We verified the effectiveness of the proposed algorithm on two public datasets. ResultsThe classification accuracy of the proposed algorithm was higher than other comparison algorithms, and the paired t-test results also showed that the performance of the proposed algorithm was significantly different from that of other algorithms. The results of the confusion matrix and feature visualization showed the effectiveness of the proposed algorithm. ConclusionExperimental results showed that the proposed algorithm can achieve higher classification accuracy than other algorithms when there was only a small amount of labeled MI EEG data at the current time. It can be promising to be applied to the field of BCI.