买专利,只认龙图腾
首页 专利交易 科技果 科技人才 科技服务 商标交易 会员权益 IP管家助手 需求市场 关于龙图腾
 /  免费注册
到顶部 到底部
清空 搜索

【发明授权】一种基于迁移学习的心冲击信号房颤计算机辅助诊断方法_东北大学_201910401402.5 

申请/专利权人:东北大学

申请日:2019-05-15

公开(公告)日:2022-11-25

公开(公告)号:CN110148466B

主分类号:G16H50/20

分类号:G16H50/20;A61B5/33;A61B5/346;A61B5/11;G06N3/04;G06N3/08

优先权:

专利状态码:有效-授权

法律状态:2022.11.25#授权;2019.09.13#实质审查的生效;2019.08.20#公开

摘要:本发明提供一种基于迁移学习的心冲击信号房颤计算机辅助诊断方法,涉及计算机辅助诊断技术领域。该方法首先对同步实测的ECG信号、BCG信号及现房颤疾病数据库中的ECG信号进行预处理,并设定所有信号数据的分帧长度为24秒;然后构建进行房颤辅助检测的卷积神经网络;训练所构建的卷积神经网络并进参数更新;提取实测BCG信号的迁移学习特征向量并构建随机森林分类器;将测试集中BCG信号作为卷积神经网络的输入,获取迁移学习模型特征参数并输入到随机森林分类器,通过随机森林分类器的输出结果判断该BCG信号是否为房颤疾病类型。本发明方法,有利于扩展BCG信号的日常应用领域,且为BCG开展心脏相关疾病辅助诊断及预测提供可行方案。

主权项:1.一种基于迁移学习的心冲击信号房颤计算机辅助诊断方法,其特征在于:包括以下步骤:步骤1:对同步实测的ECG信号、BCG信号及现房颤疾病数据库中的ECG信号进行预处理,并针对房颤疾病的特点,设定所有信号数据的分帧长度为24秒,作为神经网络的输入向量;步骤1.1:采集同一受试者的同步ECG信号、BCG信号,并分别对其信号进行归一化处理,以获取实测数据;步骤1.2:从现有的房颤疾病数据库中获取房颤ECG信号,并根据实测ECG信号的采样频率,对数据库中获取的ECG信号进行重采样处理,使其与步骤1.1中的实测数据采样频率一致;步骤1.3:针对房颤疾病的特点,设定所有信号数据的分帧长度为24秒,作为神经网络的输入向量;步骤1.4:应用实测ECG信号作为诊断标准,将每一帧实测同步BCG信号标记为房颤和非房颤二类,其中,房颤标定为1,非房颤标定为0;步骤1.5:将70%的实测BCG数据作为训练集、20%的实测BCG数据作为验证集、10%的实测BCG数据作为测试集;步骤2:构建进行房颤辅助检测的卷积神经网络;所述卷积神经网络包括4层卷积层、6层池化层和1层Flatten层以及3层全连接层,具体设计方法为:步骤2.1:以分帧长度为24秒的数据库中获取的ECG信号作为输入向量,确定输入层的神经元数目;步骤2.2:构建两个特征提取模块和一个降维模块;其中,第一个特征提取模块包括两层卷积层和两层池化层,第二个特征提取模块包括四层卷积层与四层池化层;降维模块包括一层Flatten层及三层全连接层;步骤2.3:设计输出层为2个神经元,0-1组合判定为非房颤,1-0组合判定为房颤,0-0及1-1组合则需要进行二次判定;步骤3:训练所构建的卷积神经网络并进参数更新:应用从数据库中获取的房颤ECG数据对卷积神经网络进行逐层预训练并更新网络参数;然后固定第一层卷积层和第二层卷积层的网络参数,以实测BCG信号作为输入向量,建立新型交叉熵损失函数更新网络参数,获取迁移学习后的卷积神经网络模型,具体方法为:步骤3.1:应用从数据库中获取的ECG信号对所设计的卷积神经网络进行预训练并更新网络参数;步骤3.2:应用实测BCG信号训练集对预训练后的卷积神经网络进行监督二次训练,获取迁移学习后的卷积神经网络模型,即BCG信号迁移学习模型;步骤4:提取实测BCG信号的迁移学习特征向量并构建随机森林分类器:将迁移学习后的CNN全连接层的输出作为BCG信号迁移学习模型特征参数,设计并优化随机森林分类器,使其性能最优;步骤4.1:应用实测BCG信号验证集作为二次训练后CNN网络的输入,提取降维模块所生成信号的高维特征作为BCG信号迁移学习特征参数;步骤4.2:构建随机森林分类器,设定分类器的参数,并设定分类器输出结果为0或1,其中,0为非房颤,1为非房颤;步骤4.3:将BCG信号迁移学习特征参数作为随机森林分类器的输入,进行十折交叉验证,将随机森林分类器的输出结果转化为混淆矩阵,并对随机森林分类器参数进行二次调试,以获取最优特异性、敏感性、准确度和精度的性能参数;步骤5、将测试集中BCG信号作为卷积神经网络的输入,获取迁移学习模型特征参数并输入到随机森林分类器,通过随机森林分类器的输出结果判断该BCG信号是否为房颤疾病类型。

全文数据:一种基于迁移学习的心冲击信号房颤计算机辅助诊断方法技术领域本发明涉及计算机辅助诊断技术领域,尤其涉及一种基于迁移学习的心冲击信号房颤计算机辅助诊断方法。背景技术随着穿戴式设备的日益盛行,无创式心脏功能评估领域成为当今研究热点。常规的心脏功能检测手段,包括心电图Electrocardiogram,即ECG、心磁图、心音、心阻抗图等,均需在人体体表附着电极等检测设备,对监测环境、条件及操作人员具有一定的要求,且给受试者的日常生活造成极大的不便。尤其阵发性房颤具有发作时间不确定,发作突然且临床表现不明显等特点,因此迫切需要一种非接触式实时监测心脏功能的的方法。心冲击信号Ballistocardiogram,即BCG通过检测心脏搏动传导至体表的微弱振动,间接反映心脏的工作状态,是一种无感评估人体心脏动力学性能的先进检测手段。但BCG信号存在幅值微弱、易受干扰等特点且缺少较为完善的数据库,卷积神经网络ConvolutionalNeuralNetworks,CNN等深度学习常见模型,对输入数据量具有一定的要求,通常需要大批量的数据来进行网络的训练及调试,因此不适合应用于BCG信号的分析;传统的机器学习方法虽适用于小批量数据的训练及调试,但其通常需要提取信号的波形特征,而BCG信号的波形随检测设备的不同有着较大的差异,具有较高的时间复杂度,因此机器学习方法不适合应用于日常BCG信号的处理。综上,需要一种适用于BCG信号日常监护的房颤疾病计算机辅助诊断方法,以满足心血管疾病家庭非接触检测的需求。发明内容本发明要解决的技术问题是针对上述现有技术的不足,提供一种基于迁移学习的心冲击信号房颤计算机辅助诊断方法,实现对心冲击信号阵发性房颤的辅助检测。为解决上述技术问题,本发明所采取的技术方案是:一种基于迁移学习的心冲击信号房颤计算机辅助诊断方法,包括以下步骤:步骤1:对同步实测的ECG信号、BCG信号及现有的房颤疾病数据库中的ECG信号进行预处理,并针对房颤疾病的特点,设定所有信号数据的分帧长度为24秒,作为神经网络的输入向量;步骤1.1:采集同一受试者的同步ECG信号、BCG信号,并分别对其信号进行归一化处理,以获取实测数据;步骤1.2:从现有的房颤疾病数据库中获取房颤ECG信号,并根据实测ECG信号的采样频率,对数据库中获取的ECG信号进行重采样处理,使其与步骤1.1中的实测数据采样频率一致;步骤1.3:针对房颤疾病的特点,设定所有信号数据的分帧长度为24秒,作为神经网络的输入向量;步骤1.4:应用实测ECG信号作为诊断标准,将每一帧实测同步BCG信号标记为房颤和非房颤二类,其中,房颤标定为1,非房颤标定为0;步骤1.5:将70%的实测BCG数据作为训练集、20%的实测BCG数据作为验证集、10%的实测BCG数据作为测试集;步骤2:构建进行房颤辅助检测的卷积神经网络;所述卷积神经网络包括4层卷积层、6层池化层和1层Flatten层以及3层全连接层,具体设计方法为:步骤2.1:以分帧长度为24秒的数据库中获取的ECG信号作为输入向量,确定输入层的神经元数目;步骤2.2:构建两个特征提取模块和一个降维模块;其中,第一个特征提取模块包括两层卷积层和两层池化层,第二个特征提取模块包括四层卷积层与四层池化层;降维模块包括一层Flatten层及三层全连接层;步骤2.3:设计输出层为2个神经元,0-1组合判定为非房颤,1-0组合判定为房颤,0-0及1-1组合则需要进行二次判定;步骤3:训练所构建的卷积神经网络并进参数更新:应用从数据库中获取的房颤ECG数据对卷积神经网络进行逐层预训练并更新网络参数;然后固定第一层卷积层和第二层卷积层的网络参数,以实测BCG信号作为输入向量,建立新型交叉熵损失函数更新网络参数,获取迁移学习后的卷积神经网络模型,具体方法为:步骤3.1:应用从数据库中获取的ECG信号对所设计的卷积神经网络进行预训练并更新网络参数;步骤3.2:应用实测BCG信号训练集对预训练后的卷积神经网络进行监督二次训练,获取迁移学习后的卷积神经网络模型,即BCG信号迁移学习模型;首先固定预训练后的第一层卷积层和第二层卷积层参数,然后以损失函数不再减小为基准,获取最优化网络参数;其中:卷积层与池化层的激活函数选用Relu函数,全连接层选用softmax函数,损失函数针对CNN输出层的特点,设计为采用如下公式:其中,C为新型交叉熵损失函数,y为期望的输出向量,x为神经元个数:非房颤为0-1,房颤为1-0;a为CNN输出层的实际输出向量;步骤4:提取实测BCG信号的迁移学习特征向量并构建随机森林分类器:将迁移学习后的CNN全连接层的输出作为BCG信号迁移学习模型特征参数,设计并优化随机森林分类器,使其性能最优;步骤4.1:应用实测BCG信号验证集作为二次训练后CNN网络的输入,提取降维模块所生成信号的高维特征作为BCG信号迁移学习特征参数;步骤4.2:构建随机森林分类器,设定分类器的参数,并设定分类器输出结果为0或1,其中,0为非房颤,1为非房颤;步骤4.3:将BCG信号迁移学习特征参数作为随机森林分类器的输入,进行十折交叉验证,将随机森林分类器的输出结果转化为混淆矩阵,并对随机森林分类器参数进行二次调试,以获取最优特异性、敏感性、准确度和精度的性能参数;步骤5、将测试集中BCG信号作为卷积神经网络的输入,获取迁移学习模型特征参数并输入到随机森林分类器,通过随机森林分类器的输出结果判断该BCG信号是否为房颤疾病类型。采用上述技术方案所产生的有益效果在于:本发明提供的一种基于迁移学习的心冲击信号房颤计算机辅助诊断方法,应用无感检测的BCG信号辅助诊断房颤疾病,利用ECG与BCG的同源性,将采用大数据量ECG数据库信号预训练后的CNN迁移至小数据量的BCG实测信号中,进而进行房颤及非房颤的判断。该发明有利于扩展BCG信号的日常应用领域,且为BCG开展心脏相关疾病辅助诊断及预测提供可行方案。附图说明图1为本发明实施例提供的一种基于迁移学习的心冲击信号房颤计算机辅助诊断方法的流程图;图2为本发明实施例提供的实测ECG信号图,其中,a为实测房颤ECG信号图,b实测非房颤ECG信号图;图3为本发明实施例提供的实测BCG信号图,其中,a为实测房颤BCG信号图,b实测非房颤BCG信号图;图4为本发明实施例提供的CNN结构图。图中,1、卷积层;2、池化层;3、Flatten层;4、全连接层;5、输出。具体实施方式下面结合附图和实施例,对本发明的具体实施方式作进一步详细描述。以下实施例用于说明本发明,但不用来限制本发明的范围。本实施例中,一种基于迁移学习的心冲击信号房颤计算机辅助诊断方法,如图1所示,包括以下步骤:步骤1:对同步实测的ECG信号、BCG信号及现有的房颤疾病数据库中的ECG信号进行预处理,并针对房颤疾病的特点,设定所有信号数据的分帧长度为24秒,作为神经网络的输入向量;步骤1.1:采集同一受试者的同步ECG信号、BCG信号,并分别对其信号进行归一化处理,以获取实测数据;步骤1.2:从现有的房颤疾病数据库中获取房颤ECG信号,并根据实测ECG信号的采样频率,对数据库中获取的ECG信号进行重采样处理,使其与步骤1.1中的实测数据采样频率一致;本实施例中,房颤疾病数据库使用PhysioNetComputinginCardiologyChallenge2017数据。步骤1.3:针对房颤疾病的特点,设定所有信号数据的分帧长度为24秒,作为神经网络的输入向量;步骤1.4:应用实测ECG信号作为诊断标准,将每一帧实测同步BCG信号标记为房颤和非房颤二类,其中,房颤标定为1,非房颤标定为0;步骤1.5:将70%的实测BCG数据作为训练集、20%的实测BCG数据作为验证集、10%的实测BCG数据作为测试集;本实施例实际采集的同步BCG信号、ECG信号如图2和3所示,均来自于实验室搭建的多生理信号采集座椅,其中BCG信号应用薄膜压力传感器采集,ECG信号采用三导联心电电极采集。步骤2:构建进行房颤辅助检测的卷积神经网络;所述卷积神经网络包括4层卷积层、6层池化层和1层Flatten层以及3层全连接层,具体设计方法为:步骤2.1:以分帧长度为24秒的数据库中获取的ECG信号作为输入向量,确定输入层的神经元数目;步骤2.2:构建两个特征提取模块和一个降维模块;其中,第一个特征提取模块包括两层卷积层和两层池化层,第二个特征提取模块包括四层卷积层与四层池化层;降维模块包括一层Flatten层及三层全连接层;步骤2.3:设计输出层为2个神经元,0-1组合判定为非房颤,1-0组合判定为房颤,0-0及1-1组合则需要进行二次判定,该设计主要针对BCG信号易受干扰的特点,以提高分类判断的精度;本实施例中,卷积神经网络由三个模块构成,如图4所示,其中模块1和模块2为特征提取模块,模块3为降维模块。模块1由两层卷积层与两层池化层组成,两层卷积层的卷积核数目分别为32和64,卷积核大小均为9×1,两层池化层的卷积核数目均为3,卷积核大小均为3×1。模块2由四层卷积层与两层池化层组成,前两层卷积层卷积核数目均为128、后两层卷积层卷积核数目均为256,四层卷积层的卷积核大小均为9×1,两层池化层的卷积核数目均为3,卷积核大小均为3×1;模块3由1层Flatten层以及3层全连接层组成,3层全连接层神经元数目分别为256、128、9,对应9×1的迁移学习模型特征;卷积神经网络的各层参数具体如表1所示;表1卷积神经网络各层参数层数卷积核数目卷积核大小步长1卷积层329×112池化层33×113卷积层649×114池化层33×115卷积层1289×116卷积层1289×117池化层33×118卷积层2569×119卷积层2569×1110池化层33×11全连接层256神经元数全连接层128神经元数全连接层9神经元数步骤3:训练所构建的卷积神经网络并进参数更新:应用从数据库中获取的房颤ECG数据对卷积神经网络进行逐层预训练并更新网络参数;然后固定第一层卷积层和第二层卷积层的网络参数,以实测BCG信号作为输入向量,建立新型交叉熵损失函数更新网络参数,获取迁移学习后的卷积神经网络模型,具体方法为:步骤3.1:应用大批量从数据库中获取的ECG信号对所设计的卷积神经网络进行预训练并更新网络参数;本实施例中,使用12000帧从数据库中获取的ECG信号对所设计的卷积神经网络进行预训练;步骤3.2:应用小批量实测BCG信号训练集对预训练后的卷积神经网络进行监督二次训练,获取迁移学习后的卷积神经网络模型,即BCG信号迁移学习模型;本实施例中,使用2000帧实测BCG信号训练集对预训练后的卷积神经网络进行监督二次训练;首先固定预训练后的第一层卷积层和第二层卷积层参数,然后以损失函数不再减小为基准,获取最优化网络参数;其中:卷积层与池化层的激活函数选用Relu函数,全连接层选用softmax函数,损失函数针对CNN输出层的特点,设计为采用如下公式:其中,C为新型交叉熵损失函数,y为期望的输出向量,x为神经元个数:非房颤为0-1,房颤为1-0;a为CNN输出层的实际输出向量;步骤4:提取实测BCG信号的迁移学习特征向量并构建随机森林分类器:将迁移学习后的CNN全连接层的输出作为BCG信号迁移学习模型特征参数,设计并优化随机森林分类器,使其性能最优;步骤4.1:应用实测BCG信号验证集作为二次训练后CNN网络的输入,提取降维模块所生成信号的高维特征作为BCG信号迁移学习特征参数;步骤4.2:构建随机森林分类器,设定分类器的参数,并设定分类器输出结果为0或1,其中0为非房颤,1为非房颤;本实施例中,设定随机森林分类器的参数为:min_samples_split=2,n_jobs=8,n_estimators=10,verbose=0,其余为默认。步骤4.3:将BCG信号迁移学习特征参数作为随机森林分类器的输入,进行十折交叉验证,将随机森林分类器的输出结果转化为混淆矩阵,并对随机森林分类器参数进行二次调试,以获取最优特异性、敏感性、准确度和精度的性能参数;步骤5、将测试集中BCG信号作为卷积神经网络的输入,获取迁移学习模型特征参数并输入到随机森林分类器,通过随机森林分类器的输出结果判断该BCG信号是否为房颤疾病类型。本实施例还分别应用未预训练的CNN及线性判别分析、二次判别分析、支持向量机、朴素贝叶斯、递归树、决策树等经典机器学习方法,对同组BCG信号进行房颤疾病检测对比实验,并计算各实验的特异性、敏感性、准确性、精度等性能参数,评估本发明方法的分类性能。本实施例中,将通过各种方法实现对BCG信号进行房颤疾病检测的输出结果转化为混淆矩阵表示,如表2所示:表2混淆矩阵设Z表示一个2×2混淆矩阵,其中Zij为实际类为i而预测类为j的数据量,使用Z定的分类结果的、性和、性的数量及敏感性、特异性、准确性、精度的表式如式2-9所示:性:tpk=Ckk2性:fpk=∑i≠kCki3性:tnk=∑i,j≠kCij4性:fnk=∑i≠kCik5特异性:敏感性:准确性:精度:最后应说明的是:以上实施例用以说明本发明的技术方案,而非对其限制;管参前述实施例对本发明进行详细的说明,本领域的通技术人员应当理解:其然可以对前述实施例所记的技术方案进行,或者对其中分或者全技术特征进行等同;而或者,并不使相应技术方案的本本发明利要求所限定的范围。

权利要求:1.一种基于迁移学习的心冲击信号房颤计算机辅助诊断方法,其特征在于:包括以下步骤:步骤1:对同步实测的ECG信号、BCG信号及现房颤疾病数据库中的ECG信号进行预处理,并针对房颤疾病的特点,设定所有信号数据的分帧长度为24秒,作为神经网络的输入向量;步骤1.1:采集同一受试者的同步ECG信号、BCG信号,并分别对其信号进行归一化处理,以获取实测数据;步骤1.2:从现有的房颤疾病数据库中获取房颤ECG信号,并根据实测ECG信号的采样频率,对数据库中获取的ECG信号进行重采样处理,使其与步骤1.1中的实测数据采样频率一致;步骤1.3:针对房颤疾病的特点,设定所有信号数据的分帧长度为24秒,作为神经网络的输入向量;步骤1.4:应用实测ECG信号作为诊断标准,将每一帧实测同步BCG信号标记为房颤和非房颤二类,其中,房颤标定为1,非房颤标定为0;步骤1.5:将70%的实测BCG数据作为训练集、20%的实测BCG数据作为验证集、10%的实测BCG数据作为测试集;步骤2:构建进行房颤辅助检测的卷积神经网络;所述卷积神经网络包括4层卷积层、6层池化层和1层Flatten层以及3层全连接层,具体设计方法为:步骤2.1:以分帧长度为24秒的数据库中获取的ECG信号作为输入向量,确定输入层的神经元数目;步骤2.2:构建两个特征提取模块和一个降维模块;其中,第一个特征提取模块包括两层卷积层和两层池化层,第二个特征提取模块包括四层卷积层与四层池化层;降维模块包括一层Flatten层及三层全连接层;步骤2.3:设计输出层为2个神经元,0-1组合判定为非房颤,1-0组合判定为房颤,0-0及1-1组合则需要进行二次判定;步骤3:训练所构建的卷积神经网络并进参数更新:应用从数据库中获取的房颤ECG数据对卷积神经网络进行逐层预训练并更新网络参数;然后固定第一层卷积层和第二层卷积层的网络参数,以实测BCG信号作为输入向量,建立新型交叉熵损失函数更新网络参数,获取迁移学习后的卷积神经网络模型,具体方法为:步骤3.1:应用从数据库中获取的ECG信号对所设计的卷积神经网络进行预训练并更新网络参数;步骤3.2:应用实测BCG信号训练集对预训练后的卷积神经网络进行监督二次训练,获取迁移学习后的卷积神经网络模型,即BCG信号迁移学习模型;步骤4:提取实测BCG信号的迁移学习特征向量并构建随机森林分类器:将迁移学习后的CNN全连接层的输出作为BCG信号迁移学习模型特征参数,设计并优化随机森林分类器,使其性能最优;步骤4.1:应用实测BCG信号验证集作为二次训练后CNN网络的输入,提取降维模块所生成信号的高维特征作为BCG信号迁移学习特征参数;步骤4.2:构建随机森林分类器,设定分类器的参数,并设定分类器输出结果为0或1,其中,0为非房颤,1为非房颤;步骤4.3:将BCG信号迁移学习特征参数作为随机森林分类器的输入,进行十折交叉验证,将随机森林分类器的输出结果转化为混淆矩阵,并对随机森林分类器参数进行二次调试,以获取最优特异性、敏感性、准确度和精度的性能参数;步骤5、将测试集中BCG信号作为卷积神经网络的输入,获取迁移学习模型特征参数并输入到随机森林分类器,通过随机森林分类器的输出结果判断该BCG信号是否为房颤疾病类型。2.根据权利要求1所述的一种基于迁移学习的心冲击信号房颤计算机辅助诊断方法,其特征在于:所述步骤3.2的具体方法为:首先固定预训练后的第一层卷积层和第二层卷积层参数,然后以损失函数不再减小为基准,获取最优化网络参数;其中:卷积层与池化层的激活函数选用Relu函数,全连接层选用softmax函数,损失函数针对CNN输出层的特点,设计为采用如下公式:其中,C为新型交叉熵损失函数,y为期望的输出向量,x为神经元个数:非房颤为0-1,房颤为1-0;a为CNN输出层的实际输出向量。

百度查询: 东北大学 一种基于迁移学习的心冲击信号房颤计算机辅助诊断方法

免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。