轻量级心电图心律失常分类模型,兼顾准确与可解释性。
ArrhythmiaVision: Resource-Conscious Deep Learning Models with Visual Explanations for ECG Arrhythmia Classification
- 设计轻量1D CNN模型,内存仅302KB和157KB
- MIT-BIH数据集上五类分类准确率达0.99和0.98
- 集成SHAP与Grad-CAM实现生理可解释性
心律失常是导致致命心脏事件的主要原因,亟需精准及时的检测。心电图(ECG)仍是临床诊断金标准,但人工解读耗时且依赖经验,易出错。尽管深度学习推动了自动分析,现有模型常忽略信号的时间与形态特征,缺乏可解释性,计算开销大,难以部署于资源受限设备。本文提出两个新型轻量级1D卷积神经网络ArrhythmiNet V1与V2,专为边缘设备实时分类优化。受MobileNet启发,模型内存分别仅302.18 KB与157.76 KB,于MIT-BIH心律失常数据集上对正常窦性心律、左束支阻滞、右束支阻滞、房性早搏、室性早搏五类分类准确率达0.99(V1)与0.98(V2)。为确保临床透明性,引入SHAP与梯度加权类激活映射,揭示QRS波群与T波等生理相关模式对预测的贡献。讨论性能-效率权衡,指出数据多样性与泛化能力局限。结果表明,该方法可在可穿戴与嵌入式系统中实现高精度、可解释、低资源的心电监测。
原文摘要 · Abstract (English)
Cardiac arrhythmias are a leading cause of life-threatening cardiac events, highlighting the urgent need for accurate and timely detection. Electrocardiography (ECG) remains the clinical gold standard for arrhythmia diagnosis; however, manual interpretation is time-consuming, dependent on clinical expertise, and prone to human error. Although deep learning has advanced automated ECG analysis, many existing models abstract away the signal's intrinsic temporal and morphological features, lack interpretability, and are computationally intensive-hindering their deployment on resource-constrained platforms. In this work, we propose two novel lightweight 1D convolutional neural networks, ArrhythmiNet V1 and V2, optimized for efficient, real-time arrhythmia classification on edge devices. Inspired by MobileNet's depthwise separable convolutional design, these models maintain memory footprints of just 302.18 KB and 157.76 KB, respectively, while achieving classification accuracies of 0.99 (V1) and 0.98 (V2) on the MIT-BIH Arrhythmia Dataset across five classes: Normal Sinus Rhythm, Left Bundle Branch Block, Right Bundle Branch Block, Atrial Premature Contraction, and Premature Ventricular Contraction. In order to ensure clinical transparency and relevance, we integrate Shapley Additive Explanations and Gradient-weighted Class Activation Mapping, enabling both local and global interpretability. These techniques highlight physiologically meaningful patterns such as the QRS complex and T-wave that contribute to the model's predictions. We also discuss performance-efficiency trade-offs and address current limitations related to dataset diversity and generalizability. Overall, our findings demonstrate the feasibility of combining interpretability, predictive accuracy, and computational efficiency in practical, wearable, and embedded ECG monitoring systems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。