轻量级心电图模型可精准识别多种心脏病,助力临床诊断。
FoundationalECGNet: A Lightweight Foundational Model for ECG-based Multitask Cardiac Analysis
- 融合小波去噪与注意力机制,同时捕捉心电信号时空特征。
- 正常与异常心电图分类F1达99%,多种疾病识别效果领先。
- 适合医疗AI研究者及临床辅助诊断系统开发者使用。
心血管疾病仍是全球主要死亡原因,准确高效的诊断系统至关重要。心电图(ECG)分析在检测心脏异常中起核心作用,但噪声、类别不平衡和数据集异质性等问题限制了现有方法性能。为此,我们提出FoundationalECGNet,一种用于自动化心电图多任务分类的基础模型。该模型结合莫莱特与达布歇小波双重去噪、卷积块注意力模块(CBAM)、图注意力网络(GAT)和时间序列变换器(TST),联合建模多导联心电信号的时空依赖关系。模型首先区分正常与异常心电图,再将异常信号分类为五类心脏疾病:心律失常、传导障碍、心肌梗死、QT异常或肥厚。在多个数据集上,正常/异常分类达到99% F1分数,多分类任务表现达到当前最优水平,其中传导障碍和肥厚的F1均为99%,心律失常达98.9%。此外,模型还能提供风险等级评估,辅助临床决策。综上,FoundationalECGNet是一种可扩展、可解释且通用的心电图分析方案,有望提升诊断精度与患者预后。代码将在论文接收后公开。
原文摘要 · Abstract (English)
Cardiovascular diseases (CVDs) remain a leading cause of mortality worldwide, underscoring the importance of accurate and scalable diagnostic systems. Electrocardiogram (ECG) analysis is central to detecting cardiac abnormalities, yet challenges such as noise, class imbalance, and dataset heterogeneity limit current methods. To address these issues, we propose FoundationalECGNet, a foundational framework for automated ECG classification. The model integrates a dual-stage denoising by Morlet and Daubechies wavelets transformation, Convolutional Block Attention Module (CBAM), Graph Attention Networks (GAT), and Time Series Transformers (TST) to jointly capture spatial and temporal dependencies in multi-channel ECG signals. FoundationalECGNet first distinguishes between Normal and Abnormal ECG signals, and then classifies the Abnormal signals into one of five cardiac conditions: Arrhythmias, Conduction Disorders, Myocardial Infarction, QT Abnormalities, or Hypertrophy. Across multiple datasets, the model achieves a 99% F1-score for Normal vs. Abnormal classification and shows state-of-the-art performance in multi-class disease detection, including a 99% F1-score for Conduction Disorders and Hypertrophy, as well as a 98.9% F1-score for Arrhythmias. Additionally, the model provides risk level estimations to facilitate clinical decision-making. In conclusion, FoundationalECGNet represents a scalable, interpretable, and generalizable solution for automated ECG analysis, with the potential to improve diagnostic precision and patient outcomes in healthcare settings. We'll share the code after acceptance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。