arXiv:2502.17887cs.LGcs.AI2025-02被引 9

用深度学习模型分析12导联心电图,实现高效心律失常分类。

Arrhythmia Classification from 12-Lead ECG Signals Using Convolutional and Transformer-Based Deep Learning Models

  • 结合一维卷积与Transformer模型,处理原始心电波形和图像化信号
  • 一维CNN+GRU架构在原始信号上达到93.4%准确率,表现最佳
  • 适用于资源有限的医疗环境,可快速部署于基层诊疗

罗马尼亚心血管疾病是死亡主因,占年死亡人数近三分之一。为应对这一挑战,本文探索适用于资源受限医疗环境的心律失常快速诊断方法。由于缺乏本地公开医学数据,研究采用国际通用数据集:PTB-XL、PTB Diagnostic ECG Database、China 12-Lead ECG Challenge Database、Georgia 12-Lead ECG Challenge Database 和 St. Petersburg INCART 12-lead Arrhythmia Database。采用改进的Pan-Tompkins算法提取QRS波群,作为特征输入。模型包括一维/二维卷积神经网络(1D CNNs, 2D CNNs, ResNet)及视觉变压器(ViTs)。评估指标为准确率与F1分数。结果显示,基于GRU的一维卷积模型在原始信号上取得最高准确率93.4%;将信号转为图像后,2D CNN模型达92.16%准确率。

原文摘要 · Abstract (English)

In Romania, cardiovascular problems are the leading cause of death, accounting for nearly one-third of annual fatalities. The severity of this situation calls for innovative diagnosis method for cardiovascular diseases. This article aims to explore efficient, light-weight and rapid methods for arrhythmia diagnosis, in resource-constrained healthcare settings. Due to the lack of Romanian public medical data, we trained our systems using international public datasets, having in mind that the ECG signals are the same regardless the patients' nationality. Within this purpose, we combined multiple datasets, usually used in the field of arrhythmias classification: PTB-XL electrocardiography dataset , PTB Diagnostic ECG Database, China 12-Lead ECG Challenge Database, Georgia 12-Lead ECG Challenge Database, and St. Petersburg INCART 12-lead Arrhythmia Database. For the input data, we employed ECG signal processing methods, specifically a variant of the Pan-Tompkins algorithm, useful in arrhythmia classification because it provides a robust and efficient method for detecting QRS complexes in ECG signals. Additionally, we used machine learning techniques, widely used for the task of classification, including convolutional neural networks (1D CNNs, 2D CNNs, ResNet) and Vision Transformers (ViTs). The systems were evaluated in terms of accuracy and F1 score. We annalysed our dataset from two perspectives. First, we fed the systems with the ECG signals and the GRU-based 1D CNN model achieved the highest accuracy of 93.4% among all the tested architectures. Secondly, we transformed ECG signals into images and the CNN2D model achieved an accuracy of 92.16%.

心律失常12导联心电图深度学习医疗AI

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。