arXiv:2505.05538cs.LGcs.AI2025-05被引 3

Cardioformer通过多粒度分块提升心电图分析精度,兼顾局部细节与长程依赖。

Cardioformer: Advancing AI in ECG Analysis with Multi-Granularity Patching and ResNet

  • 采用跨通道分块与分层残差学习捕捉多尺度特征
  • 在三个数据集上实现最高96.34%的AUROC性能
  • 适合医疗AI研究者和临床辅助诊断系统开发者

心电图分类对自动化心脏疾病诊断至关重要,但现有方法难以同时捕捉局部形态细节与长程时间依赖。为此,我们提出Cardioformer,一种融合跨通道分块、分层残差学习与两阶段自注意力机制的新型多粒度混合模型。该模型首先生成多尺度标记嵌入以捕获细粒度局部特征与全局上下文信息,再通过内部与跨粒度自注意力选择性融合表示。在三种基准心电图数据集(MIMIC-IV、PTB-XL、PTB)的受试者独立设置下进行广泛评估,结果表明其持续优于四种先进基线模型。Cardioformer在MIMIC-IV、PTB-XL和PTB数据集上的AUROC分别达到96.34±0.11、89.99±0.12和95.59±1.66,显著超越PatchTST、Reformer、Transformer和Medformer。此外,其跨数据集泛化能力突出,在仅用MIMIC-IV训练时,于PTB和PTB-XL上分别取得49.18%与68.41%的AUROC。这些结果凸显Cardioformer在推动自动化心电图分析方面的潜力,为更精准、鲁棒的心血管疾病诊断铺平道路。源代码已开源:https://github.com/KMobin555/Cardioformer。

原文摘要 · Abstract (English)

Electrocardiogram (ECG) classification is crucial for automated cardiac disease diagnosis, yet existing methods often struggle to capture local morphological details and long-range temporal dependencies simultaneously. To address these challenges, we propose Cardioformer, a novel multi-granularity hybrid model that integrates cross-channel patching, hierarchical residual learning, and a two-stage self-attention mechanism. Cardioformer first encodes multi-scale token embeddings to capture fine-grained local features and global contextual information and then selectively fuses these representations through intra- and inter-granularity self-attention. Extensive evaluations on three benchmark ECG datasets under subject-independent settings demonstrate that model consistently outperforms four state-of-the-art baselines. Our Cardioformer model achieves the AUROC of 96.34$\pm$0.11, 89.99$\pm$0.12, and 95.59$\pm$1.66 in MIMIC-IV, PTB-XL and PTB dataset respectively outperforming PatchTST, Reformer, Transformer, and Medformer models. It also demonstrates strong cross-dataset generalization, achieving 49.18% AUROC on PTB and 68.41% on PTB-XL when trained on MIMIC-IV. These findings underscore the potential of Cardioformer to advance automated ECG analysis, paving the way for more accurate and robust cardiovascular disease diagnosis. We release the source code at https://github.com/KMobin555/Cardioformer.

心电图分析自注意力多粒度医疗AI

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