本发明涉及癫痫检测领域,具体涉及一种基于一维cnn-lstm的多通道脑电图癫痫自动检测装置。
背景技术:
癫痫是一种主要的神经系统疾病,是由于大脑的异常电活动造成的。由于异常放电的起始部位和传递方式的不同,癫痫发作的临床表现复杂多样,可表现为发作性运动、感觉、自主神经、意识及精神障碍。这些发作症状都给患者带来极大的生活不便。
脑电信号中包含了大量的生理与疾病信息,临床上经常通过脑电信号来对某些脑疾病做出分析和诊断。传统癫痫自动检测即是从脑电图中提取出某些特征来对癫痫是否发作进行分类,这能极大减轻医生工作负担,为临床辅助诊断提供帮助。
传统的癫痫检测依赖信号特征的有效提取,特征提取的好坏往往直接影响到最终分类结果。在时域分析方法中,有提取信号幅值、信号熵、信号节律性等研究方法[1]。在频域分析方法中,有先经过傅里叶、小波分解等变换,再提取特征的方法[2]。还有一些时频结合的方法,将时域特征和频域特征相结合,再通过一些数据压缩方式使特征更精简,有效[3]。
[1]j.gotman,“automaticrecognitionofepilepticseizuresintheeeg”clinicalneurophysiology,vol.54,pp.530–540,1982.
[2]meierr,dittrichh,andschulze-bonhagea,“detectingepilepticseizuresinlong-termhumaneeg:anewapproachtoautomaticonlineandread-timedetectionandclassificationofpolymorphicseizurepatterns”journalofclinicalneurophysiology,,pp.119–131.
[3]lingguo,danielrivero,juliándorado,juanr.rabu~nal,andalejandropazos,“automaticepilepticseizuredetectionineegsbasedonlinelengthfeatureandartificialneuralnetworks,”journalofneurosciencemethods191(),pp.101–109.
传统技术存在以下技术问题:
1.传统特征的提取需要人工设计特征,这带来了一些问题:
(1)人工设计特征依赖大量的先验知识,难度较大
(2)额外的特征提取和选择增加了算法的复杂度;
(3)特征设计的好坏直接影响到分类的准确率;
(4)当使用固定的人工设计特征时,难以保持算法的泛化能力;
2.大多基于单通道脑电图,而忽视了其他导联对结果的影响。综合其他导联往往能得到更优异的结果。
3.在数据集上往往能得到不错的分类结果,但在实际临床脑电图中仍然会出现一些漏检和误检,泛化能力不强。
技术实现要素:
本发明要解决的技术问题是提供一种基于一维cnn-lstm的多通道脑电图癫痫自动检测装置,本发明能对多通道静态脑电图进行癫痫发作的自动检测本发明无须人工设计特征,而是基于深度学习技术,通过一维cnn和lstm神经网络来自动学习信号的特征;本发明采用多导联的脑电图信号作为待处理信号。相比只使用单导联信号,多导联信号信息更丰富,得到的结果更平稳,更具有一般性,同组信号,准确率更加稳定;本发明对比其他现有方法,在开源数据库中有更好的分类结果。同时,在实际临床数据中也有良好的效果。
为了解决上述技术问题,本发明提供了一种基于一维cnn-lstm的多通道脑电图癫痫自动检测装置,包括:计算机,所述计算机被编程以便执行如下步骤:
获取采集数据,其中采集数据通过以下方式采集:数据采集电极位置均按照国际标准的10-20系统电极法放置;
对所述采集数据进行小波分解去噪;
将去噪后的长时程的23通道脑电图信号以256个点为一帧来划分;
对每一帧每一通道的数据进行z-score归一化;
给每一帧数据制作对应的标签;
将以上处理过的数据输入深度学习神经网络,其中,所述深度学习神经网络以cnn和lstm的级联为主体;
用测试集进行测试,如果测试集经过训练好的深度学习神经网络输出得到的结果和对应的标签一致,则预测准确;若不一致,则错误。
本发明的有益效果:
传统的癫痫发作检测不同,无需人工设计特征来进行分类,而是直接将多通道的原信号输入训练网络,通过一维cnn和lstm神经网络来自动学习信号的特征,最后进行分类。由于采用了多通道的信号,比只使用单通道信号的方法效果更好,更具稳定性和一般性;除了在数据库中性能表现优异,在实际的临床数据中也有很不俗的效果。
在其中一个实施例中,“获取采集数据,其中采集数据通过以下方式采集:数据采集电极位置均按照国际标准的10-20系统电极法放置”中,脑电信号数据采样率为256hz。
在其中一个实施例中,“获取采集数据,其中采集数据通过以下方式采集:数据采集电极位置均按照国际标准的10-20系统电极法放置”中,所有的脑电信号都是两个电极之间的电压差经过放大后得到。
在其中一个实施例中,“对所述采集数据进行小波分解去噪;”中,利用小波分解提取信号0.5-32hz的频段。
在其中一个实施例中,“将去噪后的长时程的23通道脑电图信号以256个点为一帧来划分;”中,输入数据的大小为n*256*23,其中n表示一共有n帧数据样本;256表示每一帧的单通道数据大小为256个点;23表示一共有23个通道。
在其中一个实施例中,所述z-score归一化公式具体如下:
其中xi表示样本x中的第i个点,mean(x)表示样本信号x的均值,std(x)表示样本信号x的标准差。
在其中一个实施例中,“给每一帧数据制作对应的标签;”中,癫痫发作期对应的帧统一用1表示,其余帧统一用0表示。
在其中一个实施例中,所述深度学习神经网络的具体结构如下:
layer1:1维卷积层,卷积核大小为16*1,步长为2,激活函数为relu;
layer2:1维卷积层,卷积核大小维32*1,步长为2,激活函数为relu;
layer3:最大值池化层,pool_size=2;
layer4:1维卷积层,卷积核大小为64*1,步长为2,激活函数为relu;
layer5:1维卷积层,卷积核大小为128*1,步长为2,激活函数为relu;
layer6:最大值池化层,pool_size=2;
layer7:全连接层,神经元个数为200,含batchnormalization,激活函数为relu;
layer8:lstm层,units=10,activation=tanh,return_sequences=true;
layer9:lstm层,units=10,activation=tanh,return_sequences=true;
layer10:全连接层,神经元个数为100,含batchnormalization,激活函数为relu;
layer11:全连接层,神经元个数为30,含batchnormalization,激活函数为relu;
layer12:softmax层,对结果进行概率预测和分类。
在其中一个实施例中,所述深度学习神经网络的训练方法为adam算法,参数设置如下:学习率lr=0.0005,beta_1=0.9,beta_2=0.999,epsilon=1e-08,clipvalue=0.5。
在其中一个实施例中,所述深度学习神经网络的训练方法为mini-batch梯度下降法,其中batch_size=128,训练次数为200。
附图说明
图1是本发明基于一维cnn-lstm的多通道脑电图癫痫自动检测装置中的计算机执行的流程图。
图2是本发明基于一维cnn-lstm的多通道脑电图癫痫自动检测中的10-20电极放置示意图。
图3是本发明基于一维cnn-lstm的多通道脑电图癫痫自动检测中的数据截取示意图。
图4是本发明基于一维cnn-lstm的多通道脑电图癫痫自动检测中的网络结构示意图。
图5是本发明基于一维cnn-lstm的多通道脑电图癫痫自动检测中的信息提取示意图之一。(s为原信号,a8为0-0.5hz频段信息,d2为32-64hz频段信息,d1为64-128hz频段信息)
图6是本发明基于一维cnn-lstm的多通道脑电图癫痫自动检测中的信息提取示意图之二。(要提取的信号,其中d8,d7,d6,d5,d4,d3分别为0.5-1hz,1-2hz,2-4hz,4-8hz,8-16hz,16-32hz频段的信息)
具体实施方式
下面结合附图和具体实施例对本发明作进一步说明,以使本领域的技术人员可以更好地理解本发明并能予以实施,但所举实施例不作为对本发明的限定。
本发明基于一维cnn-lstm的多通道脑电图癫痫自动检测装置,包括:所述计算机,所述计算机被编程以便执行如下步骤:
s1.数据准备
1.本发明所使用的数据采集电极位置均按照国际标准的10-20系统电极法放置(见图2),所有的脑电信号都是两个电极之间的电压差经过放大后得到。脑电信号数据采样率为256hz。
2.首先对输入信号进行小波分解去噪。利用小波分解提取信号0.5-32hz的频段。
3.将去噪后的长时程的23通道脑电图信号以256个点(1s)为一帧来划分。输入数据的大小为n*256*23,其中n表示一共有n帧数据样本;256表示每一帧的单通道数据大小为256个点;23表示一共有23个通道。
4.参阅图3,为了扩充癫痫发作数据集,对数据进行增强。用滑动窗口连续截取的发作段的两帧数据之间都有四分之三的重复段。
5.对每一帧每一通道的数据进行z-score归一化,这样能在训练深度神经网络时保证梯度不会消失。z-score归一化公式如下,其中xi表示样本x中的第i个点,mean(x)表示样本信号x的均值,std(x)表示样本信号x的标准差。
6.给每一帧数据制作对应的标签,其中癫痫发作期对应的帧统一用1表示,其余帧统一用0表示。因此标签的大小为n*1,n表示总的样本数据组数。
s2.网络搭建
经过s1部分预处理后,数据被规整为统一尺寸n*256*23,即可作为网络模型输入数据了。本方法所采用的深度学习神经网络以cnn和lstm的级联为主体,融合了两种网络的特点和优势,这是此前其他方法所不具备的。cnn-lstm神经网络的具体结构(见图4)如下(layern指神经网络的第n层):
layer1:1维卷积层,卷积核大小为16*1,步长为2,激活函数为relu。
layer2:1维卷积层,卷积核大小维32*1,步长为2,激活函数为relu。
layer3:最大值池化层,pool_size=2。
layer4:1维卷积层,卷积核大小为64*1,步长为2,激活函数为relu。
layer5:1维卷积层,卷积核大小为128*1,步长为2,激活函数为relu。
layer6:最大值池化层,pool_size=2。
layer7:全连接层,神经元个数为200,含batchnormalization,激活函数为relu。
layer8:lstm层,units=10,activation=tanh,return_sequences=true。
layer9:lstm层,units=10,activation=tanh,return_sequences=true。
layer10:全连接层,神经元个数为100,含batchnormalization,激活函数为relu。
layer11:全连接层,神经元个数为30,含batchnormalization,激活函数为relu。
layer12:softmax层,对结果进行概率预测和分类。
训练方法为adam算法,参数设置如下:学习率lr=0.0005,beta_1=0.9,beta_2=0.999,epsilon=1e-08,clipvalue=0.5。采用mini-batch梯度下降法,其中batch_size=128,训练次数为200。
s3.结果处理
用测试集进行测试,如果测试集经过训练好的神经网络输出得到的结果和对应的标签一致,则预测准确(如发作期的标签为1,网络输出结果也为1);若不一致,则错误。所有正确预测的样本数和总样本数之比即是测试集准确率,它是网络性能好坏的重要指标。
更详细的具体应用场景:
s1.预处理阶段
1.对信号进行小波分解去噪。选取db5小波作为母小波,分解级数为8级,这样可以提取到0.5-32hz的信号(见图5,图6)。
2.以256个点为一帧,对去噪后的信号进行分帧,并制作标签。其中,发作段用标签1表示;未发作段用标签0表示。每一帧数据的大小是256*23,假设共有n帧数据,则数据组的总大小为n*256*23,标签数据组的大小为n*1。
3.将得到的n组数据随机打乱,选取其中五分之四的数据作为训练集,其余五分之一作为测试集。
4.将每一组数据按各自通道进行z-score归一化。
s2.训练阶段
1.将训练数据输入到上述网络模型中进行训练,训练方法为adam算法,参数设置如下:学习率lr=0.0005,beta_1=0.9,beta_2=0.999,epsilon=1e-08,clipvalue=0.5。采用mini-batch梯度下降法,其中batch_size=128,训练次数为200。保存训练得到的网络模型。训练方法为adam算法,这是梯度下降法的一种变形,本质也是梯度下降法,它优化的是方法;mini-batch是处理训练数据的方法,一次训练不训练所有的数据,只训练一批数据。这两个是一起用的。
s3.测试阶段
1.载入得到的模型,将测试数据输入网络,得到测试结果。假设有测试数据m组,则网络会输出只含有0或1的,大小为m*1的数组。每组数据的输出值若与其对应的标签一致,则网络预测结果正确。
2.采用无重复抽样的五折交叉验证的方法,即将总数据平均分成五份,每一份轮流做测试集,其余的做测试集。交叉验证的方法能够找到使得模型泛化性能最优的超参值。
3.预测的准确率越高,则网络性能越好。
本发明有益效果:
1.相比传统单通道数据的处理和分析,本发明使用了23通道的数据作为处理对象。综合考虑各个通道的特性和变化,让实验结果更加准确可靠。
2.无需手动设计特征,本方法设计的一维cnn-lstm级联深度学习神经网络能够自动地学习样本信号的特征。
3.本发明将cnn和lstm级联,通过cnn学习输入信号的特征,通过lstm网络来捕捉前序特征的时间相关特性。整个结构融合了这两个网络的优点,提高了癫痫发作检测的准确率和网络的泛化能力,这是之前相关研究中从来没有的。
以上所述实施例仅是为充分说明本发明而所举的较佳的实施例,本发明的保护范围不限于此。本技术领域的技术人员在本发明基础上所作的等同替代或变换,均在本发明的保护范围之内。本发明的保护范围以权利要求书为准。
技术特征:
1.一种基于一维cnn-lstm的多通道脑电图癫痫自动检测装置,其特征在于,包括:计算机,所述计算机被编程以便执行如下步骤:
获取采集数据,其中采集数据通过以下方式采集:数据采集电极位置均按照国际标准的10-20系统电极法放置;
对所述采集数据进行小波分解去噪;
将去噪后的长时程的23通道脑电图信号以256个点为一帧来划分;
对每一帧每一通道的数据进行z-score归一化;
给每一帧数据制作对应的标签;
将以上处理过的数据输入深度学习神经网络,其中,所述深度学习神经网络以cnn和lstm的级联为主体;
用测试集进行测试,如果测试集经过训练好的深度学习神经网络输出得到的结果和对应的标签一致,则预测准确;若不一致,则错误。
2.如权利要求1所述的基于一维cnn-lstm的多通道脑电图癫痫自动检测装置,其特征在于,“获取采集数据,其中采集数据通过以下方式采集:数据采集电极位置均按照国际标准的10-20系统电极法放置”中,脑电信号数据采样率为256hz。
3.如权利要求1所述的基于一维cnn-lstm的多通道脑电图癫痫自动检测装置,其特征在于,“获取采集数据,其中采集数据通过以下方式采集:数据采集电极位置均按照国际标准的10-20系统电极法放置”中,所有的脑电信号都是两个电极之间的电压差经过放大后得到。
4.如权利要求1所述的基于一维cnn-lstm的多通道脑电图癫痫自动检测装置,其特征在于,“对所述采集数据进行小波分解去噪;”中,利用小波分解提取信号0.5-32hz的频段。
5.如权利要求1所述的基于一维cnn-lstm的多通道脑电图癫痫自动检测装置,其特征在于,“将去噪后的长时程的23通道脑电图信号以256个点为一帧来划分;”中,输入数据的大小为n*256*23,其中n表示一共有n帧数据样本;256表示每一帧的单通道数据大小为256个点;23表示一共有23个通道。
6.如权利要求1所述的基于一维cnn-lstm的多通道脑电图癫痫自动检测装置,其特征在于,所述z-score归一化公式具体如下:
其中xi表示样本x中的第i个点,mean(x)表示样本信号x的均值,std(x)表示样本信号x的标准差。
7.如权利要求1所述的基于一维cnn-lstm的多通道脑电图癫痫自动检测装置,其特征在于,“给每一帧数据制作对应的标签;”中,癫痫发作期对应的帧统一用1表示,其余帧统一用0表示。
8.如权利要求1所述的基于一维cnn-lstm的多通道脑电图癫痫自动检测装置,其特征在于,所述深度学习神经网络的具体结构如下:
layer1:1维卷积层,卷积核大小为16*1,步长为2,激活函数为relu;
layer2:1维卷积层,卷积核大小维32*1,步长为2,激活函数为relu;
layer3:最大值池化层,pool_size=2;
layer4:1维卷积层,卷积核大小为64*1,步长为2,激活函数为relu;
layer5:1维卷积层,卷积核大小为128*1,步长为2,激活函数为relu;
layer6:最大值池化层,pool_size=2;
layer7:全连接层,神经元个数为200,含batchnormalization,激活函数为relu;
layer8:lstm层,units=10,activation=tanh,return_sequences=true;
layer9:lstm层,units=10,activation=tanh,return_sequences=true;
layer10:全连接层,神经元个数为100,含batchnormalization,激活函数为relu;
layer11:全连接层,神经元个数为30,含batchnormalization,激活函数为relu;
layer12:softmax层,对结果进行概率预测和分类。
9.如权利要求1所述的基于一维cnn-lstm的多通道脑电图癫痫自动检测装置,其特征在于,所述深度学习神经网络的训练方法为adam算法,参数设置如下:学习率lr=0.0005,beta_1=0.9,beta_2=0.999,epsilon=1e-08,clipvalue=0.5。
10.如权利要求1所述的基于一维cnn-lstm的多通道脑电图癫痫自动检测装置,其特征在于,所述深度学习神经网络的训练方法为mini-batch梯度下降法,其中batch_size=128,训练次数为200。
技术总结
本发明公开了一种基于一维CNN‑LSTM的多通道脑电图癫痫自动检测装置。本发明一种基于一维CNN‑LSTM的多通道脑电图癫痫自动检测装置,包括:所述计算机,所述计算机被编程以便执行如下步骤:获取采集数据,其中采集数据通过以下方式采集:数据采集电极位置均按照国际标准的10‑20系统电极法放置。本发明的有益效果:与传统的癫痫发作检测不同,无需人工设计特征来进行分类,而是直接将多通道的原信号输入训练网络,通过一维CNN和LSTM神经网络来自动学习信号的特征,最后进行分类。由于采用了多通道的信号,比只使用单通道信号的方法效果更好,更具稳定性和一般性;除了在数据库中性能表现优异,在实际的临床数据中也有很不俗的效果。
技术研发人员:王丽荣;陈雪勤;俞杰;邱励燊;蔡文强;李婉悦;郑乐松;邓米雪;张淼;陈颖
受保护的技术使用者:苏州大学
技术研发日:.12.02
技术公布日:.02.28
如果觉得《基于一维CNN-LSTM的多通道脑电图癫痫自动检测装置的制作方法》对你有帮助,请点赞、收藏,并留下你的观点哦!