arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2512.23347cs.LGcs.AI

ECG-RAMBA: 通过形态-节律解耦和长程建模实现零样本ECG泛化

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

HAI-Smartlink Research Lab, Anchi STE Company, Haiphong, Vietnam(HAI-Smartlink研究实验室,Anchi STE公司,越南海防市);School of Mechanical Engineering, Vietnam Maritime University, Haiphong, Vietnam(机械工程学院,越南海防大学)

Hai Duong Nguyen, Xuan-The Tran

AI总结:

ECG-RAMBA通过形态-节律解耦和长程建模实现零样本ECG泛化,提升跨域鲁棒性。

摘要:

深度学习在单个数据集上实现了对心电图(ECG)分类的强大性能,但可靠地在异质采集设置之间泛化仍然是临床应用和长期监测的主要障碍。许多模型架构的一个关键限制是形态波形模式和节律动态的隐含交织,这可以促进捷径学习并放大对分布变化的敏感性。我们提出了ECG-RAMBA框架,该框架分离形态和节律,然后通过上下文感知融合重新整合它们。ECG-RAMBA结合:(i)通过MiniRocket提取的确定性形态特征,(ii)从心率变异性(HRV)计算的全局节律描述符,以及(iii)通过双向Mamba主干进行的长程上下文建模。为了提高窗口推理下对瞬态异常的敏感性,我们引入了一个数值稳定的Power Mean池化操作符(Q=3),它强调高证据段,同时避免了最大池化和平均的脆弱性和稀释。我们在一个协议忠实的设置下进行评估,包括受试者级交叉验证、固定决策阈值和无测试时适应。在Chapman--Shaoxing数据集上,ECG-RAMBA实现了宏ROC-AUC≈0.85。在零样本迁移中,它在外部CPSC-2021数据集上实现了房颤检测的PR-AUC=0.708,显著优于可比的原始信号Mamba基线,并在PTB-XL上表现出一致的跨数据集性能。消融研究表明,确定性形态提供了强大的基础,而显式节律建模和长程上下文是跨域鲁棒性的关键驱动因素。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑