用心脏波形引导双向混合,提升少标注心电图分割效果
Bidirectional Fusion Guided by Cardiac Patterns for Semi-Supervised ECG Segmentation
- 基于心脏波形模式的双向CutMix增强标签与无标签数据交互
- 在多数据集、不同标注比例下均优于现有方法,最高提升12.3%
- 可插拔设计,适配多种半监督分割模型
准确分割心电图(ECG)中的有意义波形特征对心血管诊断至关重要。然而,标注数据稀缺给深度学习模型训练带来重大挑战。传统半监督语义分割(SemiSeg)方法主要关注无标签数据的一致性,未能充分利用标签与无标签数据间的信息交互。为此,我们提出CardioMix框架,基于心脏波形模式引导的双向CutMix策略用于ECG分割。该方法通过无标签数据生成真实生理变异丰富标签集,同时向无标签数据施加更强监督信号,因心脏波形引导的混合确保所有增强样本保持生理合理性。该框架设计为即插即用模块,在SemiSegECG——一个公开的心电图分割多数据集基准上进行了大量实验,结果表明CardioMix作为兼容多种SemiSeg算法的即插即用模块,在不同数据集和标注比例下始终优于现有基于CutMix的融合策略。
原文摘要 · Abstract (English)
Accurate delineation of electrocardiogram (ECG), the segmentation of meaningful waveform features, is crucial for cardiovascular diagnostics. However, the scarcity of annotated data poses a significant challenge for training deep learning models. Conventional semi-supervised semantic segmentation (SemiSeg) methods primarily focus on consistency from unlabeled data, underutilizing the information exchange possible between labeled and unlabeled sets. To address this, we introduce CardioMix, a framework built on a bidirectional CutMix strategy guided by cardiac patterns for ECG segmentation. This approach enriches the labeled set with realistic variations from unlabeled data while simultaneously applying stronger supervisory signals to the unlabeled set, as the cardiac pattern-guided mixing ensures all augmented samples remain physiologically meaningful. Our framework is designed as a plug-and-play module, demonstrating high compatibility with various SemiSeg algorithms. Extensive experiments on SemiSegECG, a public multi-dataset benchmark for ECG delineation, demonstrate that CardioMix consistently outperforms existing CutMix-based fusion strategies across diverse datasets and labeled ratios as a plug-and-play module compatible with various SemiSeg algorithms.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。