高级检索+

基于迁移学习的运动想象脑电信号分类研究

Classification of Motor Imagery EEG Signals Based on Transfer Learning

  • 摘要: 由于单个被试的运动想象脑电训练样本较少,且不同被试的脑电个体差异较大,导致大脑运动想象任务分类正确率不高、训练校准时间过长.本文提出一种欧几里得对齐-正则共空间模式的迁移学习方法,实现在样本和特征上同时进行迁移.通过数据对齐的迁移方法使不同被试间样本分布更相似,采用正则化迁移方法改进共空间模式,提取出更鲁棒的空间特征信息.对BCI competition运动想象脑电数据集进行仿真测试,右手和右脚运动想象任务的平均分类正确率可以达到87.10%;在小训练样本集上,本文的迁移学习方法相比共空间模式方法,其平均分类正确率提高了14.43%;对于跨被试的运动想象脑电,平均分类正确率提高了23.5%.实验结果表明,该方法可以有效提高小样本、跨被试运动想象脑电的分类正确率,减少模型校准训练时间,提高泛化能力.

     

    Abstract: Due to the small number of training samples of motor imagery EEG for a single subject and the great individual differences of EEG among different subjects, the accuracy of brain motor imagery task classification is low and the training calibration time is too long. In this paper, a transfer learning method of Euclidean aligned-regular common space pattern is proposed to realize the transfer of samples and features at the same time. Firstly, the data alignment transfer method is used to make the sample distribution more similar among different subjects, and then the regularized transfer method is applied to improve the common spatial pattern and extract more robust spatial feature information. Simulation test on BCI competition motor imagery EEG dataset shows the average classification accuracy of motor imagery tasks for right hand and right foot can reach 87.10%. On the small training sample set, the average classification accuracy of the transfer learning method is 14.43% higher than that of the common space pattern method. For the motor imagery EEG across subjects, the average classification accuracy is improved by 23.5%. The experimental results show that this method can effectively improve the classification accuracy of motor imagery EEG for small sample and cross subject, reduce the training time of model calibration, and improve the generalization ability.

     

/

返回文章
返回