Zheng Minmin, Yang Banghua
School of Mechatronic Engineering and Automation, Research Center of Brain Computer Engineering, Shanghai University, Shanghai, China; School of Mechanical and Electrical Engineering, Putian University, Fujian, China.
School of Mechatronic Engineering and Automation, Research Center of Brain Computer Engineering, Shanghai University, Shanghai, China.
Med Eng Phys. 2021 Oct;96:29-40. doi: 10.1016/j.medengphy.2021.08.006. Epub 2021 Aug 20.
The 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).
In 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.
We 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.
The 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.
Experimental 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.
脑电图(EEG)的非平稳性问题非常严重,尤其是对于自发信号而言,这导致与自发信号相关的机器学习效果不佳,特别是在跨时间的相关任务中,相应地限制了脑机接口(BCI)的实际应用。
在本文中,我们提出了一种新的迁移学习算法,该算法可以利用先前时刻的带标签运动想象(MI)脑电数据,以实现对当前时刻少量带标签脑电信号更好的分类准确率。
我们在深度卷积神经网络的全连接层中引入了一个自适应层。自适应层的目标函数设计为在最小化类内距离(DWC)和最大化各域内类间距离(DBCWD)的同时,最小化局部最大均值差异(LMMD)和预测误差。我们在两个公开数据集上验证了所提算法的有效性。
所提算法的分类准确率高于其他比较算法,配对t检验结果也表明所提算法的性能与其他算法有显著差异。混淆矩阵和特征可视化结果表明了所提算法的有效性。
实验结果表明,当当前只有少量带标签的MI脑电数据时,所提算法能够比其他算法实现更高的分类准确率。它在BCI领域具有应用前景。