具体实施方式

现在结合附图对本发明作进一步详细的说明。

需要注意的是，发明中所引用的如“上”、“下”、“左”、“右”、“前”、“后”等的用语，亦仅为便于叙述的明了，而非用以限定本发明可实施的范围，其相对关系的改变或调整，在无实质变更技术内容下，当亦视为本发明可实施的范畴。

本发明提供了一种基于对比预测编码的ECG数据特征生成模型，引入自监督学习模型对比预测编码来预测出与原ECG数据相同类别的特征，在增大样本集的同时，减少人工标注成本，同时也便于其它分类模型减少过拟合提高分类准确率。本发明旨在解决以下问题：

1)ECG数据数量不平衡。数据作为模型训练的材料，其大小往往决定了模型的性能。如今已存在各种各样针对ECG进行分类的模型，然而由于ECG数据本身的特点，一些心率类型所对应的数据非常少，计算机得不到足够的数据进行训练，从而导致分类模型出现各种各样的问题。

2)标注成本居高不下。全监督学习训练过程中需要用到大量人工标注标签的数据集，然而标注数据类型将消耗大量的人力、物力等资源。针对ECG这种特定的数据，标注则需要一定的专业知识，提高了人工标注的门槛，使得大规模、更复杂的数据集标注愈加艰难。

3)预测准确度不高。使用深度学习模型预测ECG数据特征的最终目标就是快速精准的实现预测。现有模型多是基于相同的概率分布进行预测，并没有挖掘不同类别数据之间特征的不同，往往预测准确率不高。

4)提高下游任务训练速度。通过对比预测编码可以使得编码后相同样本之间的差距越来越小，不同样本之间的差距越来越大，从而在正式的下游任务训练之前对不同类别的数据做了区分，可以加快下游任务的收敛速度。

本发明主要有以下步骤：

如图1所示，图1展示了对比预测编码应用在ECG数据上的工作步骤，首先将划分ECG训练数据,横向看是正样本对与负样本对，正样本对是相同类别的数据，负样本对是不同类别的数据，纵向看是训练数据与待训练数据，然后通过编码器将训练数据与待训练数据都进行编码,接着把训练数据经过编码得到的结果放入自回归模型得到上下文信息Context，Context进入预测模型后得到未来多步的预测值，最后将预测值与待训练数据经过编码后的值一起计算点积得到损失值。通过反向传播更新梯度逐渐减少损失值。具体执行过程如下：

步骤1：数据预处理

1.1采用数据集自身标注的R峰位置采集心拍

1.2将心拍重采样

1.3使用小波变换进行滤波

1.4为数据集重新打上标签并打乱重排，将数据集分为训练集和验证集，训练集又分为两部分，分别是训练数据与待训练数据。同时构造正样本对与负样本对。

步骤2：搭建CPC模型

2.1搭建编码器模型提取数据特征

2.2构造自回归网络模型，将2.1中提取的特征输入自回归网络模型，使用自回归模型GRU来融合历史信息，得到历史信息的特征融合向量c。

2.3搭建预测模型，将2.2得到的特征融合向量输入预测模型生成预测结果。

2.4构造模型输出，将2.3得到的预测结果与待训练数据经过编码器得到的结果进行点积运算，根据向量点积运算的特性，两部分数据约相似，结果越大，反之越小，将计算得到的结果使用sigmoid函数使其在[0,1]范围内，作为模型的输出。

2.5构造CPC模型，CPC模型的输入数据包括训练数据和待训练数据，输出为2.4计算的输出。

步骤3：训练CPC模型

3.1初始化模型参数

3.2将数据输入到模型中进行训练

3.3保存模型，并作图绘制训练集以及验证集的准确率。此时模型可以生成出与原ECG数据相同类别的特征，可用于下游任务比如分类任务的训练。

步骤4：分类任务

4.1随机打乱训练数据，用K折交叉验证的思想将训练数据划分为5折。

4.2构造分类模型，分类模型特殊地方在于使用了三份相同的训练数据，而不是只有一份。

4.3训练模型。

以下是本发明的一个具体实施例。

步骤1：数据预处理

1.1通过R峰位置截取心拍。本次实验采用MIT-BIT心律不齐数据库作为数据集。MITAB包含48条双导联的ECG记录，除少数记录外，每条记录的第一个导联都是II导联，每条记录长度为30分钟，采样率360Hz，每条记录都有650000个点。由于102、104、107、217这四条记录包含起搏心拍，将这四条记录删除。以R峰为基准点，取前0.4s和后0.5s作为一个心拍，采样率是360Hz，所以心拍长度为0.4×360+0.5×360＝324，将心拍重采样到251。按照美国医疗仪器促进协会(The Association for the Advancement of MedicalInstrumentation,简称AAMI)提出的标准，所有的心拍可以被分为这五大类：正常心拍(normal beat,简称N)、室上异位心拍(supraventricular ectopic beat,简称SVEB)、心室异位心拍(ventricular beat，简称VEB)、融合心拍(fusion beat，简称F)以及未知分类心拍(unknown beat，简称Q)。截取心拍的时候同时对心拍进行归类，得到的N、SVEB、VEB、F、Q这五种类型的心拍所对应的个数分别为：90081、2781、7008、802、15。由于Q为未分类心跳，所以最终数据集只包含N、SVEB、VEB、F这四大类，标签编码为0、1、2、3。

1.2将所有的数据与对应的标签按照互相对应的关系进行打乱重排。取前90％个数据作为训练集，后10％个数据作为验证集。

1.3对信号进行小波变换滤波。小波基使用db6，滤波将小于5hz和大于90hz的小波系数替换为0，只保留第3和第6细节子带之间的系数进行重构。

1.4生成正样本与负样本。其中正样本对的类别相同，负样本对的类别不相同，将样本数据划分为训练数据与待训练数据。训练数据与待训练数据都为(32,4,151)。

步骤2：搭建CPC模型。

2.1搭建编码器模型。编码模型整体架构如图2所示，前半部分由四个由全连接层、批归一化层、LeakyReLu激活层组成的块组成，全连接层的输出维度为64，经过一个Flatten层后，再接一个块，此块的全连接层的输出维度为256，最后一个全连接层的输出维度为10，所有的全连接层都使用线性激活函数。编码器的作用是提取训练数据的特征。

2.2构造自回归网络模型。自回归网络模型使用GRU(Gated Recurrent Unit)，输出维度为256，只返回最后一个单元的输出。这一部分得到的是融合了历史信息的特征融合向量c。

2.3搭建预测模型。预测模型如图3所示，其中全连接层输出维度10，使用线性激活函数。由于四个全连接层是放在一个列表里的，使用Lambda层将这四个全连接层横向拼接起来构成一个网络。自回归模型输出的特征融合向量输入预测模型后输出预测的结果。

2.4搭建输出模型。输出模型计算待训练数据经过编码器编码得到的结果与预测模型生成的待预测结果之间的点积，求平均值后通过sigmoid函数将其值映射到[0，1]范围内。

2.5构建整体CPC模型，输入为训练数据与待训练数据，训练数据经过编码器、自回归模型、预测模型后得到结果1，待训练数据经过编码器得到结果2，计算结果1与结果2的点积值，输出为点积值经过sigmoid后的结果。

步骤3初始化模型参数，将ECG数据输入模型进行训练。

3.1初始化模型参数。设置模型学习率为0.001，样本批次大小为32，迭代次数10。使用Adam优化学习率，损失函数使用binary_crossentropy。其中对学习率做了特殊处理以便模型更快的收敛，当2个epoch过去而模型性能不在提升时，学习率将减少为原来的1/3。

3.2生成训练数据与待训练数据。其中训练数据与待训练数据，正样本对与负样本对的关系如图4所示。正样本对的个数与负样本对的个数相同，根据训练集不同类别个数的不同，使产生N、SVEB、VEB、F类别的概率分别为0.1,0.3,0.2,0.4，使得样本个数较少的类别能得到充分的训练。训练数据经过编码器、自回归模型、预测模型后生成的数据为预测数据，待训练数据经过编码器后生成的数据成为待预测数据。将训练数据与待训练数据输入模型进行训练。

3.3保存模型。将训练好的模型保存起来，同时绘制训练过程中训练集以及验证集的准确率，如图5所示，从图中可以看出模型取得了非常好的效果。

步骤四：将训练好的CPC模型运用到下游的分类任务中

4.1划分训练数据。为了与训练好的CPC模型保持一致，训练集数据还是使用MIT-BIT心律不齐数据库，将数据集划分为5份。

4.2构造分类模型。分类模型使用的三份相同的训练数据。每一份训练数据都经过CPC的编码器部分、一维卷积层、relu激活层、一维最大池化层、一维卷积层、relu激活层、一维最大池化层。将三份数据得到的结果进行拼接后接一个Flatten层、两个全连接层，最后经过一个激活函数为softmax的全连接层得到分类结果。具体模型结构如图6所示。

4.3训练分类模型。损失函数使用categorical_crossentropy，优化器使用rmsprop，批量大小设置为64，训练10个epoches。训练过程的模型准确率如图7所示。

本发明适用于数据不平衡条件下ECG数据。

本发明适用于收集到的ECG数据较少的，不太常见的心律失常数据。针对收集到的数据较少的情况，CPC可以通过最大化自身的互信息，就可以解决因为数据量不足所带来的问题，由此可以扩展样本数量不足的数据，提高下游任务的泛化能力。

本发明提高了产生相同类别ECG数据特征的准确率。

通过编码器提取有效特征，去除了不必要的噪声，使特征更加明显，便于后续处理。对比预测编码利用自身的互信息，提高自身预测能力的同时，强化了编码器的特征提取能力，特征提取效果相当不错。模型在MIT-BIT心律不齐数据库上取得了相当好的效果。

本发明加快了下游ECG分类模型的训练速度。

通过对比预测编码将ECG数据经过编码器编码后已经对不同类别的数据做了一个区分，这样在训练分类模型时可以提高模型的收敛速度，加快模型训练。

以上仅是本发明的优选实施方式，本发明的保护范围并不仅局限于上述实施例，凡属于本发明思路下的技术方案均属于本发明的保护范围。应当指出，对于本技术领域的普通技术人员来说，在不脱离本发明原理前提下的若干改进和润饰，应视为本发明的保护范围。