用视觉语言知识提升显微图像分割泛化能力
Unifying Segment Anything in Microscopy with Vision-Language Knowledge
- 通过视觉语言对齐模块注入语义提示,指导SAM模型学习
- 跨域数据上准确率提升9.2%,在9个领域内数据集上提升11.8%
- 适合需要强泛化能力的医学图像分割场景
生物医学图像中感兴趣区域的精确分割具有重要价值。尽管现有基础分割模型在特定数据集上表现优异,但在未见领域数据上性能不佳,根源在于分割前缺乏视觉-语言知识。多模态大语言模型(MLLM)具备出色的多模态理解与推理能力,启发我们利用其注入视觉-语言知识(VLK),以增强视觉模型在跨域数据上的泛化能力。本文提出uLLSAM框架,通过MLLM无缝引导SAM学习显微图像跨域数据,实现显微图像分割的统一。具体提出视觉-语言语义对齐(VLSA)模块,将VLK注入分割任意模型(SAM)。发现接收全局VLK提示后,SAM性能显著提升,但边界感知仍不足,因此进一步引入语义边界正则化(SBR)进行优化。该方法在9个领域内显微图像数据集上平均分割准确率(SA)提升11.8%,在10个领域外数据集上提升9.2%,达到当前最优水平。代码已开源。
原文摘要 · Abstract (English)
Accurate segmentation of regions of interest in biomedical images holds substantial value in image analysis. Although several foundation models for biomedical segmentation have currently achieved excellent performance on certain datasets, they typically demonstrate sub-optimal performance on unseen domain data. We owe the deficiency to lack of vision-language knowledge before segmentation. Multimodal Large Language Models (MLLMs) bring outstanding understanding and reasoning capabilities to multimodal tasks, which inspires us to leverage MLLMs to inject Vision-Language Knowledge (VLK), thereby enabling vision models to demonstrate superior generalization capabilities on cross-domain datasets. In this paper, we propose a novel framework that seamlessly uses MLLMs to guide SAM in learning microscopy cross-domain data, unifying Segment Anything in Microscopy, named uLLSAM. Specifically, we propose the Vision-Language Semantic Alignment (VLSA) module, which injects VLK into Segment Anything Model (SAM). We find that after SAM receives global VLK prompts, its performance improves significantly, but there are deficiencies in boundary contour perception. Therefore, we further propose Semantic Boundary Regularization (SBR) to regularize SAM. Our method achieves performance improvements of 11.8% in SA across 9 in-domain microscopy datasets, achieving state-of-the-art performance. Our method also demonstrates improvements of 9.2% in SA across 10 out-of-domain datasets, exhibiting strong generalization capabilities. Code is available at https://github.com/ieellee/uLLSAM.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。