Motor Imagery EEG Decoding Method Based on a Discriminative Feature Learning Strategy

• Published in: IEEE transactions on neural systems and rehabilitation engineering Vol. 29; pp. 368 - 379
• Main Authors: Yang, Lie, Song, Yonghao, Ma, Ke, Xie, Longhan
• Format: Journal Article
• Language: English
• Published: United States IEEE 2021; The Institute of Electrical and Electronics Engineers, Inc. (IEEE)
• Subjects: central distance loss (CD-loss); central vector shift; central vector update; circular translation strategy; Classification algorithms; Convergence; Data mining; Datasets; Decoding; Deep learning; EEG; Electroencephalography; Feature extraction; Imagery; Machine learning; Mental task performance; Methods; Motor imagery electroencephalograph (EEG) decoding; Motor skill learning; Strategy; Task analysis
• ISSN: 1534-4320; 1558-0210; 1558-0210
• DOI: 10.1109/TNSRE.2021.3051958

With the rapid development of deep learning, more and more deep learning-based motor imagery electroencephalograph (EEG) decoding methods have emerged in recent years. However, the existing deep learning-based methods usually only adopt the constraint of classification loss, which hardly obtains the features with high discrimination and limits the improvement of EEG decoding accuracy. In this paper, a discriminative feature learning strategy is proposed to improve the discrimination of features, which includes the central distance loss (CD-loss), the central vector shift strategy, and the central vector update process. First, the CD-loss is proposed to make the same class of samples converge to the corresponding central vector. Then, the central vector shift strategy extends the distance between different classes of samples in the feature space. Finally, the central vector update process is adopted to avoid the non-convergence of CD-loss and weaken the influence of the initial value of central vectors on the final results. In addition, overfitting is another severe challenge for deep learning-based EEG decoding methods. To deal with this problem, a data augmentation method based on circular translation strategy is proposed to expand the experimental datasets without introducing any extra noise or losing any information of the original data. To validate the effectiveness of the proposed method, we conduct some experiments on two public motor imagery EEG datasets (BCI competition IV 2a and 2b dataset), respectively. The comparison with current state-of-the-art methods indicates that our method achieves the highest average accuracy and good stability on the two experimental datasets.