arXiv:2602.23060cs.LG2026-02被引 3

将心电图视为语言,用自监督模型识别心脏疾病。

RhythmBERT: A Self-Supervised Language Model Based on Latent Representations of ECG Waveforms for Heart Disease Detection

  • 把心电图波段转为符号化令牌,结合离散节奏与连续形态特征
  • 仅用单导联数据性能媲美12导联模型,对房颤等病种有效
  • 适合想用少量标注数据做心脏病检测的研究者

心电图分析对诊断心脏病至关重要,但现有自监督学习方法多将心电图视作通用时间序列,忽略其生理语义和节律结构。传统对比方法使用扭曲波形的增强,生成式方法则采用固定窗口分割,导致心脏周期错位。为此,我们提出RhythmBERT,一种基于自动编码器潜空间表示的心电图生成语言模型,将P、QRS、T波段编码为符号令牌,捕捉节律语义;同时保留连续嵌入以保持精细波形结构,实现波形与节律的统一表征。RhythmBERT在约80万条无标签心电图上预训练,采用掩码预测目标,实现标签高效的上下文表征学习。评估显示,尽管仅使用单导联数据,其性能仍可媲美甚至超越强基准的12导联模型,涵盖房颤等常见病及亚临床ST-T异常、心肌梗死等疑难病例。结果表明,将心电图视为结构化语言,是一种可扩展且符合生理规律的心脏病分析新路径。

原文摘要 · Abstract (English)

Electrocardiogram (ECG) analysis is crucial for diagnosing heart disease, but most self-supervised learning methods treat ECG as a generic time series, overlooking physiologic semantics and rhythm-level structure. Existing contrastive methods utilize augmentations that distort morphology, whereas generative approaches employ fixed-window segmentation, which misaligns cardiac cycles. To address these limitations, we propose RhythmBERT, a generative ECG language model that considers ECG as a language paradigm by encoding P, QRS, and T segments into symbolic tokens via autoencoder-based latent representations. These discrete tokens capture rhythm semantics, while complementary continuous embeddings retain fine-grained morphology, enabling a unified view of waveform structure and rhythm. RhythmBERT is pretrained on approximately 800,000 unlabeled ECG recordings with a masked prediction objective, allowing it to learn contextual representations in a label-efficient manner. Evaluations show that despite using only a single lead, RhythmBERT achieves comparable or superior performance to strong 12-lead baselines. This generalization extends from prevalent conditions such as atrial fibrillation to clinically challenging cases such as subtle ST-T abnormalities and myocardial infarction. Our results suggest that considering ECG as structured language offers a scalable and physiologically aligned pathway for advancing cardiac analysis.

心电图分析自监督学习语言模型心脏病检测

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