arXiv:2512.23347cs.LGcs.AI2025-12被引 1

分离心电图形态与节律特征,提升跨数据集泛化能力。

ECG-RAMBA: Zero-Shot ECG Generalization by Morphology-Rhythm Disentanglement and Long-Range Modeling

  • 解耦形态与节律特征,通过上下文融合重建信号
  • 零样本迁移下房颤检测PR-AUC达0.708,优于基线
  • 适合临床部署和长期监测场景的鲁棒模型

深度学习在单一数据集上对心电图分类表现优异,但在异构采集条件下仍缺乏可靠泛化能力。现有模型常隐式耦合波形形态与节律动态,导致捷径学习并加剧分布偏移敏感性。本文提出ECG-RAMBA框架,通过分离形态与节律特征,并采用上下文感知融合实现重构。该方法结合:(i) MiniRocket提取的确定性形态特征,(ii) 心率变异性(HRV)计算的全局节律描述符,以及 (iii) 双向Mamba架构实现的长程上下文建模。为增强窗口推理中对瞬态异常的敏感性,引入数值稳定的Power Mean池化(Q=3),在保留高置信度片段的同时避免最大池化的脆弱性与平均池化的稀释效应。在符合协议的跨受试者验证设置下,使用固定决策阈值且无测试时自适应评估。在Chapman--Shaoxing数据集上,宏平均ROC-AUC≈0.85;在外部CPSC-2021数据集上零样本迁移检测房颤,达到PR-AUC=0.708,显著优于同类原始信号Mamba基线,并在PTB-XL上保持一致跨数据集性能。消融实验表明,确定性形态特征提供坚实基础,而显式节律建模与长程上下文是跨域鲁棒性的关键驱动因素。

原文摘要 · Abstract (English)

Deep learning has achieved strong performance for electrocardiogram (ECG) classification within individual datasets, yet dependable generalization across heterogeneous acquisition settings remains a major obstacle to clinical deployment and longitudinal monitoring. A key limitation of many model architectures is the implicit entanglement of morphological waveform patterns and rhythm dynamics, which can promote shortcut learning and amplify sensitivity to distribution shifts. We propose ECG-RAMBA, a framework that separates morphology and rhythm and then re-integrates them through context-aware fusion. ECG-RAMBA combines: (i) deterministic morphological features extracted by MiniRocket, (ii) global rhythm descriptors computed from heart-rate variability (HRV), and (iii) long-range contextual modeling via a bi-directional Mamba backbone. To improve sensitivity to transient abnormalities under windowed inference, we introduce a numerically stable Power Mean pooling operator ($Q=3$) that emphasizes high-evidence segments while avoiding the brittleness of max pooling and the dilution of averaging. We evaluate under a protocol-faithful setting with subject-level cross-validation, a fixed decision threshold, and no test-time adaptation. On the Chapman--Shaoxing dataset, ECG-RAMBA achieves a macro ROC-AUC $\approx 0.85$. In zero-shot transfer, it attains PR-AUC $=0.708$ for atrial fibrillation detection on the external CPSC-2021 dataset, substantially outperforming a comparable raw-signal Mamba baseline, and shows consistent cross-dataset performance on PTB-XL. Ablation studies indicate that deterministic morphology provides a strong foundation, while explicit rhythm modeling and long-range context are critical drivers of cross-domain robustness.

心电图分析零样本迁移模型鲁棒性

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