arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.07042cs.CVcs.AI

ProtoSAM:基于基础模型的一次性医学图像分割

ProtoSAM: One-Shot Medical Image Segmentation With Foundational Models

Faculty of Engineering, Tel Aviv University(特拉维夫大学工程学院);Icahn school of Medicine, Mount Sinai(西奈山伊坎医学院)

Lev Ayzenberg, Raja Giryes, Hayit Greenspan

AI总结:

提出 ProtoSAM 框架,结合原型网络与 SAM 基础模型,实现一次性医学图像分割,无需微调,在多个数据集上达到最先进性能。

摘要:

本文介绍了一种新的框架 ProtoSAM,用于一次性医学图像分割。它结合了以少样本分割著称的原型网络与 SAM(一种自然图像基础模型)。所提出的方法使用 ALPnet 原型网络(增强有 DINOv2 编码器)生成初始粗略分割掩码。在提取初始掩码后,提取提示(如点和边界框),然后输入到 Segment Anything Model (SAM) 中。在多个医学图像数据集上展示了最先进的结果,并展示了使用单个图像示例(一次性)进行自动化分割的能力,无需对基础模型进行微调。我们的代码可在 https://github.com/levayz/ProtoSAM 获取。

英文摘要:

This work introduces a new framework, ProtoSAM, for one-shot medical image segmentation. It combines the use of prototypical networks, known for few-shot segmentation, with SAM - a natural image foundation model. The method proposed creates an initial coarse segmentation mask using the ALPnet prototypical network, augmented with a DINOv2 encoder. Following the extraction of an initial mask, prompts are extracted, such as points and bounding boxes, which are then input into the Segment Anything Model (SAM). State-of-the-art results are shown on several medical image datasets and demonstrate automated segmentation capabilities using a single image example (one shot) with no need for fine-tuning of the foundation model. Our code is available at: https://github.com/levayz/ProtoSAM

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑