CN112686091B - 基于深度神经网络的两步骤心律失常分类方法 - Google Patents
基于深度神经网络的两步骤心律失常分类方法 Download PDFInfo
- Publication number
- CN112686091B CN112686091B CN202011319283.8A CN202011319283A CN112686091B CN 112686091 B CN112686091 B CN 112686091B CN 202011319283 A CN202011319283 A CN 202011319283A CN 112686091 B CN112686091 B CN 112686091B
- Authority
- CN
- China
- Prior art keywords
- heartbeat
- data
- training
- heart beat
- gan
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Active
Links
- 238000000034 method Methods 0.000 title claims abstract description 31
- 206010003119 arrhythmia Diseases 0.000 title claims abstract description 28
- 230000006793 arrhythmia Effects 0.000 title claims abstract description 28
- 238000013528 artificial neural network Methods 0.000 title claims abstract description 21
- 238000012549 training Methods 0.000 claims description 54
- 238000013145 classification model Methods 0.000 claims description 28
- 238000012360 testing method Methods 0.000 claims description 22
- 238000009826 distribution Methods 0.000 claims description 14
- 206010047289 Ventricular extrasystoles Diseases 0.000 claims description 5
- 238000007781 pre-processing Methods 0.000 claims description 5
- 238000002790 cross-validation Methods 0.000 claims description 4
- 125000004122 cyclic group Chemical group 0.000 claims description 3
- 230000008569 process Effects 0.000 claims description 3
- 230000003416 augmentation Effects 0.000 claims description 2
- 238000005520 cutting process Methods 0.000 claims description 2
- 230000004927 fusion Effects 0.000 claims description 2
- 208000000418 Premature Cardiac Complexes Diseases 0.000 claims 2
- 230000002457 bidirectional effect Effects 0.000 claims 1
- 210000005242 cardiac chamber Anatomy 0.000 claims 1
- 210000002569 neuron Anatomy 0.000 description 6
- 238000003745 diagnosis Methods 0.000 description 5
- 238000010586 diagram Methods 0.000 description 4
- 238000011156 evaluation Methods 0.000 description 4
- 230000033764 rhythmic process Effects 0.000 description 4
- 238000012795 verification Methods 0.000 description 4
- 230000009286 beneficial effect Effects 0.000 description 3
- 238000005516 engineering process Methods 0.000 description 3
- ORILYTVJVMAKLC-UHFFFAOYSA-N Adamantane Natural products C1C(C2)CC3CC1CC2C3 ORILYTVJVMAKLC-UHFFFAOYSA-N 0.000 description 2
- 239000010754 BS 2869 Class F Substances 0.000 description 2
- 206010042602 Supraventricular extrasystoles Diseases 0.000 description 2
- 238000011161 development Methods 0.000 description 2
- 230000000694 effects Effects 0.000 description 2
- 230000006870 function Effects 0.000 description 2
- 238000004519 manufacturing process Methods 0.000 description 2
- 238000011176 pooling Methods 0.000 description 2
- 238000012545 processing Methods 0.000 description 2
- 238000005070 sampling Methods 0.000 description 2
- 238000000638 solvent extraction Methods 0.000 description 2
- 239000002028 Biomass Substances 0.000 description 1
- 206010063385 Intellectualisation Diseases 0.000 description 1
- 206010042434 Sudden death Diseases 0.000 description 1
- 230000001154 acute effect Effects 0.000 description 1
- 230000002411 adverse Effects 0.000 description 1
- 238000004458 analytical method Methods 0.000 description 1
- 230000008485 antagonism Effects 0.000 description 1
- 230000008859 change Effects 0.000 description 1
- 238000007405 data analysis Methods 0.000 description 1
- 230000007812 deficiency Effects 0.000 description 1
- 238000001514 detection method Methods 0.000 description 1
- 230000008034 disappearance Effects 0.000 description 1
- 239000003814 drug Substances 0.000 description 1
- 230000005611 electricity Effects 0.000 description 1
- 238000004880 explosion Methods 0.000 description 1
- 230000036541 health Effects 0.000 description 1
- 208000019622 heart disease Diseases 0.000 description 1
- 230000010354 integration Effects 0.000 description 1
- 230000006855 networking Effects 0.000 description 1
- 238000005457 optimization Methods 0.000 description 1
- 230000000306 recurrent effect Effects 0.000 description 1
- 230000006403 short-term memory Effects 0.000 description 1
- 239000000126 substance Substances 0.000 description 1
- 230000007704 transition Effects 0.000 description 1
Classifications
-
- Y—GENERAL TAGGING OF NEW TECHNOLOGICAL DEVELOPMENTS; GENERAL TAGGING OF CROSS-SECTIONAL TECHNOLOGIES SPANNING OVER SEVERAL SECTIONS OF THE IPC; TECHNICAL SUBJECTS COVERED BY FORMER USPC CROSS-REFERENCE ART COLLECTIONS [XRACs] AND DIGESTS
- Y02—TECHNOLOGIES OR APPLICATIONS FOR MITIGATION OR ADAPTATION AGAINST CLIMATE CHANGE
- Y02A—TECHNOLOGIES FOR ADAPTATION TO CLIMATE CHANGE
- Y02A90/00—Technologies having an indirect contribution to adaptation to climate change
- Y02A90/10—Information and communication technologies [ICT] supporting adaptation to climate change, e.g. for weather forecasting or climate simulation
Landscapes
- Measurement And Recording Of Electrical Phenomena And Electrical Characteristics Of The Living Body (AREA)
Abstract
本发明公开了一种基于深度神经网络的两步骤心律失常分类方法。属于计算机、软件类领域;具体步骤:1、预处理;2、初始化;3、Bi‑LSTM分类N,SV,F和Q;4、GAN生成不平衡心跳数据;5、Bi‑LSTM分类S和V;本发明适用于收集到的ECG心跳数据不足的心律失常的分类;可以保证较高的分类准确率;此外采用的两步骤分类框架充分考虑了医学事实,因此本发明可以保证较高的分类准确率,具有实际意义。
Description
技术领域
本发明涉及计算机、软件类领域,具体涉及一种基于深度神经网络的两步骤心律失常分类方法。
背景技术
传感器是把自然界的非电量变成可利用的电信号的装置或者器件。随着时代的变迁和技术的进步,它的定义也被不断更新。现阶段我们认为,传感器是把自然界的物理量、化学量、生物量变成可利用的信号的装置或者器件。它的特点包括:微型化、数字化、智能化、多功能化、系统化、网络化。传感器是物联网和互联网的关键硬件,可以用来收集实时数据,给人们带来了极大的便利,使得人们可以在第一时间获取相关的数据信息,并对其进行分析。传感器是智能化设备的信息入口,是大数据的重要组成部分。在我们的生产生活中,它是极为重要的信息和数据来源,是信息化和工业化深度融合的关键技术之一。传感器市场规模正呈现高速的增长态势,处于突破和创新发展的状态。
如今,传感器已经渗透到我们生活和生产的方方面面。未来,随着感知技术的不断发展和需求的不断增多,智能传感器将越来越多地应用到不同的领域,满足不同细节的需求。现阶段,传感器在医学方面应用广泛。其中较为普遍的是可穿戴的传感器,可以用于收集人们的生理数据,特别是医学上心电信号的收集。全世界每年会有数百万人死于心律失常。而心脏疾病常常具有突发性及不确定性,急性发作甚至会导致猝死。因此,诊断时间越短,越快出诊断结果,就越有利于患者的健康;细小的差错也可能将会导致非常严重的后果,有时甚至会威胁到患者的生命。人工进行数据分析耗时耗力,得出的结果往往具有主观性,易遗漏重要的信息,诊断低效且准确率不高。而利用分析收集到的实时心电数据,借助计算机来实现心律失常的自动检测,可以尽快判断出这个人的心脏有没有出现心律失常的问题以及出现的是什么类型的心律失常。它具有诊断高效且准确性高等优点。
发明内容
针对上述问题,本发明提供了一种基于深度神经网络的两步骤心律失常分类方法;本发明利用生成对抗网络(Generative Adversarial Nets,简称GAN)和堆叠的双向长短时记忆循环神经网络(Bi-directional Long Short-Term Memory,简称Bi-LSTM)建立了一个基于深度神经网络的两步骤心跳分类框架;按照标准,将心跳分为这五大类:正常心跳(normal beat,简称N)、室上异位心跳(supraventricular ectopic beat,简称S)、心室异位心跳(ventricular beat,简称V)、融合心跳(fusion beat,简称F)以及不可分心跳(unclassifiable beat,简称Q);该框架的第一步先将S类和V类作为一大类SV类,将心跳分成四个类别,第二步再对S类和V类进行划分,从而完成心跳的分类。
本发明的技术方案是:基于深度神经网络的两步骤心律失常分类方法,具体操作步骤如下:
步骤(1.1)、对收集到的ECG信号进行预处理,定位QRS波收集N、SV、F和Q类心跳对应的前一个RR间期并截取出ECG信号中的心拍;从而确定心拍的类型及各类型心拍对应的数据量;
步骤(1.2)、将收集到的所有的RR间期与截取到的ECG信号中的心拍分别按类别进行编码,得到所对应的类别标签;
步骤(1.3)、再将收集到的所有的RR间期与对应的标签按相同的顺序打乱,划分出训练集与测试集,对RR间期的Bi-LSTM分类模型进行训练,得到训练好的RR间期Bi-LSTM分类模型;
步骤(1.4)、对于数据量在一定区间的心拍类型,使用生成对抗网络GAN生成不平衡心拍数据,即使用GAN生成该类型的心拍,并将生成的心拍按照类别编码,得到对应的类别标签;
步骤(1.5)、将得到的所有的心拍数据与对应的标签按相同的顺序打乱,并划分训练集与测试集,对心拍的Bi-LSTM分类模型进行训练,得到训练好的心拍的Bi-LSTM分类模型;
步骤(1.6)、利用训练好的基于深度神经网络的两步骤心律失常分类模型对未知类别的ECG心拍进行分类。
进一步的,在步骤(1.1)中,所述对收集到的ECG信号进行预处理的具体操作方法如下:
(1.1.1)、收集N、SV、F和Q类心拍对应的前一个RR间期,将所有心拍对应的RR间期序列构成的集合表示为S,所述S={Si,i=1,2,3,…n},Si表示一个心跳所对应的RR间期序列,即Si=[Rt-NeRan,…,Rt-1,Rt,Rt+1,…,Rt+NeRan];
式中,Rt表示第t-1个心跳到第t个心跳的RR间隔,NeRan表示心跳的邻域范围;
(1.1.2)、定位QRS波截取SV类中S类和V类的心拍,将收集到的原始ECG信号时间序列分割成m段子序列,所述m段子序列构成的集合为X,则X={Xi,i=1,2,3,…,m};其中,Xi表示构成ECG信号的基本单位:心跳。
进一步的,在所述步骤(1.3)中,对RR间期的Bi-LSTM分类模型进行训练的具体操作方法如下:采用十折交叉验证的方法对Bi-LSTM模型进行训练,即将全部的所有的心拍数据分成十个相等的部分,使用其中的任意一个部分用于测试,剩余的九个部分用于训练,通过移动选择测试部分重复十次训练操作。
进一步的,在所述步骤(1.4)中,使用生成对抗网络GAN生成不平衡心拍数据的具体操作方法如下:其中,生成对抗网络GANGAN由生成器和判别器两个部分组成;其具体操作过程如下:
(1.4.1)、将需要扩充数据的类型的心拍Xi={x1,x2,x3,…,xt}作为GAN的训练集输入到GAN中,GAN先固定生成器并从训练集中随机采样心拍作为判别器的输入,使判别器获得真实数据的概率分布;当输入为生成的数据时,判别器通过判断其真实性;当判别器判定数据为真实的数据,则判别器就会输出1;当判别器判定是生成的数据,则判别器就会输出0;
(1.4.2)、GAN中生成器的输入为一组噪声信号Zi={z1,z2,z3,…,zt},该噪声信号满足高斯分布,生成器会将这一组噪声尽量输出为与真实的ECG信号具有相同概率分布的数据;生成器的任务就是使生成的数据能够“骗过”判别器,使判别器以为生成的数据就是原始的样本;生成器和判别器交替迭代训练,训练完成后整个GAN输出的结果为与真实的样本的概率分布相同的生成ECG信号X′i={x′1,x′2,x′3,…,x′t}。
进一步的,在所述步骤(1.5)中,采用十折交叉验证的方法对Bi-LSTM模型进行训练,即,将全部的ECG心拍序列分成十个相等的部分,使用其中的任意一个部分用于测试,剩余的九个部分用于训练,通过移动选择测试部分重复十次训练操作。
进一步;的在步骤(1.1)与(1.2)中,所述的所有的RR间期的个数在35-50个之间。
为了使计算机能够进行自动、高效、准确地诊断心律失常的类型,在现有的心律失常自动分类中还有一些问题亟待解决;本发明旨在解决以下问题:
1、分类模型不具备普遍性:在对心电图(Electrocardiogram,简称ECG)信号进行分类时,使用的分类模型只适用于特定心电数据集中的数据,而无法区分来自其他心电数据集的心跳;
2、无法准确诊断出数据量少的心律类别:存在不常见的心律类型,该类型所对应的数据非常少,计算机得不到足够的数据进行训练。在分析ECG信号的片段时,如果遇到这种不常见的心律类型,计算机可能会无法准确判断出该ECG片段所属的类别。
3、准确率不高:目前大多数利用计算机进行心跳分类所取得的分类准确率都不够高;如果不能实现非常准确的分类就不能对医生的诊断起到真正辅助作用。
本发明的有益效果是:本发明考虑医学事实,具有实际意义;1、本发明提出了一个基于深度神经网络的两步骤心跳分类框架;首先根据N类心跳的前一个RR间期比S类和V类的短的事实,将S类和V类心跳作为一个大类SV类,通过RR间期来区分N类和SV类;然后根据S类心跳与V类心跳ECG信号的波形有差异的事实将S类和V类心跳区分开;由于F类心跳和Q类心跳的区分没有实际意义,所以不考虑这两类心跳的分类效果;因此本发明具有实际意义;2、本发明适用于数据不平衡条件下的心律失常分类:本发明适用于收集到的ECG心跳数据不足的心律失常的分类;针对收集到的数据不足的情况,GAN可以生成和输入数据具有相同概率分布的数据,我们利用GAN对数据量少的类别的心跳进行数据的扩充;GAN生成足够数量对应的心跳数据后,就可以解决因为数据量不足所带来的问题,则不存在因为数据缺乏所导致的分类准确率低的情形;GAN生成的数据可以使原本不平衡的数据变得平衡,则分类模型能够取得足够大的训练集,从而有利于模型参数的最优化;因此本发明可以实现数据不平衡条件下的心律失常分类;3、本发明可以保证较高的分类准确率:本发明一方面利用GAN进行数据的扩充,来获取足够量的ECG数据用于训练,另一方面又使用了Bi-LSTM构建分类模型;Bi-LSTM是一种特殊的循环神经网络(Recurrent Neural Network,简称RNN),它可以全面地学习到输入数据的之间的前后关系,并避免梯度消失和梯度爆炸问题的发生;在实现分类之前,需要先对基于Bi-LSTM的分类模型进行训练,训练完成后该分类模型取得的参数可以准确提取出输入数据前后时序之间的关系;此外采用的两步骤分类框架充分考虑了医学事实,因此本发明可以保证较高的分类准确率。
附图说明
图1是本发明的结构流程图;
图2是本发明中RR间期分类模型的结构示意图;
图3是本发明中GAN中生成器和判别器的结构示意图;
图4是本发明中心跳分类模型的结构示意图。
具体实施方式
为了更清楚地说明本发明的技术方案,下面结合附图对本发明的技术方案做进一步的详细说明:
本发明叙述了整个分类心律失常运行的过程,本发明的两步骤心跳分类框架如图1所示。
步骤1、预处理:
(1.1)、对收集到的ECG信号进行预处理,定位QRS波收集N、SV、F和Q类心跳对应的前一个RR间期;使用Pan-Tompkins算法完成QRS波的定位后,计算当前R峰与前一个R峰之间的时间间隔;将所有心跳对应的RR间期序列构成的集合表示为S,且S={Si,i=1,2,3,…n};Si是一个心跳所对应的RR间期序列,Si=[Rt-NeRan,…,Rt-1,Rt,Rt+1,…,Rt+NeRan];其中Rt表示第t-1个心跳到第t个心跳的RR间隔,NeRan定义为心跳的邻域范围,本发明中NeRan的值通常设定在设置为25左右;每一个RR间期序列Si的类型与中央第t个心跳的类型一致;此外,为了提高分类的准确率,每个RR间期序列又添加了三个特征:最大的RR间期、最小的RR间期以及两者之间的差值;本发明中使用到的ECG数据是MIT-BIH心律失常数据集的MLⅡ导联;最后收集到的N、SV、F、Q这四种类型的RR间期所对应的个数分别为:85107、11394、795和15;
(1.2)、定位QRS波截取ECG信号中SV类的S类和V类的心跳;将一段收集到的原始ECG信号时间序列分割成m段子序列,这m个子序列构成的集合为X,则X={Xi,i=1,2,3,…,m};其中,Xi是构成心电信号的基本单位——心跳;使用Pan-Tompkins算法完成QRS波的定位后,取一个大小为170的窗口完成心跳的截取;窗口的尺寸依据经验值设定,为一个心跳的大小;截取时在定位到的每一个R峰的左侧取70个采样点,在R峰的右侧取100个采样点,截取的每个心跳表示为Xi={x1,x2,x3,…,xt};本发明中使用到的ECG数据来自于MIT-BIH心律失常数据集的MLⅡ导联;最后截取到的S、V这两种类型的心跳所对应的个数分别为:2703和6896;
步骤2、初始化:
(2.1)、将收集到的所有的RR间期按照类别进行编码;N、SV、F和Q这四类RR间期编码的类别标签分别为:0,1,2,3;
(2.2)、将截取到的ECG心跳按照类别进行编码;其中S和V这两类心跳进行编码后的类别标签分别为:0和1;从而得到所对应的类别标签;
步骤3、Bi-LSTM分类N,SV,F和Q:
(3.1)、按相同的顺序打乱RR间期数据集与对应标签;为了使基于Bi-LSTM的分类模型能够更全面地学习到输入的RR间期序列之间的关系,所有的数据在混合后需要打乱顺序,并且按相同的顺序打乱对应的标签,从而避免数据分布不均匀的情况发生;
(3.2)、划分RR间期数据集;将90%的数据划分为训练集,剩余10%的数据用作测试集,并将训练集中的20%用作验证集,此外还采用十折交叉验证的方法将全部的RR间期数据集分成十个相等的部分,一个部分用于测试,剩余的九个部分用于训练,通过移动测试部分重复这个操作10次;
(3.3)、将RR间期训练集的数据输入到搭建好的Bi-LSTM的模型中进行训练,获取模型最优参数;本发明提出的RR间期分类模型的整体结构如图2所示,第一层和第二层LSTM神经元的个数都为54,第三层、第四层和第五层Dense层对应的神经元的个数分别为:64、32、4,每一个防止过拟合的dropout层设置的dropout rate为0.3;每一次训练设置的轮数为100轮,训练时使用的优化器为Adam,损失函数是sparse categorical cross entropy;最后获得的评价指标的值是十次训练取得的评价指标的平均值;
(3.4)、训练完成后保留模型参数,输入RR间期测试集进行测试,此时的模型可以输入未知类别的RR间期数据实现N、SV、F和Q的自动分类;
步骤4、GAN生成不平衡心跳数据:
(4.1)、使用GAN生成数据量在一定区间(具体是指该类型的心跳的数量明显少于其余类型的心跳数量)内的类别的心跳;在此阶段,GAN会扩充数据量不足的类型的心跳对应的数据,以消除因为数据缺乏而对分类结果造成的不良影响;GAN由生成器和判别器两个部分组成,它的框架结构如图3所示;其中,生成器的全连接层神经元的个数为50*128,经过Reshape层后数据的维度为50*128,四个卷积层卷积核的个数分别为128、64、32和1,卷积核的尺寸都为6,步长都为2,批标准化层的批处理大小设置为0.8;判别器中卷积层卷积核的个数为32,卷积核的尺寸为6,步长为3,批标准化层的批处理大小也设置为0.8,最大池化层池化的尺寸为2,最后一个全连接层中神经元的个数为1;且判别器的学习率设为0.0002,生成器的学习率设为0.0001,一共训练10000轮;从步骤1中提取到的S和V的心跳个数可以发现,S和V这两类心跳所对应的个数相差较大,而F和Q这两类心跳没有实际的意义,类似于噪声,所以只用GAN生成3000条S类数据而不考虑F和Q;GAN具体生成不平衡传感器数据的过程如下:
第一步:将需要扩充数据的类型的心跳Xi={x1,x2,x3,…,xt}作为GAN的训练集输入到GAN中,GAN会先固定生成器并从训练集中随机采样心跳作为判别器的输入,使判别器获得真实数据的概率分布,那么当输入为生成的数据时,判别器就能判断出它的真实性;当判别器判定数据是真实的数据时,判别器就会输出1,当判别器判定是生成的数据时,判别器就会输出0;
第二步:GAN中生成器的输入为一组噪声信号Zi={z1,z2,z3,…,zt},该噪声信号满足高斯分布,生成器会将这一组噪声尽量输出为与真实的ECG信号具有相同概率分布的数据;生成器的任务就是使生成的数据能够“骗过”判别器,使判别器以为生成的数据就是原始的样本;生成器和判别器交替迭代训练,训练完成后整个GAN输出的结果为与真实的样本的概率分布相同的生成ECG信号X′i={x′1,x′2,x′3,…,x′t};
(4.2)、将生成的ECG心跳按照类别进行编码:将生成的S类心跳编码为0;
步骤5、Bi-LSTM分类S和V:
(5.1)、将GAN生成的心跳与截取到的原始心跳按照类别混合;将生成的X′i按照类别与对应的那一类的原始样本数据混合,则不存在数据不平衡的问题,每一类心律的数据都足够多;
(5.2)、按相同的顺序打乱心跳数据集与对应标签;为了使基于Bi-LSTM的分类模型能够更全面地学习到输入的心跳的时间序列之间的关系,与步骤3相同,所有的数据在混合后需要打乱顺序,并且按相同的顺序打乱对应的标签以避免数据中各个类型的心跳分布不均匀的情况发生;
(5.3)、划分数据集:将90%的数据划分为训练集,剩余10%的数据用作测试集,并将训练集中的20%用作验证集,此外还采用十折交叉验证的方法;即将全部的ECG心跳序列分成十个相等的部分,一个部分用于测试,剩余的九个部分用于训练,我们通过移动测试部分重复这个操作10次;
(5.4)、将训练集的数据输入到搭建好的基于Bi-LSTM分类模型中进行训练,从而获取模型的最优参数;提出的分类模型的整体结构如图4所示,第一层和第二层LSTM神经元的个数都为170,第三层、第四层和第五层Dense层对应的神经元的个数分别为:64、32、2,每一个防止过拟合的dropout层设置的dropout rate为0.3;每一次训练设置的轮数为50轮,训练时使用的优化器为Adam,损失函数是sparse categorical cross entropy;最后获得的评价指标的值是十次训练取得的评价指标的平均值;
(5.5)、模型训练好后将最后取得的最优参数保留,输入测试集进行测试,此时该模型可以实现对S和V的自动分类。
具体实例:
给定一段传感器收集到的ECG信号,先使用Pan-Tompkins算法进行QRS波的定位来收集所有心跳对应的RR间期以及心拍数据;
接着将待分类心跳的RR间期序列输入RR间期的Bi-LSTM分类模型进行分类。如果输出的类别是N、F或者Q,则该心跳完成分类;如果输出的是SV,则需要进行下一步以做进一步的划分;
针对上述输出为SV的心跳,将心拍数据输入心拍的Bi-LSTM分类模型,从而将S与V类心跳区分开。
最后,应当理解的是,本发明中所述实施例仅用以说明本发明实施例的原则;其他的变形也可能属于本发明的范围;因此,作为示例而非限制,本发明实施例的替代配置可视为与本发明的教导一致;相应地,本发明的实施例不限于本发明明确介绍和描述的实施例。
Claims (6)
1.基于深度神经网络的两步骤心律失常分类方法,其特征在于:具体操作步骤如下:
步骤(1.1)、对收集到的ECG信号进行预处理,定位QRS波收集N、SV、F和Q类心跳对应的前一个RR间期并截取出ECG信号中的心跳;从而确定心跳的类型及各类型心跳对应的数据量;
其中,N表示正常心跳,S表示室上异位心跳,V表示心室异位心跳,F表示融合心跳,Q表示不可分心跳,将S类和V类作为一大类SV类;
步骤(1.2)、将收集到的所有的RR间期与截取到的ECG信号中的心跳分别按类别进行编码,得到所对应的类别标签;
步骤(1.3)、再将收集到的所有的RR间期与对应的标签按相同的顺序打乱,划分出训练集与测试集,对RR间期的Bi-LSTM分类模型进行训练,得到训练好的RR间期Bi-LSTM分类模型;
步骤(1.4)、对于数据量在一定区间的心跳类型,使用生成对抗网络GAN生成不平衡心跳数据,即使用GAN生成该类型的心跳,并将生成的心跳按照类别编码,得到对应的类别标签;
步骤(1.5)、将得到的所有的心跳数据与对应的标签按相同的顺序打乱,并划分训练集与测试集,对心跳的Bi-LSTM分类模型进行训练,得到训练好的心跳的Bi-LSTM分类模型;
步骤(1.6)、利用训练好的基于深度神经网络的两步骤心律失常分类模型对未知类别的ECG心跳进行分类;
所述两步骤心律失常分类模型即:利用生成对抗网络和堆叠的双向长短时记忆循环神经网络建立了一个基于深度神经网络的两步骤心跳分类框架;按照标准,将心跳分为五大类:正常心跳、室上异位心跳、心室异位心跳、融合心跳及不可分心跳;该框架的第一步先将室上异位心跳类和心室异位心跳类作为一大类,将心跳分成四个类别,第二步再对室上异位心跳类和心室异位心跳类进行划分,从而完成心跳的分类。
2.根据权利要求1所述的基于深度神经网络的两步骤心律失常分类方法,其特征在于:在步骤(1.1)中,所述对收集到的ECG信号进行预处理的具体操作方法如下:
(1.1.1)、收集N、SV、F和Q类心跳对应的前一个RR间期,将所有心跳对应的RR间期序列构成的集合表示为S,所述S={Si,i=1,2,3,...,n},Si表示一个心跳所对应的RR间期序列,即Si=[Rt-NeRan,…,Rt-1,Rt,Rt+1,…,Rt+NeRan];
式中,Rt表示第t-1个心跳到第t个心跳的RR间隔,NeRan表示心跳的邻域范围;
(1.1.2)、定位QRS波截取SV类中S类和V类的心跳,将收集到的原始ECG信号时间序列分割成m段子序列,所述m段子序列构成的集合为X,则X={Xi,i=1,2,3,…,m};其中,Xi表示构成ECG信号的基本单位:心跳。
3.根据权利要求1所述的基于深度神经网络的两步骤心律失常分类方法,其特征在于:在所述步骤(1.3)中,对RR间期的Bi-LSTM分类模型进行训练的具体操作方法如下:采用十折交叉验证的方法对Bi-LSTM模型进行训练,即将全部的所有的心跳数据分成十个相等的部分,使用其中的任意一个部分用于测试,剩余的九个部分用于训练,通过移动选择测试部分重复十次训练操作。
4.根据权利要求1所述的基于深度神经网络的两步骤心律失常分类方法,其特征在于:在所述步骤(1.4)中,使用生成对抗网络GAN生成不平衡心跳数据的具体操作方法如下:其中,生成对抗网络GAN由生成器和判别器两个部分组成;其具体操作过程如下:
(1.4.1)、将需要扩充数据的类型的心跳Xi={x1,x2,x3,…,xt}作为GAN的训练集输入到GAN中,GAN先固定生成器并从训练集中随机采样心跳作为判别器的输入,使判别器获得真实数据的概率分布;当输入为生成的数据时,判别器就判断其真实性;当判别器判定数据为真实的数据,则判别器就会输出1;当判别器判定是生成的数据,则判别器就会输出0;
(1.4.2)、GAN中生成器的输入为一组噪声信号Zi={z1,z2,z3,…,zt},该噪声信号满足高斯分布,生成器会将这一组噪声尽量输出为与真实的ECG信号具有相同概率分布的数据;生成器的任务就是使生成的数据能够“骗过”判别器,使判别器以为生成的数据就是原始的样本;生成器和判别器交替迭代训练,训练完成后整个GAN输出的结果为与真实的样本的概率分布相同的生成ECG信号X′i={x′1,x'2,x'3,…,x′t}。
5.根据权利要求1所述的基于深度神经网络的两步骤心律失常分类方法,其特征在于:在所述步骤(1.5)中,采用十折交叉验证的方法对Bi-LSTM模型进行训练,即,将全部的ECG心跳序列分成十个相等的部分,使用其中的任意一个部分用于测试,剩余的九个部分用于训练,通过移动选择测试部分重复十次训练操作。
6.根据权利要求1所述的基于深度神经网络的两步骤心律失常分类方法,其特征在于:在步骤(1.2)中,所述的所有的RR间期的个数在35-50个之间。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202011319283.8A CN112686091B (zh) | 2020-11-23 | 2020-11-23 | 基于深度神经网络的两步骤心律失常分类方法 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202011319283.8A CN112686091B (zh) | 2020-11-23 | 2020-11-23 | 基于深度神经网络的两步骤心律失常分类方法 |
Publications (2)
Publication Number | Publication Date |
---|---|
CN112686091A CN112686091A (zh) | 2021-04-20 |
CN112686091B true CN112686091B (zh) | 2024-02-23 |
Family
ID=75446750
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202011319283.8A Active CN112686091B (zh) | 2020-11-23 | 2020-11-23 | 基于深度神经网络的两步骤心律失常分类方法 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN112686091B (zh) |
Families Citing this family (1)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
KR102589471B1 (ko) | 2021-12-15 | 2023-10-16 | 한양대학교 에리카산학협력단 | 데이터 증강 장치 및 방법 |
Citations (5)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN107822622A (zh) * | 2017-09-22 | 2018-03-23 | 成都比特律动科技有限责任公司 | 基于深度卷积神经网络的心电图诊断方法和系统 |
CN110384482A (zh) * | 2019-06-26 | 2019-10-29 | 广州视源电子科技股份有限公司 | 心电信号分类方法、装置、计算机设备和存储介质 |
KR20190141326A (ko) * | 2018-06-14 | 2019-12-24 | 한국과학기술원 | 심층 컨볼루션 신경망을 이용한 심전도 부정맥 분류 방법 및 장치 |
CN110897628A (zh) * | 2018-09-14 | 2020-03-24 | 杭州脉流科技有限公司 | 基于深度神经网络的心电图信号特征提取方法、装置、系统、设备和分类方法 |
CN111460956A (zh) * | 2020-03-26 | 2020-07-28 | 山东科技大学 | 一种基于数据增强与损失加权的不平衡心电样本分类方法 |
Family Cites Families (1)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
US10383539B2 (en) * | 2017-06-22 | 2019-08-20 | Smart Solutions Technologies, S.L. | System and methods for classifying arrhythmia-related heartbeats |
-
2020
- 2020-11-23 CN CN202011319283.8A patent/CN112686091B/zh active Active
Patent Citations (5)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN107822622A (zh) * | 2017-09-22 | 2018-03-23 | 成都比特律动科技有限责任公司 | 基于深度卷积神经网络的心电图诊断方法和系统 |
KR20190141326A (ko) * | 2018-06-14 | 2019-12-24 | 한국과학기술원 | 심층 컨볼루션 신경망을 이용한 심전도 부정맥 분류 방법 및 장치 |
CN110897628A (zh) * | 2018-09-14 | 2020-03-24 | 杭州脉流科技有限公司 | 基于深度神经网络的心电图信号特征提取方法、装置、系统、设备和分类方法 |
CN110384482A (zh) * | 2019-06-26 | 2019-10-29 | 广州视源电子科技股份有限公司 | 心电信号分类方法、装置、计算机设备和存储介质 |
CN111460956A (zh) * | 2020-03-26 | 2020-07-28 | 山东科技大学 | 一种基于数据增强与损失加权的不平衡心电样本分类方法 |
Non-Patent Citations (6)
Title |
---|
An advanced two-step DNN-based framework for arrhythmia detection;Jinyuan He等;《Advances in Knowledge Discovery and Data Mining》;第422-434页 * |
Automated detection of artial fibrillation ECG signals using two stage VMD and atrial fibrillation diagnosis inddex;R.K.TRIPATHY等;《Journal of Mechanics in Medicine and Biology》;第17卷(第07期);第1740044页 * |
BeatClass:A sustainable ECG classification system in IoT-based eHealth;Le Sun等;《IEEE Internet of Things Journal》;第9卷(第10期);第7178-7195页 * |
基于卷积长短时记忆网络的心律失常分类方法;柯丽等;《电子与信息学报》;第42卷(第08期);第1990-1998页 * |
基于深度学习的心律失常分类研究与实现;张敬启;《中国优秀硕士学位论文全文数据库 医药卫生科技辑》;第2020卷(第08期);第E062-172页 * |
类不平衡条件下的心电图心跳分类算法的研究;王逸琳;《中国优秀硕士学位论文全文数据库 医药卫生科技辑》;第2023卷(第01期);第E060-14页 * |
Also Published As
Publication number | Publication date |
---|---|
CN112686091A (zh) | 2021-04-20 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
US20080103403A1 (en) | Method and System for Diagnosis of Cardiac Diseases Utilizing Neural Networks | |
Karthiga et al. | Early prediction of heart disease using decision tree algorithm | |
CN110021397A (zh) | 基于人体生理参数预测给药量的方法及存储介质 | |
Mahmud et al. | An improved framework for reliable cardiovascular disease prediction using hybrid ensemble learning | |
Gnaneswar et al. | A review on prediction and diagnosis of heart failure | |
Bishnoi et al. | Artificial intelligence techniques used in medical sciences: a review | |
Shen et al. | Risk prediction for cardiovascular disease using ECG data in the China Kadoorie Biobank | |
CN112686091B (zh) | 基于深度神经网络的两步骤心律失常分类方法 | |
Zhang et al. | Automatic diagnosis of arrhythmia with electrocardiogram using multiple instance learning: From rhythm annotation to heartbeat prediction | |
Tallapragada et al. | Improved atrial fibrillation detection using cnn-lstm | |
Kavitha et al. | Review on cardio vascular disease prediction using machine learning | |
Umamaheswari et al. | Prediction of myocardial infarction using K-medoid clustering algorithm | |
Usha et al. | Predicting Heart Disease Using Feature Selection Techniques Based on Data Driven Approach | |
Srivastava et al. | ECG Pattern Analysis using Artificial Neural Network | |
Bahrami et al. | Prediction of diabetic retinopathy based on a committee of random forests | |
Rajesh et al. | A Silent Cardiac Atrial Fibrillation Detection and Classification using Deep Learning Approach | |
Swetha et al. | Machine Learning and Deep Learning for Medical Analysis—A Case Study on Heart Disease Data | |
Maleki | Identification of cardiovascular diseases through ECG classification using wavelet transformation | |
Soundari et al. | ANND: Identification and Prediction of Tooth Decay based on Artificial Neural Network and DenseNet Model | |
Jadhav et al. | Monitoring and Predicting of Heart Diseases Using Machine Learning Techniques | |
Arowolo et al. | A Comparative of Classification Models for Predicting of Heart Diseases | |
Vu et al. | A Comparison of Machine Learning Algorithms for Parkinson’s Disease Detection | |
Ananthajothi et al. | Class Based Ensemble Similarity Measure Based Efficient Classification of Medical Data | |
Asgharnezhad et al. | Improving PPG Signal Classification with Machine Learning: The Power of a Second Opinion | |
Munandar | Data Mining for Heart Disease Prediction Based on Echocardiogram and Electrocardiogram Data |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
SE01 | Entry into force of request for substantive examination | ||
GR01 | Patent grant | ||
GR01 | Patent grant |