arXiv:2605.17875cs.CV2026-05

用二维扫描结构提升12导联心电图多病种诊断准确率

HexagonalWarriorMamba: Superior Threshold-Dependent Multi-label Classification of 12-Lead ECG Cardiac Abnormalities

论文配图:HexagonalWarriorMamba: Superior Threshold-Dependent Multi-label Classification of 12-Lead ECG Cardiac Abnormalities
图 1 · 摘自论文原文
  • 将心电图转为2D图像,用分层扫描机制捕捉长距离依赖
  • 在7国数据集上5项阈值相关指标超越现有最佳方法
  • 适合临床多病种筛查,尤其关注诊断精度与阈值稳定性

从12导联心电图自动诊断心脏异常对心血管疾病管理至关重要。然而,传统深度学习模型难以有效建模心电信号中固有的长程依赖关系,导致并发病症检测能力有限。本文提出HexagonalWarriorMamba(HWMamba),基于Mamba架构,将12导联心电图作为单通道二维图像处理,而非传统的1维时间序列。通过结合分层结构与2维选择性扫描机制,HWMamba可建模全局上下文与复杂空间关系。模型在PhysioNet/Computing in Cardiology Challenge 2021数据集上进行评估,该数据集包含26个诊断标签,涵盖来自四大洲七家机构的记录。结果表明,HWMamba在五项关键阈值依赖指标(包括Challenge Score和Subset Accuracy)上优于当前最优方法,展现出强区分能力与基于训练数据的合理阈值选择,同时保持接近最优的Macro AUROC表现。这种多维度稳定性能,使HWMamba成为一种鲁棒且通用的多标签心电图分类方案。

原文摘要 · Abstract (English)

The accurate automated diagnosis of cardiac abnormalities from 12-lead electrocardiograms (ECGs) is critical for managing cardiovascular disease. However, detecting concurrent conditions remains a challenge for traditional deep learning models, which often have limited ability to model the long-range dependencies inherent in ECG signals. This manuscript proposes HexagonalWarriorMamba (HWMamba), a framework built on the Mamba architecture that processes 12-lead ECGs as single-channel 2D images rather than conventional 1D time series. By integrating a hierarchical architecture with a 2D Selective Scan mechanism, HWMamba is designed to model global context and complex spatial relationships within the data. The model is evaluated on the PhysioNet/Computing in Cardiology Challenge 2021 dataset, which includes 26 diagnostic labels and comprises recordings collected from seven institutions across four countries and three continents. Results demonstrate that HWMamba outperforms current state-of-the-art (SOTA) methods across five key threshold-dependent metrics, including Challenge Score and Subset Accuracy. These improvements provide a balance between strong discriminative capability and effective threshold selection derived from the training data, while maintaining near-SOTA performance in Macro AUROC. This Hexagonal Warrior performance, reflecting consistent performance across multiple evaluation dimensions, positions HWMamba as a robust and versatile approach for multi-label ECG classification.

心电图分析多标签分类Mamba架构

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