用多模态信号和新损失函数提升儿科心律失常少样本分类准确率
Advancing Few-Shot Pediatric Arrhythmia Classification with a Novel Contrastive Loss and Multimodal Learning
- 融合体表与心内电图信号,通过注意力机制实现跨模态特征融合
- 在莱比锡心脏中心儿科数据集上达96.22%准确率,少数类性能显著提升
- 适合医学影像与少样本学习研究者,尤其关注罕见病分类场景
心律失常是儿童突发性心脏死亡的主要原因,因此从心电图(ECG)自动分类心律具有重要临床意义。然而,由于年龄相关的波形差异、数据量有限以及显著的长尾类别分布,儿科心律失常分析仍具挑战性。为此,我们提出一种端到端多模态框架,整合体表ECG与心内电图(IEGM)信号进行儿科心律失常分类。模型采用双分支特征编码器、基于注意力的跨模态融合及轻量级Transformer分类器,学习互补的电生理表征。进一步引入自适应全局类别感知对比损失(AGCACL),结合原型对齐、类别频率重加权与全局引导的硬类别调制,增强类别内紧凑性与类别间可分性。我们在莱比锡心脏中心心电数据库的儿科子集上评估该方法,并建立可复现的预处理流程,包括心律片段构建、去噪与标签分组。所提方法在Top-1准确率达96.22%,相比最强基线,宏精确率、宏召回率、宏F1分数与宏F2分数分别提升4.48、1.17、6.98与7.34个百分点。结果表明,在当前基准上实现了更优的少数类敏感分类性能。但需在个体独立与多中心设置下进一步验证,方可推进临床转化。
原文摘要 · Abstract (English)
Arrhythmias are a major cause of sudden cardiac death in children, making automated rhythm classification from electrocardiograms (ECGs) clinically important. However, pediatric arrhythmia analysis remains challenging because of age-dependent waveform variability, limited data availability, and a pronounced long-tailed class distribution that hinders recognition of rare but clinically important rhythms. To address these issues, we propose a multimodal end-to-end framework that integrates surface ECG and intracardiac electrogram (IEGM) signals for pediatric arrhythmia classification. The model combines dual-branch feature encoders, attention-based cross-modal fusion, and a lightweight Transformer classifier to learn complementary electrophysiological representations. We further introduce an Adaptive Global Class-Aware Contrastive Loss (AGCACL), which incorporates prototype-based alignment, class-frequency reweighting, and globally informed hard-class modulation to improve intra-class compactness and inter-class separability under class imbalance. We evaluate the proposed method on the pediatric subset of the Leipzig Heart Center ECG-Database and establish a reproducible preprocessing pipeline including rhythm-segment construction, denoising, and label grouping. The proposed approach achieves 96.22% Top-1 accuracy and improves macro precision, macro recall, macro F1 score, and macro F2 score by 4.48, 1.17, 6.98, and 7.34 percentage points, respectively, over the strongest baseline. These results indicate improved minority-sensitive classification performance on the current benchmark. However, further validation under subject-independent and multicenter settings is still required before clinical translation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。