arXiv:2503.06816eess.IVcs.AI2025-03被引 4

用大模型知识增强小模型,少标注也能高效做医学图像分割

Semi-Supervised Medical Image Segmentation via Knowledge Mining from Large Models

  • 从大模型生成伪标签,补全小模型训练数据
  • 在两个医疗数据集上分别提升Dice 3%和1%
  • 适合资源有限但需高精度分割的临床场景

大型视觉模型如SAM具备丰富的通用视觉知识,但其泛化性与计算开销限制了在医学图像分割等专业任务中的应用。而专用模型如U-Net++因标注数据稀疏表现不佳。本文提出一种策略性知识挖掘方法:在少量标注数据上训练U-Net++,再利用SAM对未标注图像推理生成提示,反向优化SAM并为小模型提供伪标签。该过程迭代增强模型性能,使本地网络在相同标注量(Kvasir SEG为75%,COVID-QU-Ex为50%)下,相比基线U-Net++分别提升Dice 3%和1%。即使基线使用100%标注数据,本方法仍表现更优。结果表明,通过挖掘SAM的广泛知识,可有效缓解专用模型的数据瓶颈,同时保障临床部署所需的效率。

原文摘要 · Abstract (English)

Large-scale vision models like SAM have extensive visual knowledge, yet their general nature and computational demands limit their use in specialized tasks like medical image segmentation. In contrast, task-specific models such as U-Net++ often underperform due to sparse labeled data. This study introduces a strategic knowledge mining method that leverages SAM's broad understanding to boost the performance of small, locally hosted deep learning models. In our approach, we trained a U-Net++ model on a limited labeled dataset and extend its capabilities by converting SAM's output infered on unlabeled images into prompts. This process not only harnesses SAM's generalized visual knowledge but also iteratively improves SAM's prediction to cater specialized medical segmentation tasks via U-Net++. The mined knowledge, serving as "pseudo labels", enriches the training dataset, enabling the fine-tuning of the local network. Applied to the Kvasir SEG and COVID-QU-Ex datasets which consist of gastrointestinal polyp and lung X-ray images respectively, our proposed method consistently enhanced the segmentation performance on Dice by 3% and 1% respectively over the baseline U-Net++ model, when the same amount of labelled data were used during training (75% and 50% of labelled data). Remarkably, our proposed method surpassed the baseline U-Net++ model even when the latter was trained exclusively on labeled data (100% of labelled data). These results underscore the potential of knowledge mining to overcome data limitations in specialized models by leveraging the broad, albeit general, knowledge of large-scale models like SAM, all while maintaining operational efficiency essential for clinical applications.

医学图像分割知识挖掘半监督学习

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