arXiv:2508.02265cs.CV2025-08中稿 · ECAI 2025被引 1

提出双阈值对比学习框架,提升超声图像分类与分割性能

Semi-Supervised Dual-Threshold Contrastive Learning for Ultrasound Image Classification and Segmentation

  • 设计双阈值机制筛选伪标签,避免错误标签干扰
  • 跨任务注意力模块促进分类与分割信息共享,提升特征一致性
  • 在多个超声数据集上超越主流方法,适合医学图像分析研究者

基于置信度的伪标签选择常因模型早期误判和伪标签过拟合导致过度自信却错误的预测,严重损害半监督对比学习性能。此外,分类与分割任务独立处理,特征关联未充分挖掘。为此,我们提出一种新型半监督双阈值对比学习策略——Hermes,融合对比学习与半监督学习优势,利用伪标签为对比学习提供额外指导。具体地,构建跨任务注意力与显著性模块,促进分类与分割间的知识传递;设计跨任务一致性学习策略,对齐肿瘤特征,减少特征差异,避免负向迁移。为解决公开超声数据集匮乏问题,我们构建了SZ-TUS甲状腺超声图像数据集。在两个公开及一个私有数据集上的大量实验表明,Hermes在多种半监督设置下均持续优于多个前沿方法。

原文摘要 · Abstract (English)

Confidence-based pseudo-label selection usually generates overly confident yet incorrect predictions, due to the early misleadingness of model and overfitting inaccurate pseudo-labels in the learning process, which heavily degrades the performance of semi-supervised contrastive learning. Moreover, segmentation and classification tasks are treated independently and the affinity fails to be fully explored. To address these issues, we propose a novel semi-supervised dual-threshold contrastive learning strategy for ultrasound image classification and segmentation, named Hermes. This strategy combines the strengths of contrastive learning with semi-supervised learning, where the pseudo-labels assist contrastive learning by providing additional guidance. Specifically, an inter-task attention and saliency module is also developed to facilitate information sharing between the segmentation and classification tasks. Furthermore, an inter-task consistency learning strategy is designed to align tumor features across both tasks, avoiding negative transfer for reducing features discrepancy. To solve the lack of publicly available ultrasound datasets, we have collected the SZ-TUS dataset, a thyroid ultrasound image dataset. Extensive experiments on two public ultrasound datasets and one private dataset demonstrate that Hermes consistently outperforms several state-of-the-art methods across various semi-supervised settings.

超声图像半监督对比学习医学分割

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