高级检索

TranECG:基于Transformer的多导联心电图异常检测

TranECG:Transformer based Multi-lead ECG Anomaly Detection

  • 摘要: 心电图是检测心血管疾病最常用的工具之一. 然而, 心电图分析高度依赖医学专家的经验, 导致分析效率低. 为此, 提出一种集成了Transformer与生成对抗网络思想的异常检测方法——TranECG. 首先, 将长序列心电图数据分割成若干等距的心跳节拍数据; 然后, 采用基于生成对抗网络的重建模型学习节拍数据的正常模式, 并生成相应的重建序列; 最后, 异常检测模块通过分析重建序列与原始序列的残差矩阵, 判断心电图是否存在异常. 此外, 为了显式地定位异常, 根据异常检测模块的注意力系数矩阵构造热力图, 以可视化方式定位心电图中的异常位置. 实验结果表明, 在CPSC和AIWIN两个心电信号数据集上, TranECG相较于文中对比方法, 其准确率平均提高4.7%, 曲线下面积(AUC)平均提高1.4%.

     

    Abstract: Electrocardiogram (ECG) is one of the most used tools for detecting cardiovascular diseases. However, ECG analysis heavily relies on the experience of medical experts, leading to low analysis efficiency. To address this, we propose an anomaly detection method called TranECG, which integrates the ideas of Transformer and Generative Adversarial Network (GAN). First, the long-sequence ECG data is segmented into several equidistant heartbeat segments. Then, a reconstruction model based on GAN is employed to learn the normal patterns of the heartbeat segments and generate the corresponding reconstructed sequences. Finally, the anomaly detection module analyzes the residual matrix between the reconstructed sequences and the original sequences to determine whether anomalies exist in the ECG. Additionally, to explicitly locate anomalies, a heatmap is constructed based on the attention coefficient matrix of the anomaly detection module, visualizing the positions of anomalies in the ECG. Experimental results on the two ECG datasets from CPSC and AIWIN demonstrate that the accuracy of TranECG is improved by 4.7% on average, and the area under the curve (AUC) is enhanced by 1.4% on average compared to competitive methods in the paper.

     

/

返回文章
返回