用无监督提示和偏好优化提升SAM,低标注下实现精准医学图像分割
Enhancing SAM with Efficient Prompting and Preference Optimization for Semi-supervised Medical Image Segmentation
- 通过无监督生成提示,保留语义位置形状信息
- 用虚拟标注员反馈优化策略,仅需简单评分即获高精度分割
- 在肺、肿瘤、器官分割上表现优异,适合标注稀缺场景
基础模型如分割一切模型(SAM)在医学影像分割中日益流行,可支持多种下游任务。然而,这些模型仍依赖大规模标注数据或专家提供的提示。传统主动学习等方法受限于范围,仍需持续的人工参与和复杂的领域知识来修正标签或建立奖励基准。为此,我们提出一种增强型SAM框架,采用完全无监督方式生成注释高效提示,同时通过对比语言-图像预训练和视觉问答捕捉关键语义、位置与形状信息。我们引入直接偏好优化技术,设计最优策略,使模型仅需虚拟标注员提供的简单评分或排序即可生成高保真分割结果。在肺部分割、乳腺肿瘤分割及多种模态(包括X光、超声、腹部CT)下的器官分割任务中,该框架均达到顶尖性能,证明其在低标注数据场景下的有效性。
原文摘要 · Abstract (English)
Foundational models such as the Segment Anything Model (SAM) are gaining traction in medical imaging segmentation, supporting multiple downstream tasks. However, such models are supervised in nature, still relying on large annotated datasets or prompts supplied by experts. Conventional techniques such as active learning to alleviate such limitations are limited in scope and still necessitate continuous human involvement and complex domain knowledge for label refinement or establishing reward ground truth. To address these challenges, we propose an enhanced Segment Anything Model (SAM) framework that utilizes annotation-efficient prompts generated in a fully unsupervised fashion, while still capturing essential semantic, location, and shape information through contrastive language-image pretraining and visual question answering. We adopt the direct preference optimization technique to design an optimal policy that enables the model to generate high-fidelity segmentations with simple ratings or rankings provided by a virtual annotator simulating the human annotation process. State-of-the-art performance of our framework in tasks such as lung segmentation, breast tumor segmentation, and organ segmentation across various modalities, including X-ray, ultrasound, and abdominal CT, justifies its effectiveness in low-annotation data scenarios.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。