用双向Mamba+光谱时间嵌入,提升脑电心电分类效率与准确率
BioMamba: Leveraging Spectro-Temporal Embedding in Bidirectional Mamba for Enhanced Biosignal Classification
- 引入光谱时间嵌入与稀疏前馈的双向Mamba架构
- 在六项指标上优于当前最优方法,模型更小更省资源
- 适合临床信号分析,兼顾性能、速度与泛化能力
脑电图(EEG)和心电图(ECG)等生物信号在临床诊断脑部及心律失常疾病中至关重要。现有分类方法依赖基于注意力机制的密集前馈网络,导致学习效率低、计算开销大且性能不佳。本文提出BioMamba,将光谱-时间嵌入应用于双向Mamba框架,并采用稀疏前馈层,以实现对生物信号序列的有效建模。通过整合三者,BioMamba显著克服了现有方法的局限。大量实验表明,BioMamba在分类性能上显著优于现有最先进方法。其优势包括:(1)可靠性:在六项评估指标上表现稳定;(2)效率:模型规模更小,资源消耗更低;(3)通用性:可有效处理多种任务,在不同领域均具适应性与有效性。
原文摘要 · Abstract (English)
Biological signals, such as electroencephalograms (EEGs) and electrocardiograms (ECGs), play a pivotal role in numerous clinical practices, such as diagnosing brain and cardiac arrhythmic diseases. Existing methods for biosignal classification rely on Attention-based frameworks with dense Feed Forward layers, which lead to inefficient learning, high computational overhead, and suboptimal performance. In this work, we introduce BioMamba, a Spectro-Temporal Embedding strategy applied to the Bidirectional Mamba framework with Sparse Feed Forward layers to enable effective learning of biosignal sequences. By integrating these three key components, BioMamba effectively addresses the limitations of existing methods. Extensive experiments demonstrate that BioMamba significantly outperforms state-of-the-art methods with marked improvement in classification performance. The advantages of the proposed BioMamba include (1) Reliability: BioMamba consistently delivers robust results, confirmed across six evaluation metrics. (2) Efficiency: We assess both model and training efficiency, the BioMamba demonstrates computational effectiveness by reducing model size and resource consumption compared to existing approaches. (3) Generality: With the capacity to effectively classify a diverse set of tasks, BioMamba demonstrates adaptability and effectiveness across various domains and applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。