arXiv:2504.16101eess.SPcs.AI2025-04被引 4

用xLSTM融合频域特征,提升12导联心电图多标签分类准确率

xLSTM-ECG: Multi-label ECG Classification via Feature Fusion with xLSTM

  • 通过STFT将心电信号转为频域,结合xLSTM捕捉局部与全局特征
  • 在PTB-XL数据集上实现高精度多标签分类,且在Georgia 12-Lead上验证泛化能力
  • 首个专为多标签心电图分类设计的xLSTM模型,适合临床辅助诊断场景

心血管疾病仍是全球主要死因,亟需高效精准的诊断工具。心电图(ECG)对多种心脏疾病诊断至关重要,但人工解读耗时且易出错。本文提出xLSTM-ECG,一种基于扩展长短期记忆网络(xLSTM)的多标签心电图分类方法,使用PTB-XL数据集进行验证。据我们所知,这是首个专为多标签心电图分类设计并应用xLSTM模块的工作。该方法采用短时傅里叶变换(STFT)将时序心电信号转换至频域,以增强特征提取。xLSTM架构针对12导联心电图的复杂性,有效捕捉局部与全局信号特征。在PTB-XL数据集上的全面实验表明,模型具备优异的多标签分类性能;在Georgia 12-Lead数据集上的额外测试进一步证明其鲁棒性与效率。该方法显著提升心电图分类准确率,有助于推动临床诊断与患者护理。代码将在论文接受后公开。

原文摘要 · Abstract (English)

Cardiovascular diseases (CVDs) remain the leading cause of mortality worldwide, highlighting the critical need for efficient and accurate diagnostic tools. Electrocardiograms (ECGs) are indispensable in diagnosing various heart conditions; however, their manual interpretation is time-consuming and error-prone. In this paper, we propose xLSTM-ECG, a novel approach that leverages an extended Long Short-Term Memory (xLSTM) network for multi-label classification of ECG signals, using the PTB-XL dataset. To the best of our knowledge, this work represents the first design and application of xLSTM modules specifically adapted for multi-label ECG classification. Our method employs a Short-Time Fourier Transform (STFT) to convert time-series ECG waveforms into the frequency domain, thereby enhancing feature extraction. The xLSTM architecture is specifically tailored to address the complexities of 12-lead ECG recordings by capturing both local and global signal features. Comprehensive experiments on the PTB-XL dataset reveal that our model achieves strong multi-label classification performance, while additional tests on the Georgia 12-Lead dataset underscore its robustness and efficiency. This approach significantly improves ECG classification accuracy, thereby advancing clinical diagnostics and patient care. The code will be publicly available upon acceptance.

心电图分析多标签分类xLSTM深度学习

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