arXiv:2602.17701eess.SPcs.LG2026-02被引 4

对比四种深度网络,提升心电图心律失常分类准确率

Deep Neural Network Architectures for Electrocardiogram Classification: A Comprehensive Evaluation

  • 用CNN、LSTM和注意力机制捕捉心电信号的局部特征与长期依赖
  • 动态集成策略使F1分数达0.958,优于单一模型的0.951
  • 模型可解释性通过Grad-CAM验证,适合临床辅助诊断场景

随着心血管疾病发病率上升,心电图(ECG)仍是无创检测心脏异常的重要工具。本研究对深度神经网络架构在自动心律失常分类中的表现进行了全面评估,结合时序建模、注意力机制与集成策略。针对少数类别数据稀缺问题,采用生成对抗网络(GAN)对MIT-BIH Arrhythmia数据集进行增强。构建并比较了四种架构:卷积神经网络(CNN)、CNN-LSTM、CNN-LSTM-Attention及1D残差网络(ResNet-1D),以捕捉局部形态特征与长期时序依赖。性能通过准确率、F1-score和曲线下面积(AUC)及95%置信区间进行严格评估,确保统计稳健性;同时使用梯度加权类激活映射(Grad-CAM)验证模型可解释性。实验结果表明,CNN-LSTM模型在敏感性与特异性间取得最优平衡,F1-score为0.951;而CNN-LSTM-Attention与ResNet-1D模型对类别不平衡更敏感。为此引入动态集成融合策略,其中Top2-Weighted集成达到最高性能,F1-score为0.958。结果表明,融合互补深度架构显著提升分类可靠性,为智能心律失常检测系统提供稳健且可解释的基础。

原文摘要 · Abstract (English)

With the rising prevalence of cardiovascular diseases, electrocardiograms (ECG) remain essential for the non-invasive detection of cardiac abnormalities. This study presents a comprehensive evaluation of deep neural network architectures for automated arrhythmia classification, integrating temporal modeling, attention mechanisms, and ensemble strategies. To address data scarcity in minority classes, the MIT-BIH Arrhythmia dataset was augmented using a Generative Adversarial Network (GAN). We developed and compared four distinct architectures, including Convolutional Neural Networks (CNN), CNN combined with Long Short-Term Memory (CNN-LSTM), CNN-LSTM with Attention, and 1D Residual Networks (ResNet-1D), to capture both local morphological features and long-term temporal dependencies. Performance was rigorously evaluated using accuracy, F1-score, and Area Under the Curve (AUC) with 95\% confidence intervals to ensure statistical robustness, while Gradient-weighted Class Activation Mapping (Grad-CAM) was employed to validate model interpretability. Experimental results indicate that the CNN-LSTM model achieved the optimal stand-alone balance between sensitivity and specificity, yielding an F1-score of 0.951. Conversely, the CNN-LSTM-Attention and ResNet-1D models exhibited higher sensitivity to class imbalance. To mitigate this, a dynamic ensemble fusion strategy was introduced; specifically, the Top2-Weighted ensemble achieved the highest overall performance with an F1-score of 0.958. These findings demonstrate that leveraging complementary deep architectures significantly enhances classification reliability, providing a robust and interpretable foundation for intelligent arrhythmia detection systems.

心电图分析深度学习分类医疗AI

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