一种基于深度极速学习机的脑电信号分类方法及系统与流程

本发明涉及模式识别和机器学习领域，具体涉及一种基于深度极速学习机的脑电信号分类方法及系统。

背景技术：

脑机接口是基于脑电信号实现人脑与计算机或其它电子设备通讯和控制的技术。脑机接口的核心技术是对预处理的脑电信号进行特征提取和对已经提取的脑电信号进行分类。近年来，随着计算机技术的发展，脑机接口得到了突飞猛进的进展，脑机接口应用到医学、军事等许多领域。目前，脑电信号的分类方法有决策树、贝叶斯分类器、k近邻分类器、支持向量机、极速学习机等。其中多数分类算法是浅层神经网络算法，对复杂函数的表示能力有限，而深度学习没有这种限制。

深度学习是一种多隐层多层感知器的人工神经网络学习算法，实现复杂函数的逼近并缓解了以前多层神经网络算法的局部最小性问题。2006年，深度学习的概念被首次提出。随后，深度置信网、堆叠自动编码器这些深度网络模型被提出。2013年，有学者在自动编码器和极速学习机基础上提出多层极速学习机模型。极速学习机是一种高效简洁的单隐层前馈神经网络的学习算法。

脑电信号需要在受试者大脑皮层采集，需要预处理加工。在经典的模式识别中，一般是事先提取特征，预处理的脑电信号提取到的特征的不同对分类性能影响很大。深度极速学习机是一种有效且快速的深度神经网络学习方法，通过堆叠极速学习机-自动编码器算法构建多层神经网络模型，最后把学习到的特征映射到核空间分类。深度极速学习机是一种有效且快速的深度神经网络学习方法，通过堆叠极速学习机-自动编码器算法构建多层神经网络模型，最后把学习到的特征映射到核空间分类。深度极速学习机不用对数据进行复杂的迭代操作，可以方便地把数据作为输入，通过大量的数据一次性学习得到有效的特征，然后把学习到的特征映射到核空间分类。

技术实现要素：

为了解决脑电信号分类问题，本发明提出一种基于深度学习机的脑电信号分类方法及系统，直接将脑电信号预处理后的数据作为输入，训练得到深度极速学习机神经网络模型，该模型不仅可以对预处理后的脑电信号进行多次非线性映射，实现脑电信号的深度特征提取，还能把学习到的特征映射到核空间分类，从而得到每个脑电信号对应的标签，明显提高了识别的正确率和效率。

本发明是通过以下方案实现的：

本发明涉及一种基于深度极速学习机的脑电信号分类方法，通过构建带标签的训练集作为样本集对深度极速学习机网络进行训练，并将训练好的神经网络处理预处理后的脑电信号，最后根据神经网络的输出向量判断分类结果。

本发明具体步骤如下：

步骤1，制作训练集并将脑电信号做预处理：根据未加工的脑电信号，制作自然风景图像的训练集和标签集，训练集为经过预处理后的数据。以视觉反馈为例，每个预处理后的脑电信号有5376个特征属性，它是采集6个通道3.5秒的脑电信号得到的，每个通道采样率都是256赫兹。

步骤2，构造三隐层的深度极速学习机神经网络：输入层的数据是预处理得到的脑电信号(以视觉反馈为例，每个预处理后的脑电信号有5376个特征属性，类别数为2，因此输入层与输出层的隐层节点数分别为5376与2)，前两个隐层节点数人为指定为500(其权值用极速学习机-自动编码器学习)，此时结束了特征提取的过程，随后将学习到的特征映射到核空间得到最后一层隐层的输出，因此第三层的隐层节点数不需要人为指定。

步骤3，训练深度极速学习机神经网络，网络的基本参数和规格设置完成后，首先利用极速学习机-自动编码器学习前两层隐层的连接权值，随后计算训练数据第二层隐层的输出作为最后一层隐层对应的权值，最后结合最小二乘方法根据标签与最后一层隐层的输出计算网络的输出权值，具体过程如下：

步骤3.1：计算输入层与第一层隐层间的权值：建立一个极速学习机-自动编码器的网络模型，其输入与输出都是预处理完的数据，其隐层节点数为多层神经网络第一层隐层的节点数500，网络的输入权值随机确定，用最小二乘方法一次计算出输出权值，最后此输出权值的转置对应的就是输入层与第一层隐层间的权值；

步骤3.2：计算第一层隐层与第二层隐层间的权值：同样建立一个极速学习机-自动编码器的网络模型，其输入与输出都是第一层隐层的输出，其隐层节点数为多层神经网络第二层隐层的节点数500，网络的输入权值随机确定，用最小二乘方法一次计算出输出权值，最后此输出权值的转置对应的就是第一层隐层与第二层隐层间的权值；

步骤3.3：计算第二层隐层与第三层隐层间的权值：将学习到的特征映射到核空间得到最后一层隐层的输出，所以第二层隐层与第三层隐层间的权值就是训练数据第二层隐层的输出；

步骤3.4：计算最后一层隐层与输出层间的权值：利用最小二乘算法根据最后一层隐层的输出与对应的标签计算最后一层隐层与输出层间的权值。

步骤4：脑电信号的分类，以以视觉反馈为例，每个预处理后的脑电信号有5376个特征属性，然后将其输入到训练好的神经网络中，最终得到对应的一组标签。

通过以上内容可知，本申请提供的是一种基于深度极速学习机的脑电信号分类方法及系统，首先制作根据实际需要制作训练集和标签，然后设计网络的层数、每层的节点数目等等，之后将训练集预处理后输入网络中完成对网络权值的学习，最后输入脑电信号，预处理后输入神经网络，完成对脑电信号的分类。本申请通过深度极速学习机神经网络识别脑电信号，实现了多层的特征提取，直接将脑电信号预处理后的向量作为网络的输入，分类准确率很高；且网络一经训练即可反复使用，处理效率高；训练时间短。

附图说明

为了更清楚地说明本发明实施例或现有技术中的技术方案,下面将对实施例或现有技术描述中所需要使用的附图作简单地介绍,显而易见地,下面描述中的附图仅仅是本发明的一些实施例,对于本领域普通技术人员来讲,在不付出创造性劳动的前提下,还可以根据这些附图获得其他的附图。

图1为本申请实施例提供的一种脑电信号分类方法的流程图。

图2为本申请所使用的深度极速学习机神经网络结构示意图。

具体实施方式

下面将结合本申请实施例中的附图,对本申请实施例中的技术方案进行清楚、完整地描述,显然,所描述的实施例仅仅是本申请一部分实施例,而不是全部的实施例。基于本申请中的实施例,本领域普通技术人员在没有付出创造性劳动前提下所获得的所有其它实施例,都属于本申请保护的范围。

实施例1

如图1所示，本实施例包括以下步骤：

步骤1，脑电信号预处理：

步骤1.1：确定每次试验的通道数以及每个通道采样率，视觉反馈试验中有6个通道，每个通道采样率都是256赫兹；

步骤1.2：确定每次试验的有效时长，视觉反馈试验中的视觉反馈是从第2秒到第5.5秒，每次试验只有这3.5秒的时间间隔用于训练或测试，因此每个预处理后的脑电信号有5376个特征属性，每个预处理后的脑电信号对应一组标签，2*1的矩阵代表每个预处理后的脑电信号的标签。

步骤2，构建深度极速学习机神经网络：

本实施例中采用的深度极速学习机-径向基神经网络是一个多层的神经网络，由输入层、隐层和输出层等多层组成。

步骤2.1：前两层隐层节点数人为指定为500；

步骤2.2：随后将第二层隐层的输出映射到核空间得到最后一层隐层的输出，因此第三层的隐层节点数不需要人为指定，输出层设置为2个节点。

步骤3，训练多层极速学习机-径向基神经网络：

步骤3.1：利用极速学习机-自动编码器的网络计算前两层隐层的连接权值；

步骤3.2：将学习到的特征映射到核空间得到最后一层隐层的输出，所以第二层隐层与第三层隐层间的权值就是训练数据第二层隐层的输出；

步骤3.3：利用最小二乘算法根据最后一层隐层的输出与对应的标签计算最后一层隐层与输出层间的权值。

步骤4，脑电信号的分类：

步骤4.1：在得到脑电信号之后，首先进行预处理，视觉反馈试验中的脑电信号预处理后可以得到5376*1的向量；

步骤4.2：得到预处理之后的数据，输入到已经训练保存好的神经网络中，等待输出，然后依据输出向量完成对脑电信号的分类。