用双向状态空间模型实现心电图多导联高效融合,兼顾精度与轻量化。
S2M2ECG: Spatio-temporal bi-directional State Space Model Enabled Multi-branch Mamba for ECG
- 设计三层融合机制:时序双向建模、导联内信息增强、导联间空间交互。
- 在多个心电图任务上性能超越现有模型,参数量最少,适合部署。
- 专为心电图多导联特性优化,适合医疗边缘设备实时诊断。
作为心血管疾病(CVD)诊断最有效的方法之一,多导联心电图(ECG)信号呈现出典型的多传感器信息融合挑战,长期受到深度学习领域的关注。尽管已有多种深度学习架构被提出,但如何在性能、计算复杂度和多源心电特征融合之间取得平衡仍具挑战。最近,状态空间模型(SSM)特别是Mamba,在多个领域展现出卓越效果,其固有的高效率计算和线性复杂度使其特别适合低维数据如心电图。本文提出S2M2ECG,一种具有三级融合机制的SSM架构:(1) 基于分段标记的时空双向SSM用于低层信号融合;(2) 导联内双向扫描的时间信息融合以提升正向与反向识别准确率;(3) 跨导联特征交互模块实现空间信息融合。为充分利用心电图信号的多导联特性,引入多分支设计与导联融合模块,支持各导联独立分析并实现无缝整合。实验结果表明,S2M2ECG在节律、形态和临床场景中均表现优异,且其轻量化架构参数量最少,极适合高效推理与便捷部署。总体而言,S2M2ECG在性能、计算复杂度与心电图特性之间达到良好平衡,为心血管疾病诊断中的高性能轻量计算提供新路径。
原文摘要 · Abstract (English)
As one of the most effective methods for cardiovascular disease (CVD) diagnosis, multi-lead Electrocardiogram (ECG) signals present a characteristic multi-sensor information fusion challenge that has been continuously researched in deep learning domains. Despite the numerous algorithms proposed with different DL architectures, maintaining a balance among performance, computational complexity, and multi-source ECG feature fusion remains challenging. Recently, state space models (SSMs), particularly Mamba, have demonstrated remarkable effectiveness across various fields. Their inherent design for high-efficiency computation and linear complexity makes them particularly suitable for low-dimensional data like ECGs. This work proposes S2M2ECG, an SSM architecture featuring three-level fusion mechanisms: (1) Spatio-temporal bi-directional SSMs with segment tokenization for low-level signal fusion, (2) Intra-lead temporal information fusion with bi-directional scanning to enhance recognition accuracy in both forward and backward directions, (3) Cross-lead feature interaction modules for spatial information fusion. To fully leverage the ECG-specific multi-lead mechanisms inherent in ECG signals, a multi-branch design and lead fusion modules are incorporated, enabling individual analysis of each lead while ensuring seamless integration with others. Experimental results reveal that S2M2ECG achieves superior performance in the rhythmic, morphological, and clinical scenarios. Moreover, its lightweight architecture ensures it has nearly the fewest parameters among existing models, making it highly suitable for efficient inference and convenient deployment. Collectively, S2M2ECG offers a promising alternative that strikes an excellent balance among performance, computational complexity, and ECG-specific characteristics, paving the way for high-performance, lightweight computations in CVD diagnosis.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。