Dong Aimei, Li Zhigang, Zheng Qiuyu
School of Computer Science and Technology, Qilu University of Technology (Shandong Academy of Science), Jinan, China.
Front Neurosci. 2021 Mar 24;15:647393. doi: 10.3389/fnins.2021.647393. eCollection 2021.
EEG signal classification has been a research hotspot recently. The combination of EEG signal classification with machine learning technology is very popular. Traditional machine leaning methods for EEG signal classification assume that the EEG signals are drawn from the same distribution. However, the assumption is not always satisfied with the practical applications. In practical applications, the training dataset and the testing dataset are from different but related domains. How to make best use of the training dataset knowledge to improve the testing dataset is critical for these circumstances. In this paper, a novel method combining the non-negative matrix factorization technology and the transfer learning (NMF-TL) is proposed for EEG signal classification. Specifically, the shared subspace is extracted from the testing dataset and training dataset using non-negative matrix factorization firstly and then the shared subspace and the original feature space are combined to obtain the final EEG signal classification results. On the one hand, the non-negative matrix factorization can assure to obtain essential information between the testing and the training dataset; on the other hand, the combination of shared subspace and the original feature space can fully use all the signals including the testing and the training dataset. Extensive experiments on Bonn EEG confirmed the effectiveness of the proposed method.
脑电图(EEG)信号分类近来一直是研究热点。EEG信号分类与机器学习技术相结合的方式非常流行。用于EEG信号分类的传统机器学习方法假定EEG信号来自相同分布。然而,实际应用中该假设并非总能成立。在实际应用中,训练数据集和测试数据集来自不同但相关的领域。在这些情况下,如何充分利用训练数据集的知识来提升测试数据集至关重要。本文提出了一种将非负矩阵分解技术与迁移学习相结合的新方法(NMF-TL)用于EEG信号分类。具体而言,首先使用非负矩阵分解从测试数据集和训练数据集中提取共享子空间,然后将共享子空间与原始特征空间相结合以获得最终的EEG信号分类结果。一方面,非负矩阵分解能够确保获取测试数据集与训练数据集之间的关键信息;另一方面,共享子空间与原始特征空间的结合能够充分利用包括测试数据集和训练数据集在内的所有信号。在Bonn EEG上进行的大量实验证实了该方法的有效性。