[发明专利]基于注意力机制的无监督多模态心电图异常检测方法

说明书

本发明公开了一种基于注意力机制的无监督多模态心电图异常检测方法，根据体表电生理信号无监督地自动进行心电图的异常检测，模型在时序电信号的基础上还考虑了频域信号特性，通过对输入信号编码再解码重构出原始信号，当重构的时序信号与输入时序信号差异大于一定值，则判断该心电图为异常。本发明通过对ECG的时域与频域信号进行特征提取，在无监督不使用标签的情况下，能够区分正常与异常的ECG心拍数据；在有标签的心电图数据有限的情况下，辅助医生自动筛查异常心电图，具有临床的诊断及治疗上的参考意义。

技术领域

本发明属于心脏电生理疾病预测技术领域，具体涉及一种基于注意力机制的无监督多模态心电图异常检测方法。

背景技术

心血管疾病一系列心脏或者血管相关疾病总称，一直都是威胁人类生命的主要疾病之一，其引起的死亡率仍占首位。ECG(心电信号)是人类最早研究并应用于医学临床的生物信号之一，心电图反映了人体心脏健康状况，是临床诊断心血管类疾病的重要依据。随着心电图数量的快速增长，利用AI算法辅助医生进行自动化异常诊断极为重要。

现有的许多ECG自动诊断算法是基于专家医生分析和标注的数据进行监督学习得到的。但由于专家只能处理少量的心电图数据，并且这些数据大都来自于同一种模式，因此基于监督学习的自动诊断算法有其局限性。另外，标签数据中各类型标签的分布通常不平衡，这会对模型的诊断分析效果造成影响；未经标注的数据也是有价值的，它可以进一步揭示患者心血管系统内部的潜在病理信息。因此，研发一种能够进行无监督诊断心电图数据的算法是非常重要的。

目前，现有的无监督ECG异常检测或分类方法大多都没有充分利用和挖掘数据本身的特征，大致可分为两类：一类是使用聚类算法来实现无监督决策，例如文献[Abawajy JH,Kelarev A V,Chowdhury M.Multistage approach for clustering andclassification of ECG data[J].Computer methods and programs in biomedicine,2013,112(3):720-730]使用高斯混合模型和K-means聚类算法，将ECG数据转换为数值特征，从而完成异常分类。另一类是通过深度学习端到端地实现ECG的异常检测，例如文献[Pereira J,Silveira M.Unsupervised representation learning and anomalydetection in ECG sequences[J].International Journal of Data Mining andBioinformatics,2019,22(4):389-]在递归网络作为自编码器的基础上，再对其编码特征进行解码重构输入。尽管如此，但上述方法存在的缺陷是没有充分利用和挖掘ECG的信息，模型缺乏结合更多维度的特征。

发明内容

鉴于上述，本发明提供了一种基于注意力机制的无监督多模态心电图异常检测方法，利用注意力机制去挖掘时域与频域数据中的隐藏特征，无监督地重构时域输入，能够很好地解决含有标签的心电图数据缺少，有监督的模型难以取得广泛的自动诊断效果的现象。

一种基于注意力机制的无监督多模态心电图异常检测方法，包括如下步骤：

(1)从病人体表采集多导联心电信号，以每一个心跳周期为一组心电时域序列；

(2)对每组心电时域序列进行归一化处理以及频域转换，得到对应的心电频域序列；

(3)构建基于注意力机制的重构模型，其包含两个编码模块以及一个解码模块，两个编码模块分别用于对心电时域序列和心电频域序列进行编码，得到的编码特征拼接后经由解码模块解码成与输入维度一致的ECG波形序列；

(4)将心电时域序列和心电频域序列成对逐一输入至模型，以模型输出的ECG波形序列与输入心电时域序列平均误差最小为损失函数，从而对模型进行训练；