arXiv:2504.09876cs.CVcs.AI2025-04CVPR被引 2

用分层蒸馏提升胎儿超声分割的精度与效率

HDC: Hierarchical Distillation for Multi-level Noisy Consistency in Semi-Supervised Fetal Ultrasound Segmentation

  • 单教师架构+分层蒸馏,降低模型复杂度
  • 在FUGC和PSFH数据集上性能优于多教师模型
  • 适合标注稀缺的医学图像分割场景

经阴道超声是评估宫颈解剖结构和检测生理变化的关键影像手段,但宫颈结构分割因对比度低、阴影伪影和边界模糊而困难。尽管卷积神经网络在医学图像分割中表现良好,但其对大规模标注数据的依赖限制了在临床超声中的应用。半监督学习可通过未标注数据缓解此问题,但现有师生框架常出现确认偏差且计算成本高。本文提出HDC框架,采用单教师架构结合自适应一致性学习,引入分层蒸馏机制:相关性引导损失用于对齐特征表示,互信息损失用于稳定噪声学生学习。该方法在保持高性能的同时降低模型复杂度。在胎儿超声数据集FUGC和PSFH上的实验表明,HDC相较多教师模型具备更优性能与更低计算开销。

原文摘要 · Abstract (English)

Transvaginal ultrasound is a critical imaging modality for evaluating cervical anatomy and detecting physiological changes. However, accurate segmentation of cervical structures remains challenging due to low contrast, shadow artifacts, and indistinct boundaries. While convolutional neural networks (CNNs) have demonstrated efficacy in medical image segmentation, their reliance on large-scale annotated datasets presents a significant limitation in clinical ultrasound imaging. Semi-supervised learning (SSL) offers a potential solution by utilizing unlabeled data, yet existing teacher-student frameworks often encounter confirmation bias and high computational costs. In this paper, a novel semi-supervised segmentation framework, called HDC, is proposed incorporating adaptive consistency learning with a single-teacher architecture. The framework introduces a hierarchical distillation mechanism with two objectives: Correlation Guidance Loss for aligning feature representations and Mutual Information Loss for stabilizing noisy student learning. The proposed approach reduces model complexity while enhancing generalization. Experiments on fetal ultrasound datasets, FUGC and PSFH, demonstrate competitive performance with reduced computational overhead compared to multi-teacher models.

医学图像半监督分割

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