摘要
arXiv:2505.05538v1 交叉类型公告
摘要:心电图(ECG)分类对于自动心脏疾病诊断至关重要,但现有方法往往难以同时捕捉局部形态细节和长时序依赖性。为了解决这些问题,我们提出了一种名为Cardioformer的新颖多功能混合模型,该模型结合了跨通道补丁、分层残差学习和两阶段自注意力机制。Cardioformer首先编码多尺度标记嵌入以捕获细粒度局部特征和全局上下文信息,然后通过跨内粒度和跨粒度自注意力选择性地融合这些表示。在受试者独立设置下对三个标准ECG数据集进行广泛评估表明,该模型一致地超越了四种当前最先进的baseline模型。我们的Cardioformer模型分别在MIMIC-IV、PTB-XL和PTB数据集中达到了AUROC为96.34±0.11、89.99±0.12和95.59±1.66的成绩,优于PatchTST、Reformer、Transformer和Medformer模型。此外,Cardioformer还展示了强大的跨数据集泛化能力,在使用MIMIC-IV训练时,其在PTB数据集上的AUROC为49.18%,在PTB-XL上的AUROC为68.41%。这些发现凸显了Cardioformer在促进自动化ECG分析方面的能力,为更准确和稳健的心血管疾病诊断铺平了道路。我们已在https://github.com/KMobin555/Cardioformer发布了源代码。