基于约束极速学习机的脑电信号分类方法
【专利摘要】本发明涉及一种基于约束极速学习机的脑电信号分类方法，包括采用固定的滑动时间窗将原始运动想象脑电信号平均分为S段子信号；对每一段子信号通过主成分分析方法进行降维；对降维后的特征向量通过线性判别分析方法再次降维，对于K种类别的脑电数据，得到K-1维的特征向量；对每一段子信号进行处理，得到S个K-1维的特征向量，组合后得到S*(K-1)维的特征；将S*(K-1)维特征送入约束极速学习机(CELM)进行分类。本发明应用CELM通过改变输入层节点与隐层节点之间的权重的选取方式，限制权重参数的随机性，从而既能提高运动想象脑电信号的分类准确率，又能保持ELM训练速度快的优势。
【专利说明】基于约束极速学习机的脑电信号分类方法
【技术领域】
[0001] 本发明属于模式识别领域，涉及一种基于约束极速学习机的脑电信号分类方法。
【背景技术】
[0002] 脑机接口（Brain Computer Interface, BCI)可以让用户使用大脑内的活动而不 是外部的肌肉与神经控制电脑或外部的设备。近几十年来，随着信号处理和机器学习等技 术研究的发展，BCI研究逐渐成为热点。BCI技术不仅为脑疾病患者提供一种新的诊断方 式，更重要的是实现一种人与外界通讯的新途径。运动想象脑电信号在BCI领域很常见，通 过采集分析人们在想象身体某个部位运动或进行某种思维活动时的脑电信号，识别人们大 脑的状态，进而控制外界装置。
[0003] 在BCI研究中，对反应大脑不同思维状态的脑电信号进行识别是研究过程中的一 个重要环节。寻求有效的特征提取方法和分类方法是提高识别准确率的关键。常用的脑电 特征有频带能量，功率谱密度，自回归参数，自适应自回归参数等等。常用的脑电信号分类 方法有支持向量机法，最近邻法，贝叶斯法，人工神经网络法等等。
[0004] 在BCI系统中，脑电信号尤其是实时脑电信号要求较短的处理时间。一些分类算 法只考虑到了较高的准确率忽略了处理耗时。因此，为适用于脑电信号的分类，应寻找一种 算法达到高分类准确率与低处理耗时的平衡非常重要。
[0005] Huang提出的基于单隐层前馈神经网络（Single-hidden Layer Feed forward Neural Network，SLFN)的极速学习机（Extreme Learning Machine，ELM)方法，其训练速 度与BP神经网络以及支持向量机（SVM)相比有明显提升。但ELM完全随机地选择输入层 节点与隐层节点之间的权重，这种随机性会降低了应用ELM对脑电信号进行分类的正确 率。Zhu提出了一种改进ELM的方法称为约束极速学习机（Constrained Extreme Learning Machine，CELM)，约束了 ELM的随机性。在CELM中，输入层节点与隐层节点之间的权重随机 地从相邻两类样本的差分向量中选择，而不是从完全任意的向量中选择，权重选择的随机 性受到了限制，而且仍然保持了 ELM比一般分类方法训练速度快的特性。因此，如果将CELM 应用于脑电信号分类，既可以保持ELM的分类方法训练速度快的优点，又能提高分类的准 确率。
【发明内容】
[0006] 针对现有技术中存在的上述问题，本发明采用改进的ELM即CELM对运动想象任务 的脑电信号进行分类，提高其分类的准确率。
[0007] CELM的输入层节点与隐层节点之间的权重随机地从相邻两类样本的差分向量 (差分向量的示意图如附图2所示）中选择，而不是像ELM那样从完全任意的向量中选择。 ELM中，完全任意的向量选取出的参数会产生更多的隐层节点以满足算法的泛化能力要求。 更多的隐层节点会消耗更多的处理时间，占用更多的计算资源，容易导致过拟合。为避免上 面的问题，本发明中采用CELM的方法对运动想象脑电信号进行分类。
[0008] 本发明所述方法包括以下步骤：
[0009] 步骤1，对脑电信号进行特征提取，得到脑电信号的特征。
[0010] 步骤1.1，采用固定的滑动时间窗将原始运动想象脑电信号分为S段子信号。S的 取值取决于滑动时间窗的长度与原始脑电信号的长度。
[0011] 步骤1. 2,对步骤I. 1所得到的每一段子信号通过主成分分析（Principal Component Analysis, PCA)方法进行降维，得到降维后的特征向量。
[0012] 步骤1. 3,将步骤1. 2中得到的特征向量通过线性判别分析（Liner Discriminate Analysis，LDA)方法再次降维，对于K种类别的脑电数据，得到K-I维的特征向量。
[0013] 步骤1. 4,对每一段子信号均按照步骤1. 2和步骤1. 3进行处理，得到S个K-I维 的特征向量，将这S个K-I维的特征向量进行组合，得到最终为S*(K-I)维的特征。
[0014] 步骤2,将步骤1. 4所得到的S* (K-I)维特征送入CELM进行分类，得到最终分类结 果。
[0015] 与现有技术相比，本发明的方法具有以下优点：
[0016] 传统的ELM方法的输入层节点与隐层节点之间的权重是从完全随机任意的向量 中选取的，会产生更多的隐层节点，降低分类识别率。本发明应用CELM通过改变输入层节 点与隐层节点之间的权重的选取方式，限制权重参数的随机性，从而既能提高运动想象脑 电信号的分类准确率，又能保持ELM训练速度快的优势。实验表明，应用本发明所述方法分 类结果的正确率是92. 78%，优于现有其它方法。同样特征的情况下，采用SVM和ELM的平 均正确率分别为92. 15%和89. 04%。
【专利附图】
【附图说明】
[0017] 图1本发明所涉及方法的主流程图；
[0018] 图2为CELM中差分向量的不意图：(a)为随机样本产生差分向量，（b)为完全随机 向量，（c)为差分向量；
[0019] 图3为CELM方法流程图。
【具体实施方式】
[0020] 下面结合附图和实施实例对本发明做进一步的详细描述。
[0021] 假设有一组训练数据集TrainData和一组测试数据集TestData，TrainData的样 本量为N，维度为D JestData的样本量为M，维度同样为D。其中TrainData与TestData中 样本属于K个类别。
[0022] 图1是本发明所涉及方法的主流程图，具体包括以下步骤：
[0023] 步骤一，采用固定时间窗把TrainData和TestData均分成S段脑电子信号。 TrainData i代表训练数据集中第i段子信号，每段子信号的维度为Di (i = 1，2,…，S)。 TestDatai代表测试数据集中第i段子信号，每段子信号的维度为Di (i = 1，2,…，S)。因为 采用固定时间窗，窗口大小是固定的，所以D1 = D2 =…=Dp
[0024] 步骤二，对步骤一所得到的每一段子信号TrainDatai和TestData i通过主成分分 析方法进行降维。将特征值从大到小进行排序后，再根据累计贡献率，只保留前m个最大特 征值对应的特征向量组合Mpca= [O1, O 2, ...，作为投影空间向量。将TrainDatai和 TestDatai同时投影到MrcA上，可得到PCA降维后的训练数据Train i和测试数据Testi :
[0025] Traini = TrainDatai ?Mpca
[0026] Testi = TestDatai ?Mpca
[0027] 步骤三，对步骤二中得到的特征向量通过LDA方法进行二次降维，具体方法如下：
[0028] (1)根据LDA准则，利用Traini中不同类别样本的类间离散度矩阵以及同一类别 样本的类内离散度矩阵计算出LDA的投影空间向量w'
[0029] (2)把Traini与Testi投影到上，得到第i段脑电子信号的特征：
[0030] TrainFeaturei = Traini ? w*
[0031] TestFeaturei = Testi ? w*
[0032] 步骤四，计算出所有的TrainFeaturei与TestFeature i,并进行组合，得到最终的 特征 TrainFeature 与 TestFeature :
[0033] TrainFeature = (TrainFeature1, TrainFeature2,…，TrainFeatureJ
[0034] TestFeature = (TestFeature1, TestFeature2,…，TestFeatureJ
[0035] 步骤五，用步骤四所得到的特征TrainFeature训练CELM分类器模型，将 TestFeature送入训练好的模型进行分类。流程图如图3所示，具体方法如下：
[0036] (1)给定隐层节点个数L和激励函数g(x)。随机产生输入权值％和偏置值Iv Xi 代表输入的第i个训练样本，Yi代表Xi对应的标签。因为送入分类器的是已经提取好的特 征，故在本发明中，X i实际上代表TestFeature^
[0037] (2)从TrainFeature的两个不同类C1和C2中分别随机取两个样本夂和4 2 '计算 差分向量夂一'。
[0038] (3)如果夂_\足够小，则从差分向量中删除夂-'，继续执行步骤（2);如果 \ _夂与差分向量里面已有的向量平行，则从差分向量中删除弋-戈，继续执行步骤（2)。
[0039] (4)计算标准化差分向量
【权利要求】
1. 基于约束极速学习机的脑电信号分类方法，包括： 步骤1，对脑电信号进行特征提取，得到脑电信号的特征； 处理对象为一组训练数据集TrainData和一组测试数据集TestData，TrainData的样 本量为N，维度为D JestData的样本量为M，维度同样为D ;其中TrainData与TestData中 样本属于K个类别； 步骤1. 1，采用固定时间窗把TrainData和TestData均分成S段脑电子信号； TrainDatai代表训练数据集中第i段子信号，每段子信号的维度为Di, i = 1，2,…，S ; TestDatai代表测试数据集中第i段子信号，每段子信号的维度为Di, i = 1，2,…，S，D1 = D2 =--?= Di = W ； 步骤1. 2,对步骤I. 1所得到的每一段子信号TrainDatai和TestDatai通过主成分分 析方法进行降维；将特征值从大到小进行排序后，再根据累计贡献率，只保留前m个最大特 征值对应的特征向量组合MrcA = [O1, 〇2, ...，Om]作为投影空间向量；将TrainDatai和 TestDatai同时投影到MrcA上，可得到PCA降维后的训练数据Traini和测试数据Test i ; 步骤1. 3,对步骤1. 2中得到的特征向量通过线性判别分析LDA方法进行二次降维： 根据LDA准则，利用Traini中不同类别样本的类间离散度矩阵以及同一类别样本的类 内离散度矩阵计算出LDA的投影空间向量; 把Traini与Testi投影到上，得到第i段脑电子信号的特征： TrainFeaturei = Traini ? w* TestFeaturei = Testi ? w* 步骤I. 4,对每一段子信号均按照步骤I. 2和步骤I. 3进行处理，计算出所有 的TrainFeaturei与TestFeaturei,进行组合后得到最终的特征TrainFeature与 TestFeature ： TrainFeature = (TrainFeature1, TrainFeature2, TrainFeaturexI TestFeature = (TestFeature1, TestFeature2,…，TestFeatureJ 其特征在于还包括： 步骤2,用所述步骤1. 4得到的特征TrainFeature训练约束极速学习机分类器模型，将 TestFeature送入训练好的模型进行分类。
2. 根据权利要求1所述的应用约束极速学习机对脑电信号进行分类的方法，其特征在 于，所述步骤2进行分类的方法还包括以下步骤： 步骤2. 1，给定隐层节点个数L和激励函数g(x);随机产生输入权值ai和偏置值h ;Xi代表输入的第i个训练样本，即TestFeaturei, Yi代表Xi对应的标签； 步骤2. 2,从TrainFeature的两个不同类C1和C2中分别随机取两个样本,和,计算 差分向量; 步骤2. 3,如果夂足够小，则从差分向量中删除\ 继续执行步骤2. 2 ;如果 \ -?与差分向量里面已有的向量平行，则从差分向量中删除夂，继续执行步骤2. 2 ; 步骤2. 4,计算标准化差分向量w和对应的偏置b :
步骤2. 5,用向量w和偏置b构造权重矩阵Wn>a和偏置向量Id1M ; 步骤2. 6,如果差分向量的个数小于L个，重复执行步骤2. 2?2. 5 ; 步骤2. 7,计算隐含层的输出矩阵：
式中，G (a" b" X) = g (X ? afbi); 步骤2. 8,通过最小二乘法计算隐层节点的输出权重矩阵P : ^ LXM = H+Y
步骤2. 9,通过训练分类器模型，将TestFeature送入分类器进行分类，得到一组预测 标签；预测标签与真实标签进行比较求出分类正确率； 步骤2. 10,重复步骤2. 1?2. 9Z次，Z不小于50,计算出Z次的平均分类正确率作为 最终的分类正确率。
【文档编号】G06K9/62GK104361345SQ201410529244
【公开日】2015年2月18日 申请日期:2014年10月10日 优先权日:2014年10月10日
【发明者】段立娟, 续艳慧, 苗军, 马伟, 杨震, 葛卉 申请人:北京工业大学