1、头皮脑电信号(ElectroEncephaloGram,EEG)因其具有较高的时间分辨率、较好的便携性以及非侵入式的优势,现已成为解码大脑认知活动最有效的数据来源之一。基于EEG 的运动想象脑机接口(Motor Imagery Brain-Computer Interface,MI-BCI)属于自发型脑机接口类别,其目的在于准确地辨认出用户的肢体运动意图,常见的包括左手、右手、双脚以及舌头运动的想象1-2,这对于医疗康复3、休闲娱乐等领域具有重要意义。运动想象脑电信号识别任务的传统研究方法首先需要对EEG 信号进行预处理,随后选择合适的方法提取不同认知状态下的脑电特征并挑选出最具识别性的特征子
2、集,最后通过机器学习方法完成对运动想象脑电信号的识别工作。比如Kevric 等利用小波包分解子带中提取的高阶统计特征结合多尺度主成成分分析(Multiscale Principal Component Analysis,MSPCA)的降噪方法,在BCI Competition dataseta 数据集上得到了92.8%的平均识别准确率。Baig 等首先利用公共空间模式(Common Spatial Pattern,CSP)算法提取出左右手运动想象状态下的特征集合,然后使用差分进化的优化算法提取各个被试的最优特征子集,最后送入支持向量机(Support Vector Machine,SVM)中构
3、建分类模型,在BCI Competition dataset a数据集上得到了95%以上的平均分类准确率。传统的机器学习和统计方法面对复杂且往往是非结构化的数据时常常要求一定的特征工程能力,以提取更为有效的特征或者选择更加合适的模型参数,而深度学习可以训练出更抽象也更有效的特征来完成端到端9的学习任务。近年来深度学习方法在机器视觉10-11和语音识别12-13等领域取得了巨大成功,证明深度学习方法可以很好地找出高维数据中隐藏的复杂结构。Tabar 等14利用短时傅里叶变换(Short-Time Fourier Transform,STFT)提取出各通道中 和节律的时频信息,并将它们组合成2D
4、信息作为网络的输入,该网络是由卷积神经网络(Convolutional Neural Network,CNN)和堆栈自编码器(Stacked AutoEncoder,SAE)相结合形成的深层网络,并通过SAE 对CNN 提取到的特征进行分类,所提出的方法在BCI Competition dataset 2b 数据集上的Kappa 值为0.547,相较于竞赛第一名有9%的改进。Tang等15提出了一种新型卷积神经网络用于运动想象EEG 的模式识别,该网络设置了2 层卷积层分别负责提取信号序列中的空间域特征和时间域特征。该文中使用的数据源自浙江大学自行设计的运动想象实验,每个被试执行左、右手运动想
5、象任务各230 次,实验结果表明使用本文提出的新型CNN 方法可以获得比传统方法更高的分类精度。综上所述:1)卷积神经网络避免了复杂的特征提取过程,且因其具有权值共享和稀疏连接等特性16,有效地降低了模型的复杂程度,因此,本文考虑将卷积神经网络引入到运动想象脑电信号的识别任务当中。2)以经验风险最小化原则17(Empirical Risk Minimization,ERM)作为学习规则的网络模型,鼓励网络记忆而不是泛化训练数据,特别是当网络在处理小样本问题时,仅从模型结构上做调整来避免过拟合往往是不够的,还需要在数据层面作进一步处理。3)深层的网络模型会构成一个庞大的超参数集合,一直以来由于缺
6、乏求解超参集最优组合的理论支撑以及足够穷举所有组合的计算资源,因此需要根据训练样本的特性从不同的角度对超参数的组合进行尝试,以求达到最优的泛化效果。1 数据准备和分类方法1.1 数据集本文选用BCI Competition dataset 运动想象数据集,该数据集记录了一位正常女性被试共280 次随机的左右手运动想象实验,以随机的方式挑选出140 次实验作为训练集,剩余的140 次实验作为测试集,其中训练集包括70 次左手想象和70 次右手想象。整个采集过程以128 Hz 采样频率记录脑电信号,并进行0.530 Hz带通滤波。数据采集过程中每单次实验的时间流程如图1所示。图1 单次运动想象任务
7、时序图Fig.1 Sequence diagram of single motor imagery task每次实验持续时间共9 s,实验开始之后会有2 s 准备时间,在第3 秒开始时会有短暂的声音提示被试即将执行运动想象任务,同时屏幕显示一个交叉十字“+”持续时间为1 s,从第4 秒开始屏幕会出现一个箭头,被试通过运动想象拖动反馈条向箭头指示方向移动,并保持至第9秒结束。1.2 数据预处理人在进行肢体运动想象时大脑的运动感觉皮层中的特定位置会发生规律性的电位变化18,当被试进行单侧肢体运动想象时,大脑对侧皮层中 节律(812 Hz)的强度减小,同侧皮层中 节律(1225 Hz)的强度增加,称
8、为事件相关去同步(Event Related Desynchronization,ERD)和事件相关同步(Event Related Synchronization,ERS)现象19。这两种现象是判别不同类型脑电信号的重要依据,其中时频域分析法是最为高效的分析方法之一20。1.2.1 短时距傅里叶变换短时距傅里叶变换先将整段时间序列切分成若干等长的时间片段,再通过傅里叶变换计算出各个时间片段当中的频谱信息,从而获得各频率成分关于时间的变化。其计算公式如下:其中:S(n)代表脑电信号的时间序列;W(n)表窗口函数;N代表记录的时间点总数;k表示不同时间窗的索引;f代表信号中的频率成分;n代表时间
9、点。公式中要求划分的时间窗口长度相同,决定了该算法在测量高频成分时表现良好,而在测量低频成分时往往会伴有失真现象。为有效测量信号中节律和 节律的变化趋势,本文选用长度为0.5 s 的时间窗口配合Hamming 窗口函数得到的时频矩阵如图2 所示。最后将C3、C4 两个通道上的时频矩阵进行组合,得到33352 大小的三维张量作为后续卷积神经网络的输入。1.2.2 连续Morlet小波变换Morlet 小波变换属于连续型小波变换(Continuous Wavelet Transform,CWT),它以有限长度且会衰减的小波作为基底来测量信号中各个频率成分强度随时间的变化,其公式如下:x(t)代表信
10、号序列;(t)代表小波基底;t代表时间点;参数a控制小波函数的伸缩,当a从小到大取值时,小波函数逐渐变宽,于是可以更好地对低频成分作出测量,并且通过对参数b的调整控制小波函数的平移进而得到不同时域位置上各个频带的强度信息。Morlet 小波中心时间和时域跨度的计算公式如下:()是(t)经过傅里叶变换以后得到的频率成分信息。由式(3)(7)可知,小波变换在测量高频成分时,由于使用的小波较窄,可以获得较小的时域跨度,但相应地会使频域跨度变大,因此在输出的时频矩阵中,高频部分频率维度的解析度相对偏低,低频部分则正好相反。本文使用Morlet 小波变换得到的结果如图3 所示。同样地,将C3、C4 通道
11、位置信息进行整合,得到大小为351 1522 的样本矩阵作为神经网络的输入。图2 STFT运动想象时频图Fig.2 STFT motor imagery time-frequency map图3 CWT运动想象时频图Fig.3 CWT motor imagery time-frequency map2 卷积神经网络的构建2.1 网络拓扑结构为了能够训练出同时间、频率和空间三个维度均相关的高阶特征,构建出的CNN 模型如图4 所示。为方便后续讨论称该网络为MixedCNN。具体的拓扑描述如下:Input输入层接收的样本尺寸为FTC。C1 卷积层使用的卷积核大小为332,经该层卷积运算后共向外输出
12、64个特征矩阵。C2 卷积层使用的卷积核大小为364,并向后输出64个特征矩阵。Pool 池化层用于对C2 层的输出进行降采样操作以缩减模型的计算规模,池化窗口大小为22。FC1 和FC2 全连接层分别包含600 和300 个神经元,为避免出现过拟合现象,引入50%的Dropout 策略,使用的激活函数为ReLu。Output层仅共包含2个神经元,代表网络的2类输出结果(左手运动想象、右手运动想象)。为了避免在同一个卷积核内混入不同维度的信息,本文另外设计了一套可以依次对空、时、频三个维度的特征进行提取的网络模型如图5 所示。为方便后续讨论将该网络记作StepByStepCNN。具体的网络拓扑
13、描述如下:C1卷积层使用大小为112的卷积核对C3、C4通道的空间信息进行整合,并向后层输出4个特征矩阵。C2卷积层针对C1层输出的每一个特征矩阵设计了16个大小为F1的卷积核以提取其频域特征,共得到64个1T大小的特征矩阵。C3 卷积层使用16 个大小为16 的卷积核对输入的每张特征矩阵进行时域上的特征提取。FC1 和FC2 分别包含500 个和200 个神经元,激活函数选用ReLU,且引入了50%的Dropout 机制并最终配合输出层完成对样本的识别工作。图4 MixedCNN拓扑结构Fig.4 MixedCNN topology structure图5 StepByStepCNN拓扑结构
14、Fig.5 StepByStepCNN topology structure2.2 CNN的学习过程在MixingCNN 网络中将FC的输入张量记作a(0),网络中各层的计算描述如下:Convolution1:对输入张量执行卷积操作。上标代表层数;下标s和i分别代表该卷积层内卷积核的个数和卷积操作执行的次数分别表示第s个卷积核的权重和偏置均为32 大小的矩阵;f(x)选择ReLU 激活函数,输出的每个特征矩阵的维度为FT。Convolution2:对C1层的输出数据进行卷积计算:与C1 卷积层的学习过程类似分别代表364大小卷积核的权重和偏置。Pool1:对C2的输出执行最大池化操作,最大子采
15、样函数取区域内所有输出的最大值(max-pooling),输出矩阵的大小npool可以由式(12)计算:为特征矩阵的原始尺寸;p为边缘填充数量;f为池化窗口的大小;s代表步长;池化层的输出记为a(3)。FC1:该层中每一个神经元均与前一层所有神经元相连接,负责对训练得到的特征进行分类。p(4)为Dropout 比例,且该操作仅在训练过程中执行21;分别代表输入权重和偏置。FC2:由于深层结构的网络比相同神经元数量的浅层网络具有更好的表达非线性关系的能力,为此增加了FC2 全连接层,其计算过程和FC1完全相同:同样地,在StepByStepCNN网络中将输入张量记作a(0),网络中各层的计算描述
16、如下:对输入张量做空间域卷积,为确保上述两种网络可以得到有效的训练,采用在期望为0,方差为2/nl正态分布中随机采样的方式对各层神经元的权重进行初始化,其中nl为第l层神经元的个数,以避免损失函数全程不下降问题。网络采用带有动量的梯度下降算法对损失函数进行优化,学习率设置为0.001,最大迭代次数设置为300。2.3 分类本文根据EEG 信号的时-频-空特性,设计了两种不同卷积策略的网络模型;然后将STFT 和CWT 预处理得到的样本分别放入MixedCNN 和StepByStepCNN 进行训练,最终得到四组模型;接着,使用准确率、精确率、灵敏度、特异性以及ROC曲线下面积(Area Und
17、er Curve,AUC)对其进行评估,并同时与CSP+SVM22、自适应自回归模型(Adaptive Autoregressive Modeling,AAR)+LDA23、离散小波变换(Discrete Wavelet Transform,DWT)+长短时记忆网络(Long Short-Term Memory,LSTM24)和STFT+SAE14四种方法相比较;最后,挑选出识别性能最佳的模型并以此作为baseline 探讨mixup 数据增强对模型泛化性能的影响。3 mixup数据增强不同于翻转、平移、高斯噪声等通用数据增强方法,mixup25采用对不同类别之间进行建模的方式实现数据增强。m
18、ixup 数据增强受邻域风险最小化策略26启发构建出一种通用的邻域分布来近似真实分布,不同于Chapelle 等26提出的高斯邻域,mixup邻域的数学表示如下:Beta(,),(0,);mixup 使用从该邻域采样得到的虚拟样本-标签对构成新的训练数据,并以最小化式(28)对网络参数进行调整。f()表示模型函数;()表示损失函数;m表示虚拟样本的数量。Zhang 等25指出当在区间0.1,0.4内取值时可以获得比ERM学习规则更好的泛化性能。在实验中将超参数分别设置为0.1、0.2、0.3、0.4 并在每个batch 范围内将训练数据和标签分别做加权求和组成新的虚拟样本集合。在损失函数构建方
19、面,需要将输出的预测值分别与两组标签求出损失,并与之前真实样本分配的权重进行加权求和。4 实验与结果分析4.1 模型分析及对比本文提出的4 种分类模型:CWT+MixedCNN、CWT+StepByStepCNN、STFT+MixedCNN、STFT+StepByStepCNN 在测试集上计算出的评价指标如表1所示。表1 实验组模型的评价指标Tab.1 Evaluation indeces of models in experimental group对于当前被试的运动想象脑电数据,在相同预处理方式下,选择MixedCNN 比选择StepByStepCNN 训练出的分类模型具有更好的识别准确率
20、,且使用CWT 预处理方法配合MixedCNN 训练得到的模型准确率最高,达到了92.14%。另一方面,相较于CWT,STFT 预处理方法得到的样本在StepByStepCNN 上训练出的模型在准确率指标上更好。在使用相同卷积网络模型的前提下,采用STFT预处理方式得到的模型在精确率指标上表现更加突出。STFT+MixedCNN 和CWT+StepByStepCNN 两种方案在灵敏度指标上保持一致,CWT 配合MixedCNN 方法在灵敏度指标上相较于其他3 种方法至少提升了超过4 个百分点。在采用相同卷积策略的条件下,预处理方式选择STFT 相较于CWT 训练出的模型在特异性指标上表现更好,
21、其中采用STFT+StepByStepCNN 方案所得到的特异性指标最高,达到了95.71%。最后,采用MixedCNN结构的网络训练出的模型在AUC 指标上略优于StepByStepCNN训练出的模型。为与以上方法相比较,本文在BCI Competition dataset数据集合上复现了另外4 种分析方法(CSP+SVM、AAR+LDA、DWT+LSTM、STFT+SAE),并在测试集上得到其评价指标如表2 所示。其中STFT+SVM 方法在准确率指标上高于本文提出的CWT+StepByStepCNN、STFT+MixedCNN、STFT+StepByStepCNN 方法,以及另外3 种对
22、比方法。在特异性指标上STFT+SAE 方法和STFT+StepByStepCNN 方法保持一致,且在精确率指标上高于STFT+StepByStepCNN 方法。CWT+MixedCNN 方法在准确率、灵敏度和AUC 三种指标上分别高出STFT+SAE 方法0.71个百分点,4.29个百分点和0.69个百分点。表2 对照组模型的评价指标Tab.2 Evaluation indeces of models in control group4.2 mixup数据增强分析本文选用准确率最高的CWT+MixedCNN 方法作为mixup数据增强策略的baseline,其训练损失和测试损失如图6 所示,
23、当迭代训练达到500 次左右时模型基本处于收敛状态,此后验证损失保持在0.6左右上下浮动。图6 CWT+MixingCNN 损失曲线Fig.6 Loss curves of CWT+MixedCNN样本经mixup 数据增强处理之后,送入MixedCNN 训练过程中模型在测试集上的损失曲线如图7所示。当=0.1、0.2、0.4时其测试损失相较于baseline 均无明显改善;当=0.3,且模型处收敛状态时,测试损失稳定在0.5 上下,相较于baseline降低了0.1左右。图7 四种取值下的模型测试损失Fig.7 Model test loss under fourvalues取值对模型准确率
24、的影响如表3 所示,当=0.1、0.2时,相较于baseline 方法准确率降低了0.71 个百分点;当=0.3、0.4 时,模型的准确率分别提高了1.43 个百分点和0.71个百分点。表3对模型准确率的影响Tab.3 Influence ofon accuracy5 结语本文创新性地将深度学习框架引入对运动想象EEG 信号的识别任务当中。首先,利用STFT 和CWT 获得EEG 信号的时频信息,并将其与电极通道的位置信息相结合构成样本数据,然后针对所构建样本数据的特性分别设计了两种基于不同卷积策略的CNN,并在10折交叉验证过程中完成最优超参数的选择。其次,在Competition data
25、set 数据集上分别建立了4组实验组分类模型和4组对照组分类模型,测试结果表明,本文所提出的所有方法相较于CSP+SVM、AAR+LDA、DWT+LSTM 有了明显的改善,且CWT+MixedCNN 方法在准确率、灵敏度、AUC 指标上均高于STFT+SAE,体现了更好的分类性能。最后,为进一步提高模型的泛化能力,在CWT+MixedCNN 方法上引入了mixup 数据增强策略,通过对比实验发现,当取值为0.3 时mixup 策略显著降低了模型的测试损失并将模型的准确率指标提升了1.43 个百分点。本文所做工作验证了深度卷积神经网络和mixup 数据增强策略在运动想象脑电信号识别任务当中的有效性,可为运动想象脑机接口的应用提供方法和思路上的参考。
copyright@ 2008-2022 冰豆网网站版权所有
经营许可证编号:鄂ICP备2022015515号-1