CN110890155A - 一种基于导联注意力机制的多类心律失常检测方法 - Google Patents

一种基于导联注意力机制的多类心律失常检测方法 Download PDF

Info

Publication number
CN110890155A
CN110890155A CN201911167791.6A CN201911167791A CN110890155A CN 110890155 A CN110890155 A CN 110890155A CN 201911167791 A CN201911167791 A CN 201911167791A CN 110890155 A CN110890155 A CN 110890155A
Authority
CN
China
Prior art keywords
lead
ecg signal
attention mechanism
classification
global
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.)
Granted
Application number
CN201911167791.6A
Other languages
English (en)
Other versions
CN110890155B (zh
Inventor
陈勋
张静
梁邓
高敏
张旭
陈香
Current Assignee (The listed assignees may be inaccurate. Google has not performed a legal analysis and makes no representation or warranty as to the accuracy of the list.)
University of Science and Technology of China USTC
Original Assignee
University of Science and Technology of China USTC
Priority date (The priority date 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 date listed.)
Filing date
Publication date
Application filed by University of Science and Technology of China USTC filed Critical University of Science and Technology of China USTC
Priority to CN201911167791.6A priority Critical patent/CN110890155B/zh
Publication of CN110890155A publication Critical patent/CN110890155A/zh
Application granted granted Critical
Publication of CN110890155B publication Critical patent/CN110890155B/zh
Active legal-status Critical Current
Anticipated expiration legal-status Critical

Links

Images

Classifications

    • GPHYSICS
    • G16INFORMATION AND COMMUNICATION TECHNOLOGY [ICT] SPECIALLY ADAPTED FOR SPECIFIC APPLICATION FIELDS
    • G16HHEALTHCARE INFORMATICS, i.e. INFORMATION AND COMMUNICATION TECHNOLOGY [ICT] SPECIALLY ADAPTED FOR THE HANDLING OR PROCESSING OF MEDICAL OR HEALTHCARE DATA
    • G16H50/00ICT specially adapted for medical diagnosis, medical simulation or medical data mining; ICT specially adapted for detecting, monitoring or modelling epidemics or pandemics
    • G16H50/20ICT specially adapted for medical diagnosis, medical simulation or medical data mining; ICT specially adapted for detecting, monitoring or modelling epidemics or pandemics for computer-aided diagnosis, e.g. based on medical expert systems
    • AHUMAN NECESSITIES
    • A61MEDICAL OR VETERINARY SCIENCE; HYGIENE
    • A61BDIAGNOSIS; SURGERY; IDENTIFICATION
    • A61B5/00Measuring for diagnostic purposes; Identification of persons
    • A61B5/24Detecting, measuring or recording bioelectric or biomagnetic signals of the body or parts thereof
    • A61B5/316Modalities, i.e. specific diagnostic methods
    • AHUMAN NECESSITIES
    • A61MEDICAL OR VETERINARY SCIENCE; HYGIENE
    • A61BDIAGNOSIS; SURGERY; IDENTIFICATION
    • A61B5/00Measuring for diagnostic purposes; Identification of persons
    • A61B5/24Detecting, measuring or recording bioelectric or biomagnetic signals of the body or parts thereof
    • A61B5/316Modalities, i.e. specific diagnostic methods
    • A61B5/318Heart-related electrical modalities, e.g. electrocardiography [ECG]
    • AHUMAN NECESSITIES
    • A61MEDICAL OR VETERINARY SCIENCE; HYGIENE
    • A61BDIAGNOSIS; SURGERY; IDENTIFICATION
    • A61B5/00Measuring for diagnostic purposes; Identification of persons
    • A61B5/72Signal processing specially adapted for physiological signals or for diagnostic purposes
    • A61B5/7235Details of waveform analysis
    • A61B5/7264Classification of physiological signals or data, e.g. using neural networks, statistical classifiers, expert systems or fuzzy systems
    • A61B5/7267Classification of physiological signals or data, e.g. using neural networks, statistical classifiers, expert systems or fuzzy systems involving training the classification device
    • AHUMAN NECESSITIES
    • A61MEDICAL OR VETERINARY SCIENCE; HYGIENE
    • A61BDIAGNOSIS; SURGERY; IDENTIFICATION
    • A61B5/00Measuring for diagnostic purposes; Identification of persons
    • A61B5/72Signal processing specially adapted for physiological signals or for diagnostic purposes
    • A61B5/7271Specific aspects of physiological measurement analysis

Abstract

本发明公开了一种基于导联注意力机制的多类心律失常检测方法,其步骤包括:1、对原始ECG信号进行下采样和补齐到固定长度的预处理;2、设计分类模型,集成了四种网络结构,包括导联注意力机制、卷积神经网络、双向门控循环单元和时间注意力机制;3、在公开数据集上采用四折交叉验证训练上述模型;4、利用训练好的模型实现心律失常分类任务。本发明能实现高准确率心律失常自动检测,从而为医生的诊断提供辅助。

Description

一种基于导联注意力机制的多类心律失常检测方法
技术领域
本发明涉及医学信号处理领域,尤其涉及从心电图信号中检测心律失常的方法。
背景技术
心脏心律失常是一种心脏电活动不规则的情况。许多类型的心律失常是损害健康甚至危及生命的,如室性心动过速、心室颤动是致死性心律失常。心电图(electrocardiogram,ECG)通过放置在皮肤上的电极记录了一段时间内心脏的电活动,在临床上被广泛用于心律失常检测。心电图通过不同的导联从不同的角度捕捉心脏的电势,通过波形或节律的变化反应心脏心律情况。医生可以通过心电图了解病人患各种心脏病的风险。一个24小时的心电监测仪记录到的心拍多达几十万,分析这些心拍对医生来说是一项耗时的工作,可能会出现误判或漏判。因此,构建一个准确的自动心律失常检测方法辅助医生工作对临床心律失常诊断具有重要帮助。
在过去的几十年里,提出了大量的心律失常检测方法。这些方法主要由四个步骤组成,包括降噪、特征提取和选择、特征变换和心律失常分类。在这些步骤中,特征提取将输入的心电信号转化为多种特征,这些特征在心律失常的检测中起着重要作用,主要有P-QRS-T波特征、统计特征、形态特征、小波特征。特征变换将原始的高维心电信号转化为一个更低维的子空间,实现了降维的目的,可以通过主成分分析、线性判别分析、独立成分分析等方法实现。特征工程之后,各种各样的分类器用来对心电信号分类,常用的分类器主要有六种,包括人工神经网络、线性判别分析、k近邻、支持向量机、决策树和贝叶斯分类器。
近些年,深度神经网络(deep neural network,DNN)凭借其强大的特征提取能力,在计算机视觉领域取得了巨大的成功。不同于传统的机器学习方法,DNN是一种受数据驱动的方法,无需特征工程的设计。许多研究工作也尝试用DNN解决心律失常检测的问题。卷积神经网络(convolutional neural network,CNN)常被用于分析生理信号,随着卷积层的加深,特征以一种更加抽象的方式被学习和表示。长短时记忆网络(long short-termmemory,LSTM)是一种广泛用于分析时间序列的深度学习方法,也常用于心律失常检测。不少研究人员将LSTM应用在卷积神经网络提取的特征之后,这种堆叠的网络实现了较高的心律失常检测性能。
很多研究工作虽然在心律失常检测上取得了一定的效果,可是,他们通常忽略了不是所有的导联,不是所有的时间步对检测心律失常都有同样的贡献。最近,”Multi-classArrhythmia detection from 12-lead varied-length ECG using Attention-basedTime-Incremental Convolutional Neural Network”提出一种有效的多类心律失常检测方法,集成了卷积神经网络、循环元胞和注意力机制。具有意义地是,他们引入了时间维度的注意力机制给从不同的信号片段提取的特征分配相应的权重,帮助定位了信号的感兴趣片段。然而,该研究工作并没有关注“哪些导联”对给定的心电记录是有意义的。
为了便于分析心律失常,最好同步记录3或6个导联,甚至12个导联,以满足临床诊断的需要。换句话说,导联的信息是有冗余的。事实上,某个缺失的肢体导联可以通过其它已知肢体导联由公式计算得出。比如,根据Einthoven定律,II导联QRS波的振幅等于I和III导联QRS波振幅的总和。心前区导联V1-V6虽然不能通过计算得出,心脏病专家却可以根据经验推断出大致的波形。在之前的工作中,心电信号的12导联特性被忽略。
发明内容
本发明为克服现有技术的不足之处,提出一种基于导联注意力机制的多类心律失常检测方法,以期能实现高准确率心律失常自动检测。
本发明为达到上述发明目的,采用如下技术方案:
本发明一种基于导联注意力机制的多类心律失常检测方法的特点在于,包括以下步骤:
步骤一、获取原始ECG信号及其对应的标签,对所述原始ECG信号进行下采样,得到采样后的ECG信号;对所述采样后的ECG信号进行裁剪或均值补齐处理,得到预处理后的ECG信号;
步骤二、构建由导联注意力机制模块、卷积神经网络、双向门控循环单元和时间注意力机制模块组成的分类网络;
步骤2.1、构建导联注意力机制模块,包括两个并行的全局池化层、c个全连接层、sigmoid函数层:
将所述预处理后的ECG信号分别输入全局平均池化层和全局最大池化层,得到全局平均池化特征图和全局最大池化特征图,再分别经过所述c个全连接层的处理后,得到两个输出特征图并相加,从而得到融合的全局池化特征图,并经过Sigmoid函数层后,输出导联注意力图;将所述导联注意力图与预处理后的ECG信号相乘,最终得到沿导联轴加权后的ECG信号;
步骤2.2、构建卷积神经网络:
所述卷积神经网络由n个卷积层组成,并以m个卷积层作为一个卷积块,设置每个卷积块前m-1个卷积层的核尺寸小于第m个卷积层的核尺寸;并设置卷积层的核数量均为K;
在每个卷积层之后设置有LeakyReLU激活函数层;
将所述沿导联轴加权后的ECG信号输入所述卷积神经网络中,得到ECG信号的局部特征;
步骤2.3、将所述ECG信号的局部特征输入所述双向门控循环单元,得到ECG信号的全局特征;
步骤2.4、将所述ECG信号的全局特征输入所述时间注意力机制模块,得到沿时间轴加权后的ECG信号;
步骤三、采用K折交叉验证法训练所述分类网络,得到K个分类模型;
步骤四、利用K个分类模型对预处理后的待预测ECG信号分别进行预测,得到各个类别的K个预测概率,并对K个预测概率取均值,从而得到各类的平均预测概率,将平均预测概率最大的类别作为待预测的ECG信号预测标签。
本发明所述的多类心律失常检测方法的特点是,所述步骤三是按如下过程进行:
步骤3.1、定义外循环的最大循环次数为K;定义内迭代的最大迭代次数为Emax
定义当前外循环次数为k,定义当前内迭代次数为e;
将预处理后的ECG信号及其对应的标签划分成同等大小的K个子集;
步骤3.2、初始化k=1;
步骤3.3、将第k个子集作为验证集,其余K-1个子集作为第k次循环的训练集;
步骤3.4、将所述第k次循环的训练集以分批的方式输入到所述分类网络中;
步骤3.5、设置交叉熵作为循环的损失函数;
步骤3.6、利用He正态分布初始化器第k次初始化所述导联注意力机制模块的c个全连接层;利用Xaiver均匀分布初始化器第k次初始化所述卷积神经网络中n个卷积层的权重;利用生成随机正交矩阵的初始化器第k次初始化所述双向门控循环单元的元胞;
步骤3.7、利用Adam优化器优化所述分类模型中各个参数,设置学习率为l;
步骤3.8、在每个卷积块、所述双向门控循环单元和时间注意力机制模块的输出后分别设置Dropout层,丢弃率设置为d;
步骤3.9、初始化e=1;定义性能阈值为Fmax,并初始化Fmax=0;定义计数器count,并初始化count=0;
步骤3.10、在第k次外循环的第e次内迭代下利用Adam优化器对所述分类模型进行训练,得到第e次训练后的分类模型;
步骤3.11、计算第e次训练后的分类模型在验证集上的准确率Fe;
步骤3.12、判断Fe>Fmax是否成立,若成立,则将Fe赋值给Fmax,否则,令计数器count加“1”,并保持Fmax不变;
步骤3.13、判断e+1赋值给e后,e>Emax或count达到所设定阈值是否成立,若成立,则将Fmax所对应的分类模型作为第k次外循环下的分类模型,并执行步骤3.14;否则,返回步骤3.10;
步骤3.14、将k+1赋值给k后,判断k>K是否成立,若成立,则表示得到K个分类模型,并执行步骤四,否则,返回步骤3.3。
与已有技术相比,本发明的有益效果体现在:
1、本发明通过构建由导联注意力机制模块、卷积神经网络、双向门控循环单元和时间注意力机制模块组成的分类网络,给不同的导联、不同的信号片段分配权重,能够定位到空间和时间上ECG信号的信息性部分,在多类心律失常检测上实现了平均F1得分为0.8634的高性能;
2、本发明提出了导联注意力机制,通过给不同的导联分配权重,定位到空间上ECG信号的信息性部分,提高了3.33%的心律失常检测性能;
3、本发明通过卷积神经网络提取ECG信号的局部特征,提取的特征提高了心律失常检测性能;
4、本发明通过双向门控循环单元提取ECG信号的全局特征,提取的特征提高了心律失常检测性能;
5、本发明通过时间注意力机制给不同的信号片段分配权重,定位到时间上ECG信号的信息性部分,提高了心律失常检测性能。
附图说明
图1为本发明方法的流程图;
图2为本发明网络结构图,Conv1D 3×12代表卷积核尺寸为3,核数量为12的卷积层,以此类推;
图3为本发明导联注意力机制结构图;
图4为本发明方法和未集成导联注意力机制模块时对原始的正常导联II信号的注意力映射对比图;
图5为本发明方法和未集成导联注意力机制模块时对原始的房颤导联II信号的注意力映射对比图;
图6为本发明方法和未集成导联注意力机制模块时对原始的一度房室传导阻滞导联II信号的注意力映射对比图;
图7为本发明方法和未集成导联注意力机制模块时对原始的左束支传导阻滞导联II信号的注意力映射对比图;
图8为本发明方法和未集成导联注意力机制模块时对原始的右束支传导阻滞导联II信号的注意力映射对比图;
图9为本发明方法和未集成导联注意力机制模块时对原始的房性早搏导联II信号的注意力映射对比图;
图10为本发明方法和未集成导联注意力机制模块时对原始的室性早搏导联II信号的注意力映射对比图;
图11为本发明方法和未集成导联注意力机制模块时对原始的ST段压低导联II信号的注意力映射对比图;
图12为本发明方法和未集成导联注意力机制模块时对原始的ST段抬高导联II信号的注意力映射对比图。
具体实施方式
本实施例中,一种基于导联注意力机制的多类心律失常检测方法,命名为LTA-CNN,如图1所示,其包括如下步骤:
步骤一、获取原始ECG信号及其对应的标签,对原始ECG信号进行下采样,得到采样后的ECG信号;对采样后的ECG信号进行裁剪或均值补齐处理,得到预处理后的ECG信号;
步骤1.1、从中国生理信号挑战赛(the China Physiological SignalChallenge,CPSC)2018的公开数据获取实验所需的原始ECG信号及其对应的标签。CPSC2018的公开数据提供6877条12导联ECG记录,时间长度范围为6秒到60秒。记录收集自11家医院,采样率500Hz。ECG记录包含九种心律类型,八种心律失常和正常心律,如表1所示。
表1:数据集的配置
Figure BDA0002287908880000051
从以上数据集中随机划分出1/5作为测试集,用来测试算法性能。
步骤1.2、将原始ECG信号从500Hz下采样到256Hz,得到采样后的ECG信号。
步骤1.3、将60s作为采样后的ECG信号的固定长度,不足60s的信号补其均值,超过60s的信号裁剪其前60s,得到预处理后的ECG信号。
步骤二、构建由导联注意力机制模块、卷积神经网络、双向门控循环单元和时间注意力机制模块组成的分类网络,如图2所示;
步骤2.1构建导联注意力机制模块,如图3所示,包括两个并行的全局池化层、c个全连接层、sigmoid函数层:
将预处理后的ECG信号分别输入全局平均池化层和全局最大池化层,得到全局平均池化特征图和全局最大池化特征图,再分别经过c个全连接层的处理后,得到两个输出特征图并相加,从而得到融合的全局池化特征图,并经过Sigmoid函数层后,输出导联注意力图;将导联注意力图与预处理后的ECG信号相乘,最终得到沿导联轴加权后的ECG信号;
示例性地,可以设置:c=2,两个全连接层的输出尺寸分别设置为6、12。
步骤2.2、构建卷积神经网络:
卷积神经网络由n个卷积层组成,并以m个卷积层作为一个卷积块,设置每个卷积块前m-1个卷积层的核尺寸小于第m个卷积层的核尺寸;并设置卷积层的核数量均为K;
在每个卷积层之后设置有LeakyReLU激活函数层;
将沿导联轴加权后的ECG信号输入卷积神经网络中,得到ECG信号的局部特征;
本实施例中,可以设置:n=15,m=3,K=12;这种参数设置下共有五个卷积块,每个卷积块的前两层卷积核尺寸是3,步长为1,小卷积核用来提取心电信号的细节信息;前四个卷积块的第三层核尺寸是24、第五个卷积块的第三层核尺寸是48,步长均为2,大卷积核用来提取心电信号更全局的信息。
步骤2.3、将ECG信号的局部特征输入双向门控循环单元,得到ECG信号的全局特征;
本实施例中,可以设置:双向门控循环单元的输出维度为12。
步骤2.4、将ECG信号的全局特征输入时间注意力机制模块,得到沿时间轴加权后的ECG信号;
参考”Hierarchical Attention Networks for Document Classification”,引进时间注意力机制,实现步骤如式(1)-式(3)所示:
uit=tanh(Wwhit+bw) (1)
Figure BDA0002287908880000071
si=∑tαithit (3)
首先将双向门控循环单元输出的状态hit输入到一个单层感知机,得到hit的隐藏表示uit。然后通过Softmax函数计算出每个时间步的重要性权重αit,最后通过对hit进行加权平均得到每个信号的表示。
步骤三、采用四折交叉验证法训练分类网络:
步骤3.1、定义外循环的最大循环次数为K=4;定义内迭代的最大迭代次数为Emax=500;
定义当前外循环次数为k,定义当前内迭代次数为e;
将预处理后的ECG信号及其对应的标签划分成同等大小的K个子集;
步骤3.2、初始化k=1;
步骤3.3、将第k个子集作为验证集,其余K-1个子集作为第k次循环的训练集;
步骤3.4、将第k次循环的训练集以64个信号为一批的方式输入到分类网络中;
步骤3.5、设置交叉熵作为循环的损失函数;
步骤3.6、利用He正态分布初始化器第k次初始化导联注意力机制模块的c个全连接层;利用Xaiver均匀分布初始化器第k次初始化卷积神经网络中n个卷积层的权重;利用生成随机正交矩阵的初始化器第k次初始化双向门控循环单元的元胞;
步骤3.7、利用Adam优化器优化分类模型中各个参数,设置学习率为l=0.001;
步骤3.8、在每个卷积块、双向门控循环单元和时间注意力机制模块的输出后分别设置Dropout层,丢弃率设置为d=0.2;
步骤3.9、初始化e=1;定义性能阈值为Fmax,并初始化Fmax=0;定义计数器count,并初始化count=0,阈值设置为100;
步骤3.10、在第k次外循环的第e次内迭代下利用Adam优化器对分类模型进行训练,得到第e次训练后的分类模型;
步骤3.11、计算第e次训练后的分类模型在验证集上的准确率Fe;
步骤3.12、判断Fe>Fmax是否成立,若成立,则将Fe赋值给Fmax,否则,令计数器count加“1”,并保持Fmax不变;
步骤3.13、判断e+1赋值给e后,e>Emax或count达到所设定阈值是否成立,若成立,则将Fmax所对应的分类模型作为第k次外循环下的分类模型,并执行步骤3.14;否则,返回步骤3.10;
步骤3.14、将k+1赋值给k后,判断k>K是否成立,若成立,则表示得到K个分类模型,并执行步骤四,否则,返回步骤3.3;
步骤四、利用K个分类模型对预处理后的待预测ECG信号分别进行预测,得到各个类别的K个预测概率,并对K个预测概率取均值,从而得到各类的平均预测概率,将平均预测概率最大的类别作为待预测的ECG信号预测标签。
步骤五、为了说明本发明中导联注意力机制的性能,在有无导联注意力模块的情况下分别测试了模型性能。为了公平比较,设置相同的超参数,相同的预处理方法和训练策略等也被使用,如上述具体实施方式所描述。本发明检测每类心律失常的性能由F1值来评估,如式(4)-式(6)定义:
Figure BDA0002287908880000081
Figure BDA0002287908880000082
Figure BDA0002287908880000083
式(4)-式(6)中,TP是某类分类正确的样本数,FN是属于某类却被误分做其它类的样本数,而FP是属于其它类却被被错分到某类的样本数。模型最终的性能由F1值的平均值来评估。
步骤5.1、表2比较了集成导联注意力模块的模型LTA-CNN和没有集成该模块的模型TA-CNN在检测九类心律类型上的F1得分。结果表明LTA-CNN在所有类的F1得分均优于TA-CNN,平均F1得分分别为0.8634、0.8301,导联注意力模块提升了3.33%的分类性能。在检测PAC、PVC和STD上,比起TA-CNN,LTA-CNN有非常明显的性能提高,分别提升了5.56%、5.09%和4.1%。
表2.模型有无集成导联注意力模块时的分类性能
Figure BDA0002287908880000084
Figure BDA0002287908880000091
步骤5.2、Grad-CAM被用来可视化解释TA-CNN和LTA-CNN学习到的特征,进一步证明导联注意力机制帮助模型更好地定位心电信号中的异常,也改善了深度学习模型的可解释性。根据输出特征,对分类正确的九类心律进行注意力映射,如图4-图12所示,颜色越亮表示对应的心电信号片段越被注意,其中:
图4中的(a)部分为原始的正常导联II信号;(b)部分为分类正确时,本发明方法的注意力映射;(c)部分为分类正确时,本发明方法未集成导联注意力机制模块时的注意力映射;
图5中的(a)部分为原始的房颤导联II信号;(b)部分为分类正确时,本发明方法的注意力映射;(c)部分为分类正确时,本发明方法未集成导联注意力机制模块时的注意力映射;
图6中的(a)部分为原始的一度房室传导阻滞导联II信号;(b)部分为分类正确时,本发明方法的注意力映射;(c)部分为分类正确时,本发明方法未集成导联注意力机制模块时的注意力映射;
图7中的(a)部分为原始的左束支传导阻滞导联II信号;(b)部分为分类正确时,本发明方法的注意力映射;(c)部分为分类正确时,本发明方法未集成导联注意力机制模块时的注意力映射;
图8中的(a)部分为原始的右束支传导阻滞导联II信号;(b)部分为分类正确时,本发明方法的注意力映射;(c)部分为分类正确时,本发明方法未集成导联注意力机制模块时的注意力映射;
图9中的(a)部分为原始的房性早搏导联II信号;(b)部分为分类正确时,本发明方法的注意力映射;(c)部分为分类正确时,本发明方法未集成导联注意力机制模块时的注意力映射;
图10中的(a)部分为原始的室性早搏导联II信号;(b)部分为分类正确时,本发明方法的注意力映射;(c部分)为分类正确时,本发明方法未集成导联注意力机制模块时的注意力映射;
图11中的(a)部分为原始的ST段压低导联II信号;(b)部分为分类正确时,本发明方法的注意力映射;(c)部分为分类正确时,本发明方法未集成导联注意力机制模块时的注意力映射;
图12中的(a)部分为原始的ST段抬高导联II信号;(b)部分为分类正确时,本发明方法的注意力映射;(c)部分为分类正确时,本发明方法未集成导联注意力机制模块时的注意力映射;
从图4-图12,可以看出:
(1)如图4中的(b)部分所示,模型对正常心律的注意力集中在P波和T波;
(2)房颤发生时,P波消失,代之以连续不规则的基线波动(f波),如图5中的(b)所示,房颤的注意力集中在P波;
(3)如图6中的(b)部分-图8中的(b)部分,对于三种传导阻滞(1-AVB,LBBB和RBBB),模型的注意力集中在异常的QRS波;
(4)出现房性早搏时,P′-QRS-T波提前出现,P′形态与窦性P波略有不同,或P′波后无QRS波群,如图9中的(b)部分所示,房性早搏的注意力集中在P波和QRS波上;
(5)室性早搏的T波与主波方向相反,如图10中的(b)所示,室性早搏的注意力集中在T波;
(6)如图11中的(b)部分和12中的(b)部分所示,对于ST段改变的两种心律情况(STD和STE),模型的注意力集中在ST段。
对比图4-图12中的(b)部分和(c)部分可以看出,集成导联注意力模块的LTA-CNN比未集成导联注意力模块的TA-CNN更有效地定位了心电信号中的异常,证明了导联注意力机制的有效性。从LTA-CNN的注意力映射可以看出,模型提取的特征与临床判断相符,本发明对大多数心律失常具有高检测性能。

Claims (2)

1.一种基于导联注意力机制的多类心律失常检测方法,其特征在于,包括以下步骤:
步骤一、获取原始ECG信号及其对应的标签,对所述原始ECG信号进行下采样,得到采样后的ECG信号;对所述采样后的ECG信号进行裁剪或均值补齐处理,得到预处理后的ECG信号;
步骤二、构建由导联注意力机制模块、卷积神经网络、双向门控循环单元和时间注意力机制模块组成的分类网络;
步骤2.1、构建导联注意力机制模块,包括两个并行的全局池化层、c个全连接层、sigmoid函数层:
将所述预处理后的ECG信号分别输入全局平均池化层和全局最大池化层,得到全局平均池化特征图和全局最大池化特征图,再分别经过所述c个全连接层的处理后,得到两个输出特征图并相加,从而得到融合的全局池化特征图,并经过Sigmoid函数层后,输出导联注意力图;将所述导联注意力图与预处理后的ECG信号相乘,最终得到沿导联轴加权后的ECG信号;
步骤2.2、构建卷积神经网络:
所述卷积神经网络由n个卷积层组成,并以m个卷积层作为一个卷积块,设置每个卷积块前m-1个卷积层的核尺寸小于第m个卷积层的核尺寸;并设置卷积层的核数量均为K;
在每个卷积层之后设置有LeakyReLU激活函数层;
将所述沿导联轴加权后的ECG信号输入所述卷积神经网络中,得到ECG信号的局部特征;
步骤2.3、将所述ECG信号的局部特征输入所述双向门控循环单元,得到ECG信号的全局特征;
步骤2.4、将所述ECG信号的全局特征输入所述时间注意力机制模块,得到沿时间轴加权后的ECG信号;
步骤三、采用K折交叉验证法训练所述分类网络,得到K个分类模型;
步骤四、利用K个分类模型对预处理后的待预测ECG信号分别进行预测,得到各个类别的K个预测概率,并对K个预测概率取均值,从而得到各类的平均预测概率,将平均预测概率最大的类别作为待预测的ECG信号预测标签。
2.根据权利要求1所述的多类心律失常检测方法,其特征是,所述步骤三是按如下过程进行:
步骤3.1、定义外循环的最大循环次数为K;定义内迭代的最大迭代次数为Emax
定义当前外循环次数为k,定义当前内迭代次数为e;
将预处理后的ECG信号及其对应的标签划分成同等大小的K个子集;
步骤3.2、初始化k=1;
步骤3.3、将第k个子集作为验证集,其余K-1个子集作为第k次循环的训练集;
步骤3.4、将所述第k次循环的训练集以分批的方式输入到所述分类网络中;
步骤3.5、设置交叉熵作为循环的损失函数;
步骤3.6、利用He正态分布初始化器第k次初始化所述导联注意力机制模块的c个全连接层;利用Xaiver均匀分布初始化器第k次初始化所述卷积神经网络中n个卷积层的权重;利用生成随机正交矩阵的初始化器第k次初始化所述双向门控循环单元的元胞;
步骤3.7、利用Adam优化器优化所述分类模型中各个参数,设置学习率为l;
步骤3.8、在每个卷积块、所述双向门控循环单元和时间注意力机制模块的输出后分别设置Dropout层,丢弃率设置为d;
步骤3.9、初始化e=1;定义性能阈值为Fmax,并初始化Fmax=0;定义计数器count,并初始化count=0;
步骤3.10、在第k次外循环的第e次内迭代下利用Adam优化器对所述分类模型进行训练,得到第e次训练后的分类模型;
步骤3.11、计算第e次训练后的分类模型在验证集上的准确率Fe;
步骤3.12、判断Fe>Fmax是否成立,若成立,则将Fe赋值给Fmax,否则,令计数器count加“1”,并保持Fmax不变;
步骤3.13、判断e+1赋值给e后,e>Emax或count达到所设定阈值是否成立,若成立,则将Fmax所对应的分类模型作为第k次外循环下的分类模型,并执行步骤3.14;否则,返回步骤3.10;
步骤3.14、将k+1赋值给k后,判断k>K是否成立,若成立,则表示得到K个分类模型,并执行步骤四,否则,返回步骤3.3。
CN201911167791.6A 2019-11-25 2019-11-25 一种基于导联注意力机制的多类心律失常检测方法 Active CN110890155B (zh)

Priority Applications (1)

Application Number Priority Date Filing Date Title
CN201911167791.6A CN110890155B (zh) 2019-11-25 2019-11-25 一种基于导联注意力机制的多类心律失常检测方法

Applications Claiming Priority (1)

Application Number Priority Date Filing Date Title
CN201911167791.6A CN110890155B (zh) 2019-11-25 2019-11-25 一种基于导联注意力机制的多类心律失常检测方法

Publications (2)

Publication Number Publication Date
CN110890155A true CN110890155A (zh) 2020-03-17
CN110890155B CN110890155B (zh) 2022-10-28

Family

ID=69748701

Family Applications (1)

Application Number Title Priority Date Filing Date
CN201911167791.6A Active CN110890155B (zh) 2019-11-25 2019-11-25 一种基于导联注意力机制的多类心律失常检测方法

Country Status (1)

Country Link
CN (1) CN110890155B (zh)

Cited By (20)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN111631688A (zh) * 2020-06-24 2020-09-08 电子科技大学 睡眠自动分期的算法
CN111657926A (zh) * 2020-07-08 2020-09-15 中国科学技术大学 一种基于多导联信息融合的心律失常分类方法
CN111803059A (zh) * 2020-06-30 2020-10-23 武汉中旗生物医疗电子有限公司 一种基于时域卷积网络的心电信号分类方法及装置
CN111920405A (zh) * 2020-09-15 2020-11-13 齐鲁工业大学 一种房颤信号识别系统及方法
CN112022125A (zh) * 2020-09-28 2020-12-04 无锡博智芯科技有限公司 一种基于CNN-BiGRU模型和PPG的智能血压预测方法
CN112635047A (zh) * 2020-09-22 2021-04-09 广东工业大学 一种鲁棒的心电r峰检测方法
CN112957052A (zh) * 2021-01-25 2021-06-15 北京工业大学 一种基于nlf-cnn导联融合深度网络的多导联心电信号分类方法
CN113012807A (zh) * 2021-03-04 2021-06-22 武汉中旗生物医疗电子有限公司 一种ecg信号标记方法、系统、装置及计算机可读存储介质
CN113080988A (zh) * 2021-03-26 2021-07-09 京东方科技集团股份有限公司 基于注意力机制的12导联心电图整体分类方法和装置
CN113095302A (zh) * 2021-05-21 2021-07-09 中国人民解放军总医院 用于心律失常分类的深度模型、利用该模型的方法及装置
CN113080994A (zh) * 2021-03-30 2021-07-09 北京芯动卫士科技有限公司 一种基于卷积神经网络的多导联心电信号分类方法
CN113177482A (zh) * 2021-04-30 2021-07-27 中国科学技术大学 一种基于最小类别混淆的跨个体脑电信号分类方法
CN113288163A (zh) * 2021-06-04 2021-08-24 浙江理工大学 基于注意力机制的多特征融合心电信号分类模型建模方法
CN113598787A (zh) * 2021-09-07 2021-11-05 四川大学 一种基于多视图进行心电信号重点区域检测的方法
CN113768514A (zh) * 2021-08-09 2021-12-10 西安理工大学 基于卷积神经网络与门控循环单元的心律失常分类方法
CN114052741A (zh) * 2021-11-08 2022-02-18 西安工业大学 基于多尺度自回归模型的心电图识别方法
CN114190897A (zh) * 2021-12-15 2022-03-18 中国科学院空天信息创新研究院 睡眠分期模型的训练方法、睡眠分期方法及装置
CN114224355A (zh) * 2020-09-07 2022-03-25 中国移动通信有限公司研究院 心电信号分类训练方法、分类方法、装置及存储介质
WO2022073374A1 (zh) * 2020-10-10 2022-04-14 深圳市凯沃尔电子有限公司 基于多导联心电信号生成心搏标签数据序列的方法和装置
WO2023114742A1 (en) * 2021-12-13 2023-06-22 Irhythm Technologies, Inc. Non-invasive cardiac monitor and methods of inferring or predicting a physiological characteristic of a patient

Citations (7)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
US20180249964A1 (en) * 2017-03-02 2018-09-06 Logos Care, Inc. Deep learning algorithms for heartbeats detection
CN108720831A (zh) * 2018-05-12 2018-11-02 鲁东大学 一种基于导联深度神经网络的自动心律失常分析方法
CN108766557A (zh) * 2018-05-12 2018-11-06 鲁东大学 基于通道信号融合神经网络的自动心律失常分析方法
CN109620205A (zh) * 2018-12-26 2019-04-16 上海联影智能医疗科技有限公司 心电数据分类方法、装置、计算机设备和存储介质
CN109620210A (zh) * 2019-01-28 2019-04-16 山东科技大学 一种基于自编码模式的cnn与gru结合的心电信号分类方法
CN109711413A (zh) * 2018-12-30 2019-05-03 陕西师范大学 基于深度学习的图像语义分割方法
CN110251120A (zh) * 2019-06-06 2019-09-20 山东省计算中心(国家超级计算济南中心) 基于注意力机制的深度残差神经网络的心电图分类方法

Patent Citations (7)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
US20180249964A1 (en) * 2017-03-02 2018-09-06 Logos Care, Inc. Deep learning algorithms for heartbeats detection
CN108720831A (zh) * 2018-05-12 2018-11-02 鲁东大学 一种基于导联深度神经网络的自动心律失常分析方法
CN108766557A (zh) * 2018-05-12 2018-11-06 鲁东大学 基于通道信号融合神经网络的自动心律失常分析方法
CN109620205A (zh) * 2018-12-26 2019-04-16 上海联影智能医疗科技有限公司 心电数据分类方法、装置、计算机设备和存储介质
CN109711413A (zh) * 2018-12-30 2019-05-03 陕西师范大学 基于深度学习的图像语义分割方法
CN109620210A (zh) * 2019-01-28 2019-04-16 山东科技大学 一种基于自编码模式的cnn与gru结合的心电信号分类方法
CN110251120A (zh) * 2019-06-06 2019-09-20 山东省计算中心(国家超级计算济南中心) 基于注意力机制的深度残差神经网络的心电图分类方法

Non-Patent Citations (4)

* Cited by examiner, † Cited by third party
Title
PATRICK SCHWAB: "Beat by beat: Classifying cardiac arrhythmias with recurrent neural networks", 《2017 COMPUTING IN CARDIOLOGY (CINC)》 *
QIHANG YAO ET AL.: "Multi-class Arrhythmia detection from 12-lead varied-length ECG using Attention-based Time-Incremental Convolutional Neural Network", 《INFORMATION FUSION》 *
吴波: "基于心电信号的身份识别与心律失常识别方法研究", 《中国优秀硕士学位论文全文数据库 医药卫生科技辑》 *
张坤等: "基于深度学习的心律失常检测算法研究", 《医疗卫生装备》 *

Cited By (28)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN111631688A (zh) * 2020-06-24 2020-09-08 电子科技大学 睡眠自动分期的算法
CN111631688B (zh) * 2020-06-24 2021-10-29 电子科技大学 睡眠自动分期的算法
CN111803059A (zh) * 2020-06-30 2020-10-23 武汉中旗生物医疗电子有限公司 一种基于时域卷积网络的心电信号分类方法及装置
CN111657926A (zh) * 2020-07-08 2020-09-15 中国科学技术大学 一种基于多导联信息融合的心律失常分类方法
CN111657926B (zh) * 2020-07-08 2021-04-23 中国科学技术大学 一种基于多导联信息融合的心律失常分类方法
CN114224355A (zh) * 2020-09-07 2022-03-25 中国移动通信有限公司研究院 心电信号分类训练方法、分类方法、装置及存储介质
CN111920405A (zh) * 2020-09-15 2020-11-13 齐鲁工业大学 一种房颤信号识别系统及方法
CN112635047B (zh) * 2020-09-22 2021-09-07 广东工业大学 一种鲁棒的心电r峰检测方法
CN112635047A (zh) * 2020-09-22 2021-04-09 广东工业大学 一种鲁棒的心电r峰检测方法
CN112022125A (zh) * 2020-09-28 2020-12-04 无锡博智芯科技有限公司 一种基于CNN-BiGRU模型和PPG的智能血压预测方法
WO2022073374A1 (zh) * 2020-10-10 2022-04-14 深圳市凯沃尔电子有限公司 基于多导联心电信号生成心搏标签数据序列的方法和装置
CN112957052B (zh) * 2021-01-25 2023-06-23 北京工业大学 一种基于nlf-cnn导联融合深度网络的多导联心电信号分类方法
CN112957052A (zh) * 2021-01-25 2021-06-15 北京工业大学 一种基于nlf-cnn导联融合深度网络的多导联心电信号分类方法
CN113012807A (zh) * 2021-03-04 2021-06-22 武汉中旗生物医疗电子有限公司 一种ecg信号标记方法、系统、装置及计算机可读存储介质
CN113080988B (zh) * 2021-03-26 2024-01-16 京东方科技集团股份有限公司 基于注意力机制的12导联心电图整体分类方法和装置
CN113080988A (zh) * 2021-03-26 2021-07-09 京东方科技集团股份有限公司 基于注意力机制的12导联心电图整体分类方法和装置
CN113080994A (zh) * 2021-03-30 2021-07-09 北京芯动卫士科技有限公司 一种基于卷积神经网络的多导联心电信号分类方法
CN113177482A (zh) * 2021-04-30 2021-07-27 中国科学技术大学 一种基于最小类别混淆的跨个体脑电信号分类方法
CN113095302A (zh) * 2021-05-21 2021-07-09 中国人民解放军总医院 用于心律失常分类的深度模型、利用该模型的方法及装置
CN113288163A (zh) * 2021-06-04 2021-08-24 浙江理工大学 基于注意力机制的多特征融合心电信号分类模型建模方法
CN113768514B (zh) * 2021-08-09 2024-03-22 西安理工大学 基于卷积神经网络与门控循环单元的心律失常分类方法
CN113768514A (zh) * 2021-08-09 2021-12-10 西安理工大学 基于卷积神经网络与门控循环单元的心律失常分类方法
CN113598787B (zh) * 2021-09-07 2022-06-17 四川大学 一种基于多视图进行心电信号重点区域检测的方法
CN113598787A (zh) * 2021-09-07 2021-11-05 四川大学 一种基于多视图进行心电信号重点区域检测的方法
CN114052741A (zh) * 2021-11-08 2022-02-18 西安工业大学 基于多尺度自回归模型的心电图识别方法
WO2023114742A1 (en) * 2021-12-13 2023-06-22 Irhythm Technologies, Inc. Non-invasive cardiac monitor and methods of inferring or predicting a physiological characteristic of a patient
CN114190897A (zh) * 2021-12-15 2022-03-18 中国科学院空天信息创新研究院 睡眠分期模型的训练方法、睡眠分期方法及装置
CN114190897B (zh) * 2021-12-15 2024-04-05 中国科学院空天信息创新研究院 睡眠分期模型的训练方法、睡眠分期方法及装置

Also Published As

Publication number Publication date
CN110890155B (zh) 2022-10-28

Similar Documents

Publication Publication Date Title
CN110890155B (zh) 一种基于导联注意力机制的多类心律失常检测方法
Wang et al. Deep multi-scale fusion neural network for multi-class arrhythmia detection
CN111657926B (zh) 一种基于多导联信息融合的心律失常分类方法
CN111990989A (zh) 一种基于生成对抗及卷积循环网络的心电信号识别方法
CN110974214A (zh) 一种基于深度学习的自动心电图分类方法、系统及设备
CN110638430B (zh) 级联神经网络ecg信号心律失常分类模型的搭建方法
Emanet ECG beat classification by using discrete wavelet transform and Random Forest algorithm
CN111329469A (zh) 一种心律异常预测方法
Pandey et al. Automatic arrhythmia recognition from electrocardiogram signals using different feature methods with long short-term memory network model
Sugimoto et al. Detection and localization of myocardial infarction based on a convolutional autoencoder
CN112932433B (zh) 一种心电图的心律失常自动分类方法
CN112906748A (zh) 基于残差网络的12导联ecg心律失常检测分类模型构建方法
Wu et al. Personalizing a generic ECG heartbeat classification for arrhythmia detection: a deep learning approach
CN115470828A (zh) 基于卷积与自注意力机制的多导联心电图分类识别方法
Khan et al. Electrocardiogram heartbeat classification using convolutional neural networks for the detection of cardiac Arrhythmia
Wang et al. Atrial fibrillation detection and ECG classification based on CNN-BiLSTM
Dhyani et al. Analysis of ECG-based arrhythmia detection system using machine learning
Ozdemir et al. Abnormal ecg beat detection based on convolutional neural networks
Wang et al. Multiscale residual network based on channel spatial attention mechanism for multilabel ECG classification
Shahin et al. Adversarial multi-task learning for robust end-to-end ECG-based heartbeat classification
Sraitih et al. An overview on intra-and inter-patient paradigm for ECG Heartbeat Arrhythmia Classification
Tung et al. Multi-lead ECG classification via an information-based attention convolutional neural network
Ye et al. Explainable prediction of cardiac arrhythmia using machine learning
Pomprapa et al. Deep learning of arrhythmia analysis based on convolutional neural network
Singh et al. Classification and detection of heart rhythm irregularities using machine learning

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