一种基于卷积神经网络和EEG的大脑疲劳检测方法技术

本发明专利技术公开了一种基于卷积神经网络和EEG的大脑疲劳检测方法，步骤包括：通过傅里叶变换对不同频段的EEG信号进行提取；EEG数据输入卷积神经网络之后，经卷积池化提取的图像特征，经过全连接层转换为一维数据，将经过卷积神经网络处理后的数据作为SVM的输入，对SVM模型进行训练；使用SVM模型对经过特征提取的EEG信号进行分类，得到分类结果。本发明专利技术使用SVM代替传统网络的softmax层，使得在数据量较少的情况下，也可获得更高的准确率，进行特征提取时使用DenseNet，可以充分利用脑电图特征，并且可以减轻梯度消失问题，相比其他的网络，DenseNet所需要的参数数量较少，可以降低存储开销并且减少计算量，提高效率。提高效率。提高效率。

【技术实现步骤摘要】
一种基于卷积神经网络和EEG的大脑疲劳检测方法


[0001]本专利技术涉及大脑疲劳检测方法
，具体为一种基于卷积神经网络和EEG的大脑疲劳检测方法。

技术介绍

[0002]脑电图(EEG)是脑神经细胞电生理活动在大脑皮层或头皮表面的总体反映。脑电信号中包含了大量的生理与疾病信息，在临床医学方面，脑电信号处理可为某些脑疾病提供诊断依据和治疗手段。脑电信号可用于检测人的情感状态、睡眠质量、癫痫发作检测等方面。脑电信号在每种大脑状态时表现不同的波形，睡眠时的脑电信号时身体自发产生的，可以反映出大脑在睡眠时的生理活动，通过对脑电信号的分析可对睡眠质量进行判断，从而对大脑疲劳程度进行判断。
[0003]脑电信号中包含时域、频域和空间域信息，由于时域信息直观性强、物理意义比较明确，直接从时域上提取EEG特征信息是最早发展起来的特征，过去的EEG分析主要靠肉眼观察。随着计算机图像处理的发展，可通过计算机对EEG数据进行分析，分析方法有：K近邻、支持向量机、决策树、人工神经网络等。在多数实验中使用ANN进行EEG分析有更高的准确率。
[0004]卷积神经网络(CNN)是一类包含卷积计算且具有深度结构的前馈神经网络。CNN的输入层可以处理多维数据，由于使用梯度下降算法进行学习，CNN的输入特征需要进行标准化处理。处理后的数据通过隐含层，隐含层主要包括卷积层、池化层和全连接层，随着CNN的发展在一些模型中可能包含例如Inception模块、残差块等复杂结构。卷积层主要是对输入数据进行特征提取，内部包含多个卷积核，该层参数包括卷积核大小、步长和填充。卷积层中包含激活函数以协助表达复杂特征；池化层可对数据进行降维，减少计算量且防止过拟合。全连接层的作用则是对提取的特征进行非线性组合以得到输出。输出层对于不同的问题可设置不同的输出，对于图像分类问题，输出层设置为输出分类标签，物体识别问题，可设计为目标物体的坐标。
[0005]随着卷积神经网络的发展，许多经典模型被提出，如：AlexNet模型、VGGNet模型、GoogleNet模型、ResNet模型和DenseNet模型。卷积神经网络层数的不断增加，出现梯度消失和模型退化问题，ResNet模型的核心思想是建立前面层与后面层之间的短路连接，解决了神经网络中的退化问题。DenseNet的基本思路与ResNet一致，它是建立前面所有层与后面层的密集连接。DenseNet通过特征在通道维度上的连接实现特征重用，加强网络中特征的传递，有效的对特征进行了复用，DenseNet减轻了梯度消失问题并且在一定程度上减少了网络的参数数量。
[0006]在样本较少的情况下，利用SVM进行分类的效果要优于其他分类器，在此基础上，许多研究者开始使用SVM代替卷积神经网络中的SoftMax分类器，使图像分类的识别准确率进一步提高。
[0007]支持向量机是依据统计学理论的算法，是传统机器学习中最常用的算法之一，属
于监督学习。SVM在处理小样本数据集的问题中显示出很好的优势，在保证性能的同时，使用该算法还能够将学习过程中用到的过拟合以及维数爆炸等问题。
[0008]SVM最初的目的是为了寻找一个分类效果比较好的平面，使样本中的数据和该平面相距较远，能够把两种不同类别的样本分开，并且在其两侧各存在一个与决策边界平行的超平面，此时，两超平面之间的距离被称为间隔。
[0009]SVM分类器相比其他分类器，在数据量较少的情况下，获得较高的识别准确率，效率也比较高，同时还能防止过拟合现象的发生；因为许多分类任务大多都是非线性分类的，所以为了解决此类问题我们引入核函数，在保证计算量复杂度都不增加的基础上，将许多在低维空间不可分的问题转化为在高维空间中的线性可分问题；能够利用不同的核函数实现对数据集的分类；对于传统机器学习算法无法解决最优解的问题进行优化处理。
[0010]最近实验表明，以卷积神经网络为基础对EEG信号进行分析要优于其他的算法，可以有效提高对大脑疲劳进行分类的准确度和精度。通过引入DenseNet，相对于仅仅使用经典的卷积神经网络对睡眠时产生的脑电信号进行分析，可更有效的利用EEG信号所包含的特征，将经卷积神经网络学习到的EEG特征输入SVM能够实现对大脑疲劳状态的分类。

技术实现思路

[0011]本专利技术的目的在于提供一种基于卷积神经网络和EEG的大脑疲劳检测方法，该方法的步骤包括：
[0012]第一步，将EEG数据传入卷积神经网络输入层：通过傅里叶变换对不同频段的EEG信号进行提取；
[0013]EEG信号可分为α波、β波、δ波和θ波四种基本特征波，通过傅里叶变换对四种基本特征波进行提取。
[0014]第二步，提取EEG特征：EEG数据输入卷积神经网络之后，经卷积池化提取的图像特征，经过全连接层转换为一维数据；
[0015]为了更好的对脑电信号特征进行提取，使用具有特征重用特点的DenseNet对特征进行提取。DenseNet中具有密集连接机制，每一层都获得来自前面所有层的输出，将输出进行连接后作为该层的输入，DenseNet模型可对特征进行充分利用正是由于这种密集链接机制，该网络还在一定程度上减少了参数数量且减轻梯度消失问题的出现；
[0016]在DenseNet中，每一层的输入均来自前面的所有层，即第i层的输入与之前所有层的输出相关，假设x
i
表示第i层输出，则x
i
可以表示为
[0017]H
i
([x0,x1,
…
,x
i―1
]),i＝1,2,
…
,n；
[0018][]代表拼接操作，H
i
(
·
)表示非线性转化函数，是一个组合操作，DenseNet主要由密集块和过渡层构成，密集块中包含多层，每层的特征图大小同，通道数发生改变；过渡层是将不同的密集块进行连接的模块，主要用于整合密集块获得的特征；
[0019]在密集块中采用的非线性组合函数H
i
(
·
)是BN、激活函数和3
×
3卷积，假定输入的特征图的数量为k0，密集块各层卷积后均输出k个特征图，则第n层共有k0+(n―1)k个输入的特征图，其中k为网络增长率，在密集块中可加入 Bottleneck层，即增加BN、激活函数、1
×
1卷积，可减少输入的特征图数量，融合各通道特征；
[0020]过渡层用于密集块之间的连接，过渡层由BN、激活函数、1
×
1卷积和2
×
2平均池化
组成，由于其压缩模型的作用，假设密集块输出的特征图数量为m，经过过渡层后可产生θm个输出特征图，θ∈(0,1]为压缩因子。
[0021]第三步，构建SVM分类器：将经过卷积神经网络处理后的数据作为SVM的输入，对SVM模型进行训练；
[0022]将SVM算法模型作为分类器代替卷积神经网络中的S本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于卷积神经网络和EEG的大脑疲劳检测方法，其特征在于：该方法的步骤包括：第一步，将EEG数据传入卷积神经网络输入层：通过傅里叶变换对不同频段的EEG信号进行提取；第二步，提取EEG特征：EEG数据输入卷积神经网络之后，经卷积池化提取的图像特征，经过全连接层转换为一维数据；第三步，构建SVM分类器：将经过卷积神经网络处理后的数据作为SVM的输入，对SVM模型进行训练；第四步，EEG信号预测结果：使用SVM模型对经过特征提取的EEG信号进行分类，得到分类结果。2.根据权利要求1所述的一种基于卷积神经网络和EEG的大脑疲劳检测方法，其特征在于：EEG信号可分为α波、β波、δ波和θ波四种基本特征波，通过傅里叶变换对四种基本特征波进行提取。3.根据权利要求1所述的一种基于卷积神经网络和EEG的大脑疲劳检测方法，其特征在于：通过卷积神经网络对EEG信号特征进行提取，使用DenseNet对特征进行提取，DenseNet通过特征在通道维度上的连接实现特征重用并且可以减轻梯度消失问题；在DenseNet中，每一层的输入均来自前面的所有层，即第i层的输入与之前所有层的输出相关，假设x
i
表示第i层输出，则x
i
可以表示为H
i
([x0,x1,
…
,x
i
‑1]),i＝1,2,
…
,n；[]代表拼接操作，H
i
(
·
)表示非线性转化函数，是一个组合操作，DenseNet主要由密集块和过渡层构成，密集块中包含多层，每层的特征图大小同，通道数发生改变；过渡层是将不同的密集块进行连接的模块，主要用于整合密集块获得的特征；在密集块中采用的非线性组合函数H
i
(
·
)是BN、激活函数和3
×
3卷积，假定输入的特征图的数量为k0，密集块各层卷积后均输出k个特征图，则第n层共有k0+(n―1)k个输入的特征图，其中k为网络增长率，在密集块中可加入Bottleneck层，即增加BN、激活函数、1
×
1卷积，可减少输入的特征图数量，融合各通道特征；过渡层用于密集块之间的连接，过渡层由BN、激活函数、1
×
1卷积和2
×
...

【专利技术属性】
技术研发人员：王元全，杨俊丽，
申请(专利权)人：天津宇迪智能技术有限公司，
类型：发明
国别省市：