本发明涉及一种癫痫脑电识别装置,属于cnn-svm技术和医疗疾病诊断领域的融合。
背景技术:
癫痫会影响所有年龄人群,是一种脑部慢性疾患。大概70%的患者通过服用抗癫痫药物能有效抑制癫痫的发作(不能根治),但其余患者在服用药物后却没有效果,而且长期服用抗癫痫药物还会有副作用。目前,在医学疾病诊断领域,癫痫的诊断是通过获得详细的病史,进行神经学检查以及诸如神经成像和eeg的辅助测试来进行的,也即神经科医生通过直接目视检查脑电图(eeg),以研究癫痫样异常。视觉检查的准确率是取决于神经科医生在脑电图研究中的专业水平,并且研究的质量可能因人为信号的干扰而受到限制,这限制了医生准确识别异常的能力。癫痫预测的机器学习模型主要分为两大类,一类是线性模型,一类是非线性模型。因为脑电信号的混沌性和不稳定性,非线性方法在脑电信号的分析上有更大的成长空间。非线性两个最流行的方法就是卷积神经网络(convolutionalneuralnetwork,cnn)和支持向量机(supportvectormachine,svm)。前者的卷积层在特征提取上有较好的性能,后者在二分类的处理上有独特优势。神经学家能够通过直接观察脑电波进行相应的诊断。但是这种方法需要长时间的观察,非常消耗时间和成本,效率和准确率都很低。因此基于脑电图(eeg)的自动化分析来实现癫痫分类和预测是非常有意义的。本发明提出了一种基于多通道脑电数据和cnn-svm的癫痫脑电识别装置,以提高癫痫疾病诊断的效率和降低误诊率。
技术实现要素:
本发明的目的是提供一种基于多通道脑电数据和cnn-svm的癫痫脑电识别装置。技术方案如下:
一种基于多通道脑电数据和cnn-svm的癫痫脑电识别装置,包括下列模块:
预处理模块,用于对于癫痫病人正常期和发作期的脑电数据,使用3阶巴特沃斯带通滤波器(0.3-70hz)
消除运动伪影和肌电干扰;使用陷波滤波器去除50hz工频干扰。
特征提取模块,包括:卷积层:使用卷积层从脑电信号中提取特征,并选用relu函数作为激活函数;
池化层:采用最大值池化的方式;丢失正则化:为防止过拟合,应用丢失正则化机制;
模型构造模块,包括:
1)基于svm构造2分类模型;
2)把特征提取模块提取到的特征作为数据集,构建训练集和测试集;
3)使用sigmoid函数作为核函数。
模型训练模块,包括:
1)使用梯度下降和反向传播训练模型,寻找最优超平面;
2)使用10折交叉验证方式训练模型。
模型测试及保存模块
1)使用测试集测试模型的分类准确率;
2)保存准确率最高的模型,将其用于癫痫脑电识别。
附图说明
图1正常脑电信号和癫痫脑电信号对比图,(a)正常脑电信号(b)癫痫脑电信号。
图2本发明基于cnn-svm的癫痫脑电识别装置结构框图
图3本发明特征提取模块及模型构建模块框图
具体实施方式
下面结合附图和具体实例对本发明进行进一步的说明阐述。
本发明采用23通道的脑电信号,脑电信号分为正常脑电和癫痫脑电信号。采用cnn-svm构造2分类模型,利用该模型分类正常脑电和癫痫脑电,从而实现较高准确率的癫痫脑电识别
参见图1,本发明所使用的原始脑电图。信号来自癫痫病人正常期和发作期的脑电数据,采样频率为1024hz,单个信号长度为2048。其中,(a)为正常脑电信号,(b)为癫痫脑电信号,可以看出两类信号有明显的分类特征,因此可以使用cnn-svm对其进行分类,从而实现癫痫疾病的识别。
参见图2和图3,本发明基于cnn-svm的癫痫脑电识别装置实现框图和本发明特征提取模块及模型构建模块的框图。以下简述发明具体实施步骤。
1.预处理模块
1)使用3阶巴特沃斯带通滤波器(0.3-70hz)消除运动伪影和肌电干扰;
2)使用陷波滤波器去除50hz工频干扰。
2.特征提取模块
1)卷积层。第一层和第二层是卷积层,输入矩阵和卷积核进行卷积操作,进而提取出数据特征,卷积运算如公式(1)所示。由于卷积操作是线性运算只能进行对线性可分数据分类,但是从所使用的脑电数据是非平稳和非线性可分的,为使模型能够解决非线性问题,需要对其加入非线性因素,即在两个卷积层后增加使用如公式(2)所示的relu作为激活函数的两个非线性激活层,relu函数具有快速训练特性,所以比公式(3)所示的sigmoid函数更有优势
其中,x(t)表示输入eeg信号,w(t)表示权值矩阵,y(t)表示输出信号
relu(x)=max(0,x)(2)
2)池化层。采用最大值池化的方式,如公式(4),该操作即选取池化窗口对应的输入矩阵的最大值作为输出矩阵相应元素的值,它处于连续的卷积层中间,主要是用来减小特征图,把其中的冗余信息去掉,提取最重要的特征,对数据起到降低维度的作用。
f(x)=max(0,x)(4)
3)丢失正则化。为防止过拟合,应用了丢失正则化机制,使得网络的隐藏层以0.1的概率随机丢弃。且丢失机制仅在训练时激活,这使网络变得简单高效,增加了网络的泛化能力。
3.模型构造模块
1)基于svm构造出2分类模型;
2)把特征提取模块提取到的特征作为数据集,选取900个样本作为训练集,100个样本作为测试集;
3)使用公式(3)的sigmoid函数作为核函数。
4.模型训练模块
1)使用梯度下降和反向传播训练模型,寻找最优超平面。
2)使用10折交叉验证方式训练模型,每次训练中,900个样本中有90%被随机选为训练数据集,其余10%被选为验证集,该过程重复20次;
5.模型测试及保存模块
1)使用测试集测试模型的分类准确率;
2)保存准确率最高的模型,将其作为癫痫脑电识别装置。
癫痫脑电识别装置的评估结果:
1)准确率:95.1%
本发明的有益效果如下:
1)本发明设计了一种癫痫脑电识别装置,避免了直接目视检查脑电图,降低了诊断癫痫疾病的成本,具有较高的识别准确率。
2)由未经过训练的数据测试该装置的识别准确率,测试结果和训练结果基本一致,表明本发明装置具有较好的泛化能力。
3)结合cnn和svm算法各自的优点,为癫痫脑电识别提出了一种新思路。