arXiv:2410.08559cs.LGcs.AI2024-10被引 22

用隐空间预测替代原始信号重建,提升心电图自监督学习效果

Learning General Representation of 12-Lead Electrocardiogram with a Joint-Embedding Predictive Architecture

  • 在隐空间中进行掩码建模,避免重建噪声和原始信号误差
  • 在约18万样本上训练,下游任务表现达当前最优
  • 适合做心电图特征提取与诊断分类的研究者使用

心电图(ECG)捕捉心脏电活动,对心脏病诊断至关重要。但标注数据稀缺限制了监督学习的应用。自监督学习(SSL)提供了一种从无标签数据中学习的有效途径。本文提出一种基于隐空间掩码建模的SSL方法——ECG-JEPA,通过预测隐藏表示而非重建原始信号来学习12导联心电图的语义表征。该方法避免了生成无关噪声,并克服了原始信号间简单L2损失的局限性。核心创新包括专为12导联设计的跨模式注意力机制(CroPA)。模型在多个公开心电图数据集合并的约18万样本上训练,显著优于现有方法,在诊断分类、特征提取和分割等下游任务中均达到领先性能。代码已开源。

原文摘要 · Abstract (English)

Electrocardiogram (ECG) captures the heart's electrical signals, offering valuable information for diagnosing cardiac conditions. However, the scarcity of labeled data makes it challenging to fully leverage supervised learning in the medical domain. Self-supervised learning (SSL) offers a promising solution, enabling models to learn from unlabeled data and uncover meaningful patterns. In this paper, we show that masked modeling in the latent space can be a powerful alternative to existing self-supervised methods in the ECG domain. We introduce ECG-JEPA, an SSL model for 12-lead ECG analysis that learns semantic representations of ECG data by predicting in the hidden latent space, bypassing the need to reconstruct raw signals. This approach offers several advantages in the ECG domain: (1) it avoids producing unnecessary details, such as noise, which is common in ECG; and (2) it addresses the limitations of naive L2 loss between raw signals. Another key contribution is the introduction of Cross-Pattern Attention (CroPA), a specialized masked attention mechanism tailored for 12-lead ECG data. ECG-JEPA is trained on the union of several open ECG datasets, totaling approximately 180,000 samples, and achieves state-of-the-art performance in various downstream tasks including diagnostic classification, feature extraction, and segmentation. Our code is openly available at https://github.com/sehunfromdaegu/ECG_JEPA.

心电图自监督学习隐空间建模多导联

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