一种基于迁移学习心律失常分类识别的方法与流程

1.本发明涉及信号识别技术领域，特别涉及一种基于迁移学习心律失常分类识别的方法。


背景技术：

2.当前使用于进行心律失常监测的信号主要是ecg，其产品形态主要包括三个形式：一是基于腕带形态，通过分别用一只手的腕带外侧和另外一只手内侧的金属传感器采集单通道ecg信号的方法；二是使用holter形态的采集单(或多)通道ecg信号的方法；三是使用多导监护仪采集多通道ecg信号的方法。在这三类使用ecg信号进行心律失常识别的产品形态中：腕带形态由于需要两只手同时操作才能进行心律失常的识别，因此这种形态不适合用于心律失常的长期连续监测；holter形态适合长期连续监测，但其在舒适性上面的缺陷会影响人体睡眠、运动等其它的活动，对于非临床诊断患者反而带来生活上的不便；多导监护仪形态适合进行手术前后的重度病人。因此，开发一种便于采集可用心律失常监测的信号模型，同时用于结合可以不影响佩戴者的正常生活的设备，具有重要的实际意义。
3.在此基础上出现了一种结合腕带，结合腕带最近已经成为一种广泛使用的健康管理工具，每天都有大量的人佩戴，同时装有ppg传感器的腕带可以为心律失常提供一种低功耗、便捷的监测方法。因此，结合深度学习算法的腕带式可穿戴设备是检测心律失常的有效方法。然而，由于当前用ppg检测心律失常的数据量少，并且ppg不能识别心房活动(p波)导致检测的心律失常类别有限，如何获取能有效的识别心律失常ppg信号是一个值得探索的方向。
4.可以克服这个挑战的一个方法是迁移学习方法的能力，特别是基于监督深度学习的迁移学习模型，如卷积神经网络、递归神经网络或注意力网络模型的迁移学习模型，已经被证明可以在给定源域和目标域的基础上从源域学习到经验用于高效构建目标域的分类模型(其过程如图1所示)。
5.成功应用于ecg信号心律失常识别分类的深度学习方法，在用于ppg信号进行心律失常分类时，出现了新的挑战：为了使用ppg信号进行心律失常分类识别产生准确的结果，需要来自采集心律失常过程对应ppg信号中具有代表性和可靠的数据基础，但在稀疏数据是常见问题的心律失常ppg的信号领域却是昂贵的要求。
6.在标注心律失常ppg信号的背景下，这些数据是通过在真实患者身体上进行采集或通过模拟来收集或者通过ecg信号进行迁移获得的。三种可能性都有其缺点。
7.首先，真实的人体数据采集既费时又费钱，导致很难收集足够的数据来训练可靠的深度学习模型；
8.其次，模拟的本质是假设现实的简化视图，导致成本更低，并且在大多数情况下，具有更快的结果，然而，这确实偏离了真实的观察；
9.最后，相近数据集的迁移，可以减少在学习新任务所需的数据量和时间，或者模型预测分类性能的提高(与从头开始学习相比)，但是这建立在两个领域的解空间相似的假设
条件下。
10.因此，在仿真数据上训练的深度学习模型对于真实数据的性能高度依赖于仿真和现实之间的差距，在相近数据集上迁移的深度学习模型对于真实数据的性能高度依赖于源域和目标域之间的差距，即使最精确的模型也只能再现仿真与现实，源域和目标域之间的偏差。这就是本发明要解决的一个中心问题，即如何通过监督学习模型来克服模拟和真实数据之间、源域和目标域的差距，而不需要许多昂贵的真实实验。


技术实现要素：

11.为解决上述现有技术中所存在的心律失常情况下ppg信号采集困难、以及一致性不强等问题，本发明提供一种基于迁移学习心律失常分类识别的方法，在模型性能不明显降低的情况用于训练的真实ppg信号数据量大大减少，旨在解决传统深度学习模型对ppg信号实现心律失常分类监测的模型所需标注的真实数据样本较大的问题，同时提升模型的泛化性能。
12.为了实现上述技术目的，本发明提供了如下技术方案：
13.一种基于迁移学习心律失常分类识别的方法，包括：
14.构建神经网络模型，通过ecg信号标注样本对神经网络模型进行训练，其中，神经网络模型中包含lstm网络；
15.通过模拟ppg历史数据标注样本对通过ecg信号标注样本训练后的神经网络模型进行训练；
16.将通过模拟ppg历史数据标注样本训练后的神经网络模型中的lstm网络替换为swlstm网络，通过真实ppg历史数据标注样本对替换完成后的模型进行训练，得到识别模型，其中识别模型用于进行心律失常分类识别。
17.可选的，所述神经网络模型为未训练的cti-net模型，其中cti-net模型中包括cnn网络、lstm网络及全连接层，其中cnn网络、lstm网络及全连接层两两之间相互连接。
18.可选的，所述ecg信号标注样本、模拟ppg历史数据标注样本及真实ppg历史数据标注样本的获取过程包括：
19.分别采集ecg信号、模拟ppg历史数据、真实ppg历史数据；
20.分别对采集ecg信号、模拟ppg历史数据、真实ppg历史数据进行标准化处理；
21.分别获取采集ecg信号、模拟ppg历史数据、真实ppg历史数据对应的标签；
22.分别对应的标签与标准化处理后的数据进行整合，得到ecg信号标注样本、模拟ppg历史数据标注样本及真实ppg历史数据标注样本，其中，ecg信号、模拟ppg历史数据、真实ppg历史数据分别为相同或不同的固定长度，所述对应的标签为分类预测输出心律失常类别。
23.可选的，所述标准化处理的过程如下式：
[0024][0025]
其中，s1,
…
,sn为采集ecg信号、模拟ppg历史数据或真实ppg历史数据的变量序列，μ和σ为标准化参数，μ为平均值，σ为标准差值，标准化参数根据ecg信号、模拟ppg历史数据、真实ppg历史数据的数据类型不同进行设置。
[0026]
可选的，通过滑动窗口法采集ecg信号、模拟ppg历史数据、真实ppg历史数据。
[0027]
可选的，将通过模拟ppg历史数据标注样本训练后的神经网络模型中的lstm网络替换为swlstm网络时，通过模拟ppg历史数据标注样本训练后的神经网络模型中除lstm网络之外的其他网络模型的结构及权重参数保持不变，并在所述lstm网络替换为swlstm网络后，对swlstm网络中的权重参数进行随机初始化。
[0028]
可选的，所述识别模型为cswti-net模型。
[0029]
可选的，所述心律失常分类包括：正常、房颤、一级房室传导阻滞、左束支传导阻滞、右束支传导阻滞、房性早搏及室性早搏。
[0030]
本发明具有如下技术效果：
[0031]
本文提出的方法的创新之处体现在：1)为充分提取ecg/ppg信号的特征信息，基于卷积神经网络和门长短时记忆网络，提出了提取空时特征信息，然后通过相同全连接单元进行特征融合的深度学习模型cti-net；2)使用真实的ecg信号作为源域桥接模拟ppg信号的中间域进行数据域迁移，使得训练模型cti-net进一步接近目标域的解空间；3)使用真实的ppg信号对模型cti-net进行结构上的迁移，得到适合ppg信号进行心律失常分类识别的高性能模型，并且进一步压缩模型的运算量。同发明提出的模型结构不仅适合于ecg信号的心律失常识别分类，并且可以在减少真实ppg信号量的基础上确保基于ppg信号的心律失常识别分类性能不明显降低。
附图说明
[0032]
为了更清楚地说明本发明实施例或现有技术中的技术方案，下面将对实施例中所需要使用的附图作简单地介绍，显而易见地，下面描述中的附图仅仅是本发明的一些实施例，对于本领域普通技术人员来讲，在不付出创造性劳动性的前提下，还可以根据这些附图获得其他的附图。
[0033]
图1为本发明实施例提供的高效构建目标域的分类模型示意图；
[0034]
图2为本发明实施例提供的cti-net模型示意图；
[0035]
图3为本发明实施例提供的方法流程示意图；
[0036]
图4为本发明实施例提供的cswti-net模型示意图。
具体实施方式
[0037]
下面将结合本发明实施例中的附图，对本发明实施例中的技术方案进行清楚、完整地描述，显然，所描述的实施例仅仅是本发明一部分实施例，而不是全部的实施例。基于本发明中的实施例，本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例，都属于本发明保护的范围。
[0038]
为了解决在现有技术中存在心律失常情况下ppg信号采集困难、以及一致性不强等问题，本发明提供了如下方案：
[0039]
本发明公开一种基于迁移学习律失常分类识别方法，包含空间特征提取原理、时间特征提取原理、空时特征融合原理、数据域迁移原理和结构域迁移原理。
[0040]
cnn空间特征提取原理，其中的一维卷积即试图提取信号在某一方向的平移不变特征，这里的ecg/ppg信号也是一种时间序列。因此，使用一维卷积进行提取的ecg/ppg信号
在时间方向的平移不变特征，可以更充分提取了ecg/ppg信号中的特征信息并且筛选出具有区分度的信息，在提高了ecg/ppg信号在不同心律失常实时分类监测方法的准确性能同时，也保证了分类系统的泛化性能。其中构建的cnn层结构结果如图2所示，设置两层图卷积层，同时使用relu作为激活函数，然后使用全连接层将cnn提取的特征线性映射后输入特征融合模块。
[0041]
lstm时间特征提取原理，lstm作为循环神经网络的变种，具有非线性拟合能力，能够有效提取数据的时间特征，克服了rnn中容易出现梯度消失和梯度爆炸导致模型不收敛的问题。其中构建的lstm层结构结果如图2所示，设置一层lstm层，然后使用全连接层将lstm提取的特征线性映射后输入特征融合模块。
[0042]
swlstm时间特征提取原理，swlstm作为lstm的变种，不仅具有非线性拟合能力，能够有效提取数据的时间特征，同时在保障得到与lstm相近预测效果的同时获得更快的收敛速度。其中构建的包含有swlstm流程结构如图3所示(用共享门取代了lstm的输入门、遗忘门和输出门)，直接使用swlstm层取代lstm层，然后使用全连接层将swlstm提取的特征线性映射后输入特征融合模块。
[0043]
空时特征融合原理，将cnn提取的特征线性映射后的特征和lstm提取的特征线性映射后的特征进行进一步的线性映射，消除不同特征直接由于幅度不同导致对有区分度特征的提取不充分问题，其在模型训练过程中使用更加有效率的梯度下降以及反向传播，避免了梯度爆炸和梯度消失问题，同时简化计算过程。
[0044]
数据域迁移原理，将预先训练好的基于ecg信号(源域)进行心律失常分类的网络重新用于模拟ppg信号(中间域)进行心律失常分类的进一步训练。这样，在对中间域数据进行训练之前，所有关于模型的网络权重都用从源域中获得的值进行初始化(而非重新开始训练时所有参数进行随机化值)。
[0045]
结构域迁移原理，通过在中间域训练好的模型进行结构微调，保留卷积层参数，将lstm替换成swlstm层，在获得较高模型性能的同时，进一步降低模型的计算量，同时获得真实ppg信号(目标域)的心律失常识别分类模型。
[0046]
为了达到上述目的，本发明采用下面的技术方案予以实现：
[0047]
一种基于迁移学习心律失常分类识别的方法，基于以下定义：
[0048]
定义：选取固定长度的ecg/ppg信号作为样本数据和对应的心律失常类别作为样本标签。
[0049]
如图3所示，具体包括如下步骤：
[0050]
步骤s1：将已采集的ecg信号标注样本后进行模型训练；
[0051]
步骤s2：使用滑动窗口法将采集的固定长度的ecg信号，使用训练集共享的标准化参数进行标准化后作为输入，对应分类预测输出心律失常类别为标签；
[0052]
步骤s3：得到cti-net模型，且保留cti-net模型各层的权重；
[0053]
步骤s4：将模拟的ppg历史数据标注样本后进行模型训练，其中模型参数使用第3步的权重进行初始化；
[0054]
步骤s5：使用滑动窗口法将采集的固定长度的ppg信号，使用训练集共享的标准化参数进行标准化后作为输入，对应分类预测输出心律失常类别为标签；
[0055]
步骤s6：得到适应中间域的cti-net模型，且冻结cti-net模型各卷积层的权重；
[0056]
步骤s7：将真实的ppg历史数据标注样本后进行模型训练，其中模型卷积层参数使用第6步的权重进行保持不变，lstm用swlstm进行替换并随机初始化；
[0057]
步骤s8：得到适应目标域的cswti-net模型(具体结构如图4)。
[0058]
其中标准化的方法为，假设某一变量序列为s1,
…
,sn，记其中平均值和标准差值分别为:μ和σ,则标准化后的序列为：其中标准化后的序列值的服从正态分布，存储μ和σ的值。
[0059]
本发明提出用于ecg/ppg信号用以不同心律失常识别分类的cswti-net模型，使用第一届mit的数据库进行验证，包括正常(n)、房颤(af)、一级房室传导阻滞(i-avb)、左束支传导阻滞(lbbb)、右束支传导阻滞(rbbb)、房性早搏(pac)、室性早搏(pvc)六大类，使用准确性能指标：准确率、召回率、特异性和马太相关系数。
[0060]
其中准确率(简称ac)的计算公式为：
[0061]
其中召回率(简称re)的计算公式为：
[0062]
其中特异性(简称sp)的计算公式为：
[0063]
其中马太相关系数(mcc)的计算公式为：
[0064]
这里的tp、tn、fp和fn分别表示真阳性、真阴性、假阳性和假阴性的实例。对马太相关系数mcc的取值范围从-1到1中，mcc为-1表示最坏的可能预测，而值为1表示最好的可能预测方案。此外，mcc为0表示随机预测。将标注的10000个样本点按照7:3的比例划分为训练集和测试集，其中训练集、测试集的样本点数为7000个和3000个，其中训练后的cswti-net模型和具有相似结构的cnn模型和cti-net对测试集的分类结果用性能指标表示如表1所示。可以看出相较于cnn模型和cti-net，cswti-net模型对用于ecg信号不同心律失常实时分类监测时的测试集具有明显的性能提升。表1为cnn模型、cti-net与cswti-net模型的性能对比表。
[0065]
表1
[0066][0067]
将标注的1000个真实ppg样本点按照7:3的比例划分为训练集和测试集，其中训练集、测试集的样本点数为700个和300个，其中迁移前后的cswti-net模型和具有相似结构迁移前后cti-net对测试集的分类结果用性能指标表示如表2所示。可以看出相较于迁移前cti-net模型和迁移前的cswti-net模型，迁移后cti-net模型和迁移后的cswti-net模型的
对用于ppg信号不同心律失常实时分类监测时的测试集具有明显的性能提升，相对cti-net模型，cswti-net模型也具有明显的优势。其中表2为cti-net与模型cswti-net模型的性能对比表。
[0068]
表2
[0069][0070]
本文提出的方法的创新之处体现在：1)为充分提取ecg/ppg信号的特征信息，基于卷积神经网络和门长短时记忆网络，提出了提取空时特征信息，然后通过相同全连接单元进行特征融合的深度学习模型cti-net；2)使用真实的ecg信号作为源域桥接模拟ppg信号的中间域进行数据域迁移，使得训练模型cti-net进一步接近目标域的解空间；3)使用真实的ppg信号对模型cti-net进行结构上的迁移，得到适合ppg信号进行心律失常分类识别的高性能模型，并且进一步压缩模型的运算量。同发明提出的模型结构不仅适合于ecg信号的心律失常识别分类，并且可以在减少真实ppg信号量的基础上确保基于ppg信号的心律失常识别分类性能不明显降低。
[0071]
以上显示和描述了本发明的基本原理、主要特征和优点。本行业的技术人员应该了解，本发明不受上述实施例的限制，上述实施例和说明书中描述的只是说明本发明的原理，在不脱离本发明精神和范围的前提下，本发明还会有各种变化和改进，这些变化和改进都落入要求保护的本发明范围内。本发明要求保护范围由所附的权利要求书及其等效物界定。