CN112733774A - 一种基于BiLSTM与串并多尺度CNN结合的轻量化ECG分类方法 - Google Patents
一种基于BiLSTM与串并多尺度CNN结合的轻量化ECG分类方法 Download PDFInfo
- Publication number
- CN112733774A CN112733774A CN202110063840.2A CN202110063840A CN112733774A CN 112733774 A CN112733774 A CN 112733774A CN 202110063840 A CN202110063840 A CN 202110063840A CN 112733774 A CN112733774 A CN 112733774A
- Authority
- CN
- China
- Prior art keywords
- serial
- ecg
- scale
- classification
- extraction module
- 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.)
- Pending
Links
- 238000000034 method Methods 0.000 title claims abstract description 48
- 238000000605 extraction Methods 0.000 claims abstract description 115
- 230000010349 pulsation Effects 0.000 claims abstract description 36
- 238000012360 testing method Methods 0.000 claims abstract description 35
- 238000012549 training Methods 0.000 claims abstract description 31
- 230000004927 fusion Effects 0.000 claims abstract description 16
- 206010047289 Ventricular extrasystoles Diseases 0.000 claims abstract description 13
- 206010042602 Supraventricular extrasystoles Diseases 0.000 claims abstract description 11
- 238000011176 pooling Methods 0.000 claims description 18
- 230000008569 process Effects 0.000 claims description 11
- 238000013145 classification model Methods 0.000 claims description 9
- 238000007781 pre-processing Methods 0.000 claims description 6
- 238000002790 cross-validation Methods 0.000 claims description 5
- 230000008901 benefit Effects 0.000 claims description 4
- 238000010276 construction Methods 0.000 claims description 3
- 208000009729 Ventricular Premature Complexes Diseases 0.000 claims description 2
- 238000013461 design Methods 0.000 abstract description 5
- 238000013527 convolutional neural network Methods 0.000 description 59
- 238000010586 diagram Methods 0.000 description 14
- 238000005516 engineering process Methods 0.000 description 3
- 230000006870 function Effects 0.000 description 3
- 238000011160 research Methods 0.000 description 3
- 238000004422 calculation algorithm Methods 0.000 description 2
- 238000013135 deep learning Methods 0.000 description 2
- 230000000694 effects Effects 0.000 description 2
- 238000002474 experimental method Methods 0.000 description 2
- 208000019622 heart disease Diseases 0.000 description 2
- 230000006872 improvement Effects 0.000 description 2
- 239000011159 matrix material Substances 0.000 description 2
- 208000024172 Cardiovascular disease Diseases 0.000 description 1
- 230000002159 abnormal effect Effects 0.000 description 1
- 238000004458 analytical method Methods 0.000 description 1
- 238000013528 artificial neural network Methods 0.000 description 1
- 230000009286 beneficial effect Effects 0.000 description 1
- 230000002457 bidirectional effect Effects 0.000 description 1
- 238000007635 classification algorithm Methods 0.000 description 1
- 230000003247 decreasing effect Effects 0.000 description 1
- 230000007547 defect Effects 0.000 description 1
- 238000001514 detection method Methods 0.000 description 1
- 238000003745 diagnosis Methods 0.000 description 1
- 201000010099 disease Diseases 0.000 description 1
- 208000037265 diseases, disorders, signs and symptoms Diseases 0.000 description 1
- 239000000284 extract Substances 0.000 description 1
- 230000004217 heart function Effects 0.000 description 1
- 230000001766 physiological effect Effects 0.000 description 1
- 238000012545 processing Methods 0.000 description 1
- 230000000306 recurrent effect Effects 0.000 description 1
- 230000035945 sensitivity Effects 0.000 description 1
- 238000012706 support-vector machine Methods 0.000 description 1
- 230000009466 transformation Effects 0.000 description 1
- 238000012795 verification Methods 0.000 description 1
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F2218/00—Aspects of pattern recognition specially adapted for signal processing
- G06F2218/08—Feature extraction
-
- A—HUMAN NECESSITIES
- A61—MEDICAL OR VETERINARY SCIENCE; HYGIENE
- A61B—DIAGNOSIS; SURGERY; IDENTIFICATION
- A61B5/00—Measuring for diagnostic purposes; Identification of persons
- A61B5/72—Signal processing specially adapted for physiological signals or for diagnostic purposes
- A61B5/7203—Signal processing specially adapted for physiological signals or for diagnostic purposes for noise prevention, reduction or removal
-
- A—HUMAN NECESSITIES
- A61—MEDICAL OR VETERINARY SCIENCE; HYGIENE
- A61B—DIAGNOSIS; SURGERY; IDENTIFICATION
- A61B5/00—Measuring for diagnostic purposes; Identification of persons
- A61B5/72—Signal processing specially adapted for physiological signals or for diagnostic purposes
- A61B5/7225—Details of analog processing, e.g. isolation amplifier, gain or sensitivity adjustment, filtering, baseline or drift compensation
-
- A—HUMAN NECESSITIES
- A61—MEDICAL OR VETERINARY SCIENCE; HYGIENE
- A61B—DIAGNOSIS; SURGERY; IDENTIFICATION
- A61B5/00—Measuring for diagnostic purposes; Identification of persons
- A61B5/72—Signal processing specially adapted for physiological signals or for diagnostic purposes
- A61B5/7235—Details of waveform analysis
- A61B5/7264—Classification of physiological signals or data, e.g. using neural networks, statistical classifiers, expert systems or fuzzy systems
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/045—Combinations of networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/049—Temporal neural networks, e.g. delay elements, oscillating neurons or pulsed inputs
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/08—Learning methods
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F2218/00—Aspects of pattern recognition specially adapted for signal processing
- G06F2218/12—Classification; Matching
Landscapes
- Engineering & Computer Science (AREA)
- Health & Medical Sciences (AREA)
- Life Sciences & Earth Sciences (AREA)
- Physics & Mathematics (AREA)
- Theoretical Computer Science (AREA)
- Artificial Intelligence (AREA)
- Molecular Biology (AREA)
- Signal Processing (AREA)
- Biomedical Technology (AREA)
- Biophysics (AREA)
- General Health & Medical Sciences (AREA)
- Computer Vision & Pattern Recognition (AREA)
- General Engineering & Computer Science (AREA)
- General Physics & Mathematics (AREA)
- Mathematical Physics (AREA)
- Evolutionary Computation (AREA)
- Software Systems (AREA)
- Surgery (AREA)
- Data Mining & Analysis (AREA)
- Computational Linguistics (AREA)
- Veterinary Medicine (AREA)
- Physiology (AREA)
- Psychiatry (AREA)
- Pathology (AREA)
- Heart & Thoracic Surgery (AREA)
- Medical Informatics (AREA)
- Computing Systems (AREA)
- Animal Behavior & Ethology (AREA)
- Public Health (AREA)
- Fuzzy Systems (AREA)
- Power Engineering (AREA)
- Information Retrieval, Db Structures And Fs Structures Therefor (AREA)
Abstract
本发明公开了一种基于BiLSTM与串并多尺度CNN结合的轻量化ECG分类方法，属于ECG信号的分类领域，包括以下步骤：获取MIT‑BIH数据集中的ECG时间序列数据，对预处理后的ECG时间序列数据按照比例划分训练集数据和测试集数据，基于ECG信息提取模块、多尺度特征提取模块、维度空间提取模块及BiLSTM与全连接层结合的分类模块构建基于BiLSTM与串并多尺度CNN网络结合的轻量化网络模型；分别将训练集数据以及测试集数据输入到此轻量化网络模型中完成对此网络模型的训练以及测试，最终实现对正常搏动、室上异位搏动、室性异位搏动、融合搏动四分类；该ECG分类网络的设计方法使得本ECG分类网络模型可以达到99.6％以上的准确率。
Description
技术领域
本发明涉及ECG信号的分类领域，尤其涉及一种基于BiLSTM与串并多尺度CNN结合的轻量化ECG分类方法。
背景技术
近年来，心血管疾病发生率不断提升。不仅仅是中老年人，甚至有不少的青壮年都不幸患病，然而还低年龄段人群中却还没有引起足够的重视，这就需要我们增强技术手段，在心脏出现异常时可以做出准确的检测。心电图作为诊断心脏病和评估心脏功能的基础，记录着心脏的心理活动，准确的识别信号所显示的心脏生理活动有利于心脏病的诊断。如何提高分类的准确率，一直以来都是医学领域以及计算机领域的研究热点。
传统的ECG信号分类技术有很多，主要包括支持向量机、线性判别分析、贝叶斯、K最近邻算法等等，主要思想是通过PCA、小波变换、高阶统计量、模糊c均值聚类等特征处理技术与分类算法进行结合，以达到准确分类的目的。传统时间序列的分类方法理论成熟，学习时间相对较短。但是随着数据量的增多，传统算法也存在明显的缺陷，例如不能很好地处理非线性拟合的问题，需要人工处理特征等。并且随着计算机算力的不断提高，基于深度学习的方法逐渐成为了ECG信号分类领域的研究热点。
深度学习包括卷积神经网络(CNN)、循环神经网络(RNN)和生成对抗网络(GAN)；其中卷积神经网络经过多领域的认证其对特征提取具有非常好的效果，目前卷积神经网络应用到导联心电信号的分类中已经有大量的研究。但因为ECG信号的特殊性，不同的提取人群之间、不同的心拍之间以及每个心拍所包含的不同的特征之间，特征尺度相差都很大，而且特征与特征之间的存在的时间、尺度等关联性对ECG信号的分类也会差生极大的影响，这就容易造成对某些类别的漏学习，进而导致模型对ECG信号的识别分类产生错误。
CN111666865A，多导联心电信号卷积神经网络分类模型及其使用方法，该方法使用一个卷积层一个池化层构成若干个串联的特征提取模块，在特征提取模块之后分出与样本类别数量相同的通道个数，形成并联的输出模块，每个输出模块包括一个全连接层和一个输出层。使用新型损失函数判别网络的收敛程度，训练网络参数，以达到最优的网络模型，该方法提取的特征不够丰富，参数过多，特征之间的时间相关性没有得到很好地利用。
发明内容
根据现有技术存在的问题，本发明公开了一种基于BiLSTM与串并多尺度CNN结合的轻量化ECG分类方法，包括以下步骤：
S1：获取MIT-BIH数据集中的ECG时间序列数据，并对该ECG时间序列数据进行预处理，得到预处理后ECG时间序列数据，对预处理后的ECG时间序列数据按照比例划分训练集数据和测试集数据；
S2：构建用于提取ECG时间序列特征信息的ECG信息提取模块、用于提取多个尺度ECG时间序列特征信息的多尺度特征提取模块、用于增加描述ECG时间序列的特征数量的维度空间提取模块以及用于分类的BiLSTM与全连接层结合的分类模块，基于ECG信息提取模块、多尺度特征提取模块、维度空间提取模块及BiLSTM与全连接层结合的分类模块构建基于BiLSTM与串并多尺度CNN网络结合的轻量化网络模型；
S3:将训练集数据批次输入到BiLSTM与串并多尺度CNN结合的轻量化网络模型中对轻量化网络模型进行训练，得到训练好的轻量化ECG分类网络模型；
S4:将测试集数据输入到训练好的轻量化ECG分类网络模型里对测试集数据进行正常搏动、室上异位搏动、室性异位搏动、融合搏动四分类，并将分类结果与该测试集标签进行比较，得到分类的准确率，完成对此轻量化ECG网络模型分类性能的测试。
进一步地，所述对ECG时间序列进行预处理包括以下步骤：
S1-1：获取MIT-BIH心电信号数据，并对该数据采用9尺度小波变换对信号进行去噪处理，得到去噪后的心电信号；
S1-2：将去噪后的心电信号的每一个R峰位为基准点，取每一个R峰的前0.4s到后0.5s从去噪后的心电信号中提取一个心拍样本，得到含有100672个心拍样本的数据集；
S1-3:按照AAMI标准将含有100672个心拍样本的的数据集划分为正常搏动、室上异位搏动、室性异位搏动、融合搏动四个类别，并对四个类别进行one-hot编码；
S1-4:采用交叉验证公式将one-hot编码之后的数据集划分为训练集与测试集。
进一步地，其特征还在于：所述基于BiLSTM与串并多尺度CNN结合的轻量化ECG分类模型的构建方法，步骤如下：
S2-1:在串并多尺度CNN网络中的并行方向搭建三个串行通道，分别为串行通道1、串行通道2、串行通道3，其中串行通道1又分为串行通道1.1串行通道1.2，在串并多尺度CNN网络的每个串行通道上引入多个卷积层与最大池化层结合的ECG信息提取模块，在串并多尺度CNN网络的串行以及并行两个方向上每个卷积层中卷积核的大小皆以等差递增的形式依次增加以获取不同尺度相关性的特征信息，并使ECG信息提取模块与最大池化层进行残差连接搭建多尺度信息提取模块；
在串并多尺度CNN网络中共使用了四个ECG信息提取模块，每个模块有两个卷积层，一个最大池化层，串行通道1有两个ECG信息提取模块，从上到下为ECG信息提取模块1以及串行通道1.2中的ECG信息提取模块2、串行通道2有ECG信息提取模块3、串行通道3有ECG信息提取模块4，每一层中卷积核的大小皆以等差数列递增公式进行递增；
S2-2:串并多尺度CNN网络的串行通道1是在引入了多尺度信息提取模块之后连接了维度空间提取模块；
由串并多尺度CNN网络的串行通道1.1中的最大池化层和以及串行通道1.2中的ECG信息提取模块通过concat融合的方式搭建第一个维度空间提取模块；
S2-3:串并多尺度CNN网络的串行通道2是在多尺度信息提取模块之后引入一个残差模块来增加串行通道2中卷积核提取特征的感受野，进行对ECG时间序列特征的多级提取；
通过对卷积核大小为8的卷积层进行残差连接完成残差模块的搭建；
S2-4:串并多尺度CNN的串行通道3中只引入一个多尺度信息提取模块；
S2-5:串并多尺度CNN网络结构还分别在串行通道1、2、3的多尺度特征提取块之前引入不同大小卷积核的卷积层来保证输入不同尺度的特征；随着串行通道1、2、3的通道数不断增加，各个通道中卷积层的数量也是不断减小来避免由卷积核大小的递增所导致提取特征的感受野过大，最后将串并多尺度CNN的串行通道1、2、3通过concat融合的方式搭建第二个维度空间提取模块，用于增加描述ECG时间序列的特征的提取数量并构建成完整的串并多尺度CNN，第二个维度空间提取模块即完整的串并行多尺度CNN网络；
S2-6:在串并多尺度CNN之后，搭建以BiLSTM为主的分类模块，使用多个BiLSTM与全连接层结合的方式构建分类模块，该模块与串并多尺度CNN网络结合，引入了串并多尺度CNN网络提取的特征之间具有前后相关性的优势实现对ECG时间序列心拍样本的四分类。
进一步地，所述BiLSTM与串并多尺度CNN结合的轻量化网络模型训练过程如下：
S3-1:将训练数据集中的心拍样本批次输入到BiLSTM与串并多尺度CNN结合的轻量化网络模型中提取ECG心电信号的时间序列特征并针对提取的特征完成正常搏动、室上异位搏动、室性异位搏动、融合搏动四个类别分类，将分类的结果与训练集心拍样本的对应标签进行对比，得到分类的准确率。
S3-2:采用与one-hot编码相匹配的交叉熵损失函数计算整个ECG分类网络模型的损失并使用rmsprop进行反向传播以更新整个ECG分类网络模型的参数，直至ECG网络的损失收敛至0.02以下或者达到最高的训练次数，训练结束；保存训练过程中分类准确率最高的一次分类模型，作为最优网络模型。
进一步地，所述BiLSTM与串并多尺度CNN结合的轻量化网络模型测试过程如下:
将测试集的心拍样本输入到训练好的最优网络模型，通过对测试集数据进行正常搏动、室上异位搏动、室性异位搏动、融合搏动四分类，并将分类结果与该测试集中心拍样本对应的标签进行对比，完成对此轻量化网络模型分类性能的测试，得到分类准确率。
由于采用了上述技术方案，本发明提供的一种基于BiLSTM与串并多尺度CNN结合的轻量化ECG分类方法，通过串并结合的思维，首先是构建多层卷积核以等差数列的方式递增的卷积层与池化层结合搭建ECG信息提取模块；在ECG信息提取模块的基础上与最大池化层进行残差连接或者concat融合构建多尺度特征提取模块或者维度空间提取模块；使用多个串行通道并行输入的方式构建维度空间提取模块；串行通道一是在多尺度特征提取模块下方连接的维度空间特征提取模块；串行通道二是在多尺度特征提取模块下方接残差模块；串行通道三只保留多尺度特征提取模块。随着串行通道1、2、3通道数的增加各个卷积层中卷积核大小等差递增、卷积层数等差递减，通过这种思想可以提取更为丰富的ECG时间序列的特征，并保持卷积核进行提取特征时感受野的平衡，从而形成完整的串并行多尺度网络；使用多个双向长短时记忆网路(BiLSTM)与全连接层结合构架分类模块；最终形成完整的一种基于BiLSTM与串并多尺度CNN网络结合的轻量化ECG分类模型；整个网络只有84392个参数，远远低于现有的ECG分类网络模型的参数量，实现了在保证ECG分类网络轻量化的前提下从单个特征所含信息量、时序特征的数量多方面来丰富所提取的特征、从而使得本ECG分类网络模型可以达到99.6％以上的准确率，与现有的AAMI标准四分类技术相比也有了较大的提升。
附图说明
为了更清楚地说明本申请实施例或现有技术中的技术方案，下面将对实施例或现有技术描述中所需要使用的附图作简单地介绍，显而易见地，下面描述中的附图仅仅是本申请中记载的一些实施例，对于本领域普通技术人员来讲，在不付出创造性劳动的前提下，还可以根据这些附图获得其他的附图。
图1是本发明所述ECG时间序列分类方法的整体设计流程图；
图2是本发明所述ECG时间序列分类方法的ECG信息提取模型结构图；
图3是本发明所述ECG时间序列分类方法的多尺度信息提取模块结构图；
图4是本发明所述ECG时间序列分类方法的第一个维度空间提取模块结构图；
图5是本发明所述ECG时间序列分类方法的串并多尺度CNN中串行通道2的残差模块结构图；
图6是本发明所述ECG时间序列分类方法的分类模块结构图；
图7是本发明所述ECG时间序列分类方法的完整的轻量化ECG分类网络模型图；
图8是本发明混淆矩阵图。
具体实施方式
为使本发明的技术方案和优点更加清楚，下面结合本发明实施例中的附图，对本发明实施例中的技术方案进行清楚完整的描述：
图1是本发明所述ECG时间序列分类方法的整体设计流程图；一种基于BiLSTM与串并多尺度CNN结合的轻量化ECG分类方法，包括以下步骤：
S1：获取MIT-BIH数据集中的ECG时间序列数据，并对该ECG时间序列数据进行预处理，得到预处理后ECG时间序列数据，对预处理后的ECG时间序列数据按照比例划分训练集数据和测试数据；
S2：搭建用于提取ECG时间序列特征信息的ECG信息提取模块、用于提取多个尺度ECG时间序列特征信息的多尺度特征提取模块、用于增加描述ECG时间序列的特征数量的维度空间提取模块以及用于分类的BiLSTM与全连接层结合的分类模块，基于ECG信息提取模块、多尺度特征提取模块、维度空间提取模块及BiLSTM与全连接层结合的分类模块构建基于BiLSTM与串并多尺度CNN网络结合的轻量化网络模型；
S3:将训练集数据批次输入到BiLSTM与串并多尺度CNN结合的轻量化网络模型中对轻量化网络模型进行训练，得到训练好的轻量化ECG分类网络模型；
S4:将测试集数据输入到训练好的轻量化ECG分类网络模型里对测试集数据进行正常搏动、室上异位搏动、室性异位搏动、融合搏动四分类，并将分类结果与该测试集标签进行对比，得到分类的准确率，完成对此轻量化ECG网络模型分类性能的测试。
进一步地：所述对ECG时间序列数据进行预处理的过程如下：
S1-1：获取MIT-BIH心电信号数据，并对该数据采用9尺度小波变换对信号进行去噪处理，得到去噪后的心电信号；
S1-2：对于去噪后的心电信号，以该去噪后的心电信号的每一个R峰位为基准点，取每一个R峰的前0.4s到后0.5s从去噪后的心电信号中提取一个心拍样本，最后得到一个含有100672个心拍样本的数据集；
S1-3：按照AAMI标准将上述获得的数据集划分为正常搏动(N)，室上异位搏动(S)，室性异位搏动(V)，融合搏动(F)和未知搏动(Q)五类，由于第五类Q包含许多未知搏动类，故本实验只进行N、S、V、F四分类验证,并对数据集的N、S、V、F四种类别进行one-hot编码；
S1-4:将one-hot编码之后的数据集采用的交叉验证的方式将其划分为训练集与测试集，交叉验证的划分公式如下；
m2＝n-m1 (2)
其中：n为样本总数，k为交叉验证折数，m1为测试集样本数，m2为训练集样本数。
进一步地，所述一种基于BiLSTM与串并多尺度CNN结合的轻量化ECG分类模型的构建方法，步骤如下：
S2-1：在串并多尺度CNN网络中的并行方向共搭建三个串行通道，分别为串行通道1、串行通道2、串行通道3，其中串行通道1又分为串行通道1.1串行通道1.2，在串并多尺度CNN网络的每个串行通道上引入多个卷积层与最大池化层结合的ECG信息提取模块，在串并多尺度CNN网络的串行以及并行两个方向上每个卷积层中卷积核的大小皆以等差递增的形式依次增加以获取不同尺度相关性的特征信息；并使ECG信息提取模块与最大池化层进行残差连接搭建多尺度信息提取模块，通过这种模块搭建的思想以提取更为丰富的多尺度特征信息；
图2是本发明所述ECG时间序列分类方法的ECG信息提取模块结构图，每个ECG信息提取模块有两个卷积层，一个最大池化层；在串并多尺度CNN网络中共使用了四个ECG信息提取模块，串行通道1有两个ECG信息提取模块，从上到下为ECG信息提取模块1以及串行通道1.2中的ECG信息提取模块2、串行通道2有ECG信息提取模块3、串行通道3有ECG信息提取模块4，每一层中卷积核的大小皆以等差数列递增；
卷积核大小的递增公式为：
m1＝2×(n+1)，1≤n≤4 (3)
m2＝2×(n+2)，1≤n≤4 (4)
其中，m1、m2为ECG信息提取模块中的第一层与第二层，n为ECG信息提取模块的序号；
图3是本发明所述ECG时间序列分类方法的多尺度信息提取模块结构图，多尺度信息提取模块由ECG信息提取模块与最大池化层进行残差连接而成；
S2-2：在搭建完用来提取ECG时间序列更为丰富的多尺度特征信息的ECG信息提取模块以及多尺度信息提取模块之后，串并多尺度CNN的串行通道1首先引入一个多尺度信息提取模块，之后在多尺度信息提取模块下方连接的是维度空间模块，该维度空间提取模块用于增加描述ECG时间序列的特征提取数量；
图4是本发明所述ECG时间序列分类方法的第一个维度空间提取模块结构图，由串并多尺度CNN的串行通道1.1中的最大池化层和以及串行通道1.2中的ECG信息提取模块通过concat融合的方式搭建第一个维度空间提取模块。
S2-3:在第一个维度空间提取模块之后完成对串并多尺度CNN的串行通道1的设计，接下来串并多尺度CNN的串行通道2是在多尺度信息提取模块之后引入一个残差模块来增加串行通道2中卷积核进行特征提取的感受野，实现对ECG时间序列特征的多级提取；
图5是本发明所述ECG时间序列分类方法的串并多尺度CNN中串行通道2的残差模块结构图，只有一个卷积核大小为8的卷积层，通过对该卷积层进行残差连接完成残差模块的搭建；
S2-4:在残差模块之后完成对串并多尺度CNN的串行通道2的设计，而串并多尺度CNN的串行通道3中只引入一个多尺度信息提取模块。
S2-5:串并多尺度CNN网络的串行通道3完成之后，本串并多尺度CNN网络结构还分别在串行通道1、2、3的多尺度特征提取块之前引入不同大小的卷积层来保证输入不同尺度的特征；随着串行通道1、2、3的通道数不断增加，卷积层的数量也是不断减小避免由卷积核大小的递增所导致的卷积核进行特征提取的感受野过大，最后将串并多尺度CNN的串行通道1、2、3通过concat融合的方式搭建第二个维度空间提取模块，用于增加描述ECG时间序列的特征的提取数量并构建成完整的串并多尺度CNN，第二个维度空间提取模块即完整的串并行多尺度CNN网络；
S2-6:在串并多尺度CNN网络之后，搭建以BiLSTM为主的分类模块，使用多个BiLSTM与全连接层结合的方式构建分类模型，该模块与串并多尺度CNN网络结合，引入了串并多尺度CNN网络提取的特征之间具有前后相关性的优势实现对ECG时间序列心拍样本的四分类。
图6是本发明所述ECG时间序列分类方法的分类模块结构图，由两层BiLSTM与一层全连接层组成。
图7本发明所述ECG时间序列分类方法的完整的轻量化ECG分类网络模型图，由完整的串并多尺度CNN网络与分类模块结合而成。
进一步地，所述BiLSTM与串并多尺度CNN结合的轻量化网络模型训练过程如下：
S3-1:将训练集中的心拍样本输入到BiLSTM与串并多尺度CNN结合的轻量化网络模型提取ECG心电信号的时间序列特征并针对提取的特征完成N、S、V、F四分类，将分类的结果与训练集心拍样本的对应标签进行对比，得到分类的准确率；
S3-2:为了优化分类的准确率，本研究采用与one-hot编码相匹配的交叉熵损失函数计算整个ECG分类网络模型的损失并使用rmsprop进行反向传播以更新整个ECG分类网络模型的参数，直至此网络的损失收敛至0.02以下或者达到最高的训练次数，训练结束。保存训练过程中分类准确率最高的一次分类模型，作为我们的最优网络模型；
进一步地，所述BiLSTM与串并多尺度CNN结合的轻量化网络模型测试过程如下:
S4-1:将测试集的心拍样本输入到训练好的最优网络模型，通过对测试集数据进行N、S、V、F四分类，并将分类结果与该测试集中心拍样本对应的标签进行对比，完成对此轻量化网络模型分类性能的测试，得到最终的分类准确率。
本发明在遵循AAMI国际标准的规则之下，仅使用84392个少量参数实现了有与现有技术可比甚至更优的分类精度，图8是本发明混淆矩阵图，实验中敏感度、阳性预测值以及准确度相比现有技术皆有提升，准确度提高到99.6％。
以上所述，仅为本发明较佳的具体实施方式，但本发明的保护范围并不局限于此，任何熟悉本技术领域的技术人员在本发明揭露的技术范围内，根据本发明的技术方案及其发明构思加以等同替换或改变，都应涵盖在本发明的保护范围之内。
Claims (5)
1.一种基于BiLSTM与串并多尺度CNN结合的轻量化ECG分类方法，其特征在于：包括以下步骤：
S1：获取MIT-BIH数据集中的ECG时间序列数据，并对该ECG时间序列数据进行预处理，得到预处理后ECG时间序列数据，对预处理后的ECG时间序列数据按照比例划分训练集数据和测试集数据；
S2：构建用于提取ECG时间序列特征信息的ECG信息提取模块、用于提取多个尺度ECG时间序列特征信息的多尺度特征提取模块、用于增加描述ECG时间序列的特征数量的维度空间提取模块以及用于分类的BiLSTM与全连接层结合的分类模块，基于ECG信息提取模块、多尺度特征提取模块、维度空间提取模块及BiLSTM与全连接层结合的分类模块构建基于BiLSTM与串并多尺度CNN网络结合的轻量化网络模型；
S3:将训练集数据批次输入到BiLSTM与串并多尺度CNN结合的轻量化网络模型中对轻量化网络模型进行训练，得到训练好的轻量化ECG分类网络模型；
S4:将测试集数据输入到训练好的轻量化ECG分类网络模型里对测试集数据进行正常搏动、室上异位搏动、室性异位搏动、融合搏动四分类，并将分类结果与该测试集标签进行对比，得到分类的准确率，完成对此轻量化ECG网络模型分类性能的测试。
2.根据权利要求1所述的一种基于BiLSTM与串并多尺度CNN结合的轻量化ECG分类方法，其特征还在于：所述对ECG时间序列进行预处理包括以下步骤：
S1-1：获取MIT-BIH心电信号数据，并对该数据采用9尺度小波变换对信号进行去噪处理，得到去噪后的心电信号；
S1-2：将去噪后的心电信号的每一个R峰位为基准点，取每一个R峰的前0.4s到后0.5s从去噪后的心电信号中提取一个心拍样本，得到含有100672个心拍样本的数据集；
S1-3:按照AAMI标准将含有100672个心拍样本的的数据集划分为正常搏动、室上异位搏动、室性异位搏动、融合搏动四个类别，并对四个类别进行one-hot编码；
S1-4:采用交叉验证公式将one-hot编码之后的数据集划分为训练集与测试集。
3.根据权利要求1所述的一种基于BiLSTM与串并多尺度CNN结合的轻量化ECG分类方法，其特征还在于：所述基于BiLSTM与串并多尺度CNN结合的轻量化ECG分类模型的构建方法，步骤如下：
S2-1:在串并多尺度CNN网络中的并行方向搭建三个串行通道，分别为串行通道1、串行通道2、串行通道3，其中串行通道1又分为串行通道1.1串行通道1.2，在串并多尺度CNN网络的每个串行通道上引入多个卷积层与最大池化层结合的ECG信息提取模块，在串并多尺度CNN网络的串行以及并行两个方向上每个卷积层中卷积核的大小皆以等差递增的形式依次增加以获取不同尺度相关性的特征信息，并使ECG信息提取模块与最大池化层进行残差连接搭建多尺度信息提取模块；
在串并多尺度CNN网络中共使用了四个ECG信息提取模块，每个模块有两个卷积层，一个最大池化层，串行通道1有两个ECG信息提取模块，从上到下为ECG信息提取模块1以及串行通道1.2中的ECG信息提取模块2、串行通道2有ECG信息提取模块3、串行通道3有ECG信息提取模块4，每个ECG信息提取模块分别与最大池化层通过不同的组合方式构成多尺度信息模块或者维度空间提取模块；
S2-2:串并多尺度CNN网络的串行通道1是在引入了多尺度信息提取模块之后连接了维度空间提取模块；
由串并多尺度CNN网络的串行通道1.1中的最大池化层和以及串行通道1.2中的ECG信息提取模块通过concat融合的方式搭建第一个维度空间提取模块；
S2-3:串并多尺度CNN网络的串行通道2是在多尺度信息提取模块之后引入一个残差模块来增加串行通道2中特征提取的感受野，进行对ECG时间序列特征的多级提取；
通过对卷积核大小为8的卷积层进行残差连接完成残差模块的搭建；
S2-4:串并多尺度CNN的串行通道3中只引入一个多尺度信息提取模块；
S2-5:串并多尺度CNN网络结构还分别在串行通道1、2、3的多尺度特征提取块之前引入不同大小卷积核的卷积层来保证输入不同尺度的特征；随着串行通道1、2、3的通道数不断增加，各个通道中卷积层的数量也是不断减小来避免由卷积核大小的递增所导致提取特征的感受野过大，最后将串并多尺度CNN的串行通道1、2、3通过concat融合的方式搭建第二个维度空间提取模块，用于增加描述ECG时间序列的特征的提取数量并构建成完整的串并多尺度CNN，第二个维度空间提取模块即完整的串并行多尺度CNN网络；
S2-6:在串并多尺度CNN网络之后，搭建以BiLSTM为主的分类模块，使用多个BiLSTM与全连接层结合的方式构建分类模块，该模块与串并多尺度CNN网络结合，引入了串并多尺度CNN网络提取的特征之间具有前后相关性的优势实现对ECG时间序列心拍样本的四分类。
4.根据权利要求1所述的一种基于BiLSTM与串并多尺度CNN结合的轻量化ECG分类方法，其特征还在于：所述BiLSTM与串并多尺度CNN结合的轻量化网络模型训练过程如下：
S3-1:将训练数据集中的心拍样本输入到BiLSTM与串并多尺度CNN结合的轻量化网络模型提取ECG心电信号的时间序列特征并针对提取的特征完成正常搏动、室上异位搏动、室性异位搏动、融合搏动四个类别分类，将分类的结果与训练集心拍样本的对应标签进行对比，得到分类的准确率。
S3-2:采用与one-hot编码相匹配的交叉熵损失函数计算整个ECG分类网络模型的损失并使用rmsprop进行反向传播以更新整个ECG分类网络模型的参数，直至ECG网络的损失收敛至0.02以下或者达到最高的训练次数，训练结束；保存训练过程中分类准确率最高的一次分类模型，作为最优网络模型。
5.根据权利要求1所述的一种基于BiLSTM与串并多尺度CNN结合的轻量化ECG分类方法，其特征还在于：所述BiLSTM与串并多尺度CNN结合的轻量化网络模型测试过程如下:
将测试集的心拍样本输入到训练好的最优网络模型，通过对测试集数据进行正常搏动、室上异位搏动、室性异位搏动、融合搏动四分类，并将分类结果与该测试集中心拍样本对应的标签进行对比，完成对此轻量化网络模型分类性能的测试，得到分类准确率。
Priority Applications (1)
| Application Number | Priority Date | Filing Date | Title |
|---|---|---|---|
| CN202110063840.2A CN112733774A (zh) | 2021-01-18 | 2021-01-18 | 一种基于BiLSTM与串并多尺度CNN结合的轻量化ECG分类方法 |
Applications Claiming Priority (1)
| Application Number | Priority Date | Filing Date | Title |
|---|---|---|---|
| CN202110063840.2A CN112733774A (zh) | 2021-01-18 | 2021-01-18 | 一种基于BiLSTM与串并多尺度CNN结合的轻量化ECG分类方法 |
Publications (1)
| Publication Number | Publication Date |
|---|---|
| CN112733774A true CN112733774A (zh) | 2021-04-30 |
Family
ID=75592229
Family Applications (1)
| Application Number | Title | Priority Date | Filing Date |
|---|---|---|---|
| CN202110063840.2A Pending CN112733774A (zh) | 2021-01-18 | 2021-01-18 | 一种基于BiLSTM与串并多尺度CNN结合的轻量化ECG分类方法 |
Country Status (1)
| Country | Link |
|---|---|
| CN (1) | CN112733774A (zh) |
Cited By (6)
| Publication number | Priority date | Publication date | Assignee | Title |
|---|---|---|---|---|
| CN113499048A (zh) * | 2021-07-22 | 2021-10-15 | 重庆理工大学 | 一种基于CNN-BiLSTM的中心动脉压波形重构的系统及方法 |
| CN113616216A (zh) * | 2021-09-10 | 2021-11-09 | 郑州大学 | 一种基于BiLSTM-Treg的心电信号分类方法 |
| CN114067153A (zh) * | 2021-11-02 | 2022-02-18 | 暨南大学 | 基于并行双注意力轻量残差网络的图像分类方法及系统 |
| CN114190889A (zh) * | 2021-11-19 | 2022-03-18 | 上海联影智能医疗科技有限公司 | 心电信号的分类方法、系统、电子设备及可读存储介质 |
| CN114469120A (zh) * | 2022-01-12 | 2022-05-13 | 大连海事大学 | 一种基于相似度阈值迁移的多尺度Dtw-BiLstm-Gan心电信号生成方法 |
| CN116616792A (zh) * | 2023-07-21 | 2023-08-22 | 山东大学 | 基于轻量化设计和特征融合的心房颤动检测系统 |
Citations (2)
| Publication number | Priority date | Publication date | Assignee | Title |
|---|---|---|---|---|
| WO2020151075A1 (zh) * | 2019-01-23 | 2020-07-30 | 五邑大学 | 一种基于cnn-lstm深度学习模型的驾驶疲劳识别方法 |
| CN111695469A (zh) * | 2020-06-01 | 2020-09-22 | 西安电子科技大学 | 轻量化深度可分离卷积特征融合网的高光谱图像分类方法 |
-
2021
- 2021-01-18 CN CN202110063840.2A patent/CN112733774A/zh active Pending
Patent Citations (2)
| Publication number | Priority date | Publication date | Assignee | Title |
|---|---|---|---|---|
| WO2020151075A1 (zh) * | 2019-01-23 | 2020-07-30 | 五邑大学 | 一种基于cnn-lstm深度学习模型的驾驶疲劳识别方法 |
| CN111695469A (zh) * | 2020-06-01 | 2020-09-22 | 西安电子科技大学 | 轻量化深度可分离卷积特征融合网的高光谱图像分类方法 |
Non-Patent Citations (2)
| Title |
|---|
| 孙梦莉;王演;刘晓凯;林卉;张世强;: "基于深度迁移学习的心电信号分类方法", 计算机应用与软件, no. 05 * |
| 蒋仕宝;杜翠凤;聂丹彤;: "融合CNN与LSTM的网络质量KQI数据特征提取与投诉预警", 移动通信, no. 02 * |
Cited By (8)
| Publication number | Priority date | Publication date | Assignee | Title |
|---|---|---|---|---|
| CN113499048A (zh) * | 2021-07-22 | 2021-10-15 | 重庆理工大学 | 一种基于CNN-BiLSTM的中心动脉压波形重构的系统及方法 |
| CN113616216A (zh) * | 2021-09-10 | 2021-11-09 | 郑州大学 | 一种基于BiLSTM-Treg的心电信号分类方法 |
| CN113616216B (zh) * | 2021-09-10 | 2024-02-27 | 郑州大学 | 一种基于BiLSTM-Treg的心电信号分类方法 |
| CN114067153A (zh) * | 2021-11-02 | 2022-02-18 | 暨南大学 | 基于并行双注意力轻量残差网络的图像分类方法及系统 |
| CN114190889A (zh) * | 2021-11-19 | 2022-03-18 | 上海联影智能医疗科技有限公司 | 心电信号的分类方法、系统、电子设备及可读存储介质 |
| CN114469120A (zh) * | 2022-01-12 | 2022-05-13 | 大连海事大学 | 一种基于相似度阈值迁移的多尺度Dtw-BiLstm-Gan心电信号生成方法 |
| CN116616792A (zh) * | 2023-07-21 | 2023-08-22 | 山东大学 | 基于轻量化设计和特征融合的心房颤动检测系统 |
| CN116616792B (zh) * | 2023-07-21 | 2023-12-01 | 山东大学 | 基于轻量化设计和特征融合的心房颤动检测系统 |
Similar Documents
| Publication | Publication Date | Title |
|---|---|---|
| CN112733774A (zh) | 一种基于BiLSTM与串并多尺度CNN结合的轻量化ECG分类方法 | |
| Ruiz et al. | The great multivariate time series classification bake off: a review and experimental evaluation of recent algorithmic advances | |
| Shi et al. | Automated heartbeat classification based on deep neural network with multiple input layers | |
| CN110598793B (zh) | 一种大脑功能网络特征分类方法 | |
| CN111990989A (zh) | 一种基于生成对抗及卷积循环网络的心电信号识别方法 | |
| CN109009102B (zh) | 一种基于脑电图深度学习的辅助诊断方法及系统 | |
| CN111297349A (zh) | 一种基于机器学习的心律分类系统 | |
| CN111540467B (zh) | 精神分裂症分类识别方法、运行控制装置及医疗设备 | |
| CN113229825A (zh) | 一种基于深度神经网络的多标签多导联心电图分类方法 | |
| Wang et al. | Three-heartbeat multilead ECG recognition method for arrhythmia classification | |
| CN114970637A (zh) | 一种轻量级基于深度学习的心律失常分类方法 | |
| CN113274031A (zh) | 一种基于深度卷积残差网络的心律失常分类方法 | |
| CN113768515A (zh) | 一种基于深度卷积神经网络的心电信号分类方法 | |
| CN115530788A (zh) | 基于自注意力机制的心律失常分类方法 | |
| KR102169637B1 (ko) | 사망 위험도의 예측 방법 및 이를 이용한 사망 위험도의 예측 디바이스 | |
| Wu et al. | SRT: Improved transformer-based model for classification of 2D heartbeat images | |
| Liu et al. | Automatic classification of arrhythmias using multi-branch convolutional neural networks based on channel-based attention and bidirectional LSTM | |
| Prabhakararao et al. | Congestive heart failure detection from ECG signals using deep residual neural network | |
| CN112957052B (zh) | 一种基于nlf-cnn导联融合深度网络的多导联心电信号分类方法 | |
| CN117476226A (zh) | 基于特征融合的多尺度卷积神经网络糖尿病风险评估方法 | |
| CN117612732A (zh) | 基于深度学习的多模态健康管理方案生成方法及其系统 | |
| CN116649899A (zh) | 一种基于注意力机制特征融合的心电信号分类方法 | |
| CN115909438A (zh) | 基于深度时空域卷积神经网络的疼痛表情识别系统 | |
| CN115886833A (zh) | 心电信号分类方法及装置、计算机可读介质、电子设备 | |
| CN115500807A (zh) | 基于小型卷积神经网络的心律失常分类检测方法及系统 |
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 | ||
| AD01 | Patent right deemed abandoned | ||
| AD01 | Patent right deemed abandoned |
Effective date of abandoning: 20240524 |