用一维卷积与Mamba模型结合,提升12导联心电图异常分类准确率。
One Dimensional CNN ECG Mamba for Multilabel Abnormality Classification in 12 Lead ECG
- 融合一维卷积与选择性状态空间模型,更好捕捉心电信号时序特征。
- 在PhysioNet 2020/2021数据集上AUPRC和AUROC均优于已有方法。
- 适合临床辅助诊断、远程医疗及资源有限环境中的心电分析应用。
从心电图记录中准确检测心脏异常对临床诊断与决策支持至关重要。传统深度学习模型如残差网络和Transformer在处理长序列信号时表现受限。最近,状态空间模型被引入作为高效替代方案。本研究提出一种名为一维卷积神经网络心电图Mamba的混合框架,将卷积特征提取与Mamba——一种专为有效序列建模设计的选择性状态空间模型——结合。该模型基于双向变体Vision Mamba,增强了心电信号中时间依赖性的表征能力。在PhysioNet 2020与2021计算心脏病学挑战赛数据集上进行了全面实验,性能优于现有方法。具体而言,所提模型在十二导联心电图上的AUPRC和AUROC得分显著高于此前最佳算法报告结果。这些结果表明基于Mamba的架构具有推动可靠心电图分类的潜力,有助于早期诊断、个性化治疗,并提升远程医疗与资源匮乏医疗体系中的可及性。
原文摘要 · Abstract (English)
Accurate detection of cardiac abnormalities from electrocardiogram recordings is regarded as essential for clinical diagnostics and decision support. Traditional deep learning models such as residual networks and transformer architectures have been applied successfully to this task, but their performance has been limited when long sequential signals are processed. Recently, state space models have been introduced as an efficient alternative. In this study, a hybrid framework named One Dimensional Convolutional Neural Network Electrocardiogram Mamba is introduced, in which convolutional feature extraction is combined with Mamba, a selective state space model designed for effective sequence modeling. The model is built upon Vision Mamba, a bidirectional variant through which the representation of temporal dependencies in electrocardiogram data is enhanced. Comprehensive experiments on the PhysioNet Computing in Cardiology Challenges of 2020 and 2021 were conducted, and superior performance compared with existing methods was achieved. Specifically, the proposed model achieved substantially higher AUPRC and AUROC scores than those reported by the best previously published algorithms on twelve lead electrocardiograms. These results demonstrate the potential of Mamba-based architectures to advance reliable ECG classification. This capability supports early diagnosis and personalized treatment, while enhancing accessibility in telemedicine and resource-constrained healthcare systems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。