arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2403.09257cs.CV

WSI-SAM:用于组织病理学全切片图像的多分辨率分割一切模型(SAM)

WSI-SAM: Multi-resolution Segment Anything Model (SAM) for histopathology whole-slide images

Eindhoven University of Technology(埃因霍温理工大学);University of Surrey(萨里大学);University Medical Center Utrecht(乌得勒支大学医学中心);University Utrecht(乌得勒支大学)

Hong Liu, Haosen Yang, Paul J. van Diest, Josien P. W. Pluim, Mitko Veta

AI总结:

WSI-SAM通过引入高、低分辨率令牌和双掩码解码器,在冻结SAM基础上以最小参数开销实现多分辨率组织病理学图像分割,在DCIS和CAMELYON16任务上分别超越SAM 4.1和2.5个百分点。

摘要:

分割一切模型(SAM)标志着分割模型领域的重大进步,提供了强大的零样本能力和动态提示功能。然而,现有的医学SAM不适用于全切片图像(WSIs)的多尺度特性,限制了其有效性。为解决这一缺陷,我们提出了WSI-SAM,通过使用多分辨率补丁增强SAM对组织病理学图像的精确对象分割能力,同时保留其高效的、提示驱动的设计和零样本能力。为充分利用预训练知识同时最小化训练开销,我们保持SAM冻结,仅引入极少的额外参数和计算开销。具体而言,我们引入了高分辨率(HR)令牌、低分辨率(LR)令牌和双掩码解码器。该解码器将原始SAM掩码解码器与一个轻量级融合模块相结合,该模块整合多尺度特征。我们不在独立预测掩码,而是在中间层整合HR和LR令牌,以联合学习同一对象在多个分辨率下的特征。实验表明,我们的WSI-SAM优于最先进的SAM及其变体。特别是,我们的模型在导管原位癌(DCIS)分割任务和乳腺癌转移分割任务(CAMELYON16数据集)上分别比SAM高出4.1和2.5个百分点。代码将在https://github.com/HongLiuuuuu/WSI-SAM提供。

英文摘要:

The Segment Anything Model (SAM) marks a significant advancement in segmentation models, offering robust zero-shot abilities and dynamic prompting. However, existing medical SAMs are not suitable for the multi-scale nature of whole-slide images (WSIs), restricting their effectiveness. To resolve this drawback, we present WSI-SAM, enhancing SAM with precise object segmentation capabilities for histopathology images using multi-resolution patches, while preserving its efficient, prompt-driven design, and zero-shot abilities. To fully exploit pretrained knowledge while minimizing training overhead, we keep SAM frozen, introducing only minimal extra parameters and computational overhead. In particular, we introduce High-Resolution (HR) token, Low-Resolution (LR) token and dual mask decoder. This decoder integrates the original SAM mask decoder with a lightweight fusion module that integrates features at multiple scales. Instead of predicting a mask independently, we integrate HR and LR token at intermediate layer to jointly learn features of the same object across multiple resolutions. Experiments show that our WSI-SAM outperforms state-of-the-art SAM and its variants. In particular, our model outperforms SAM by 4.1 and 2.5 percent points on a ductal carcinoma in situ (DCIS) segmentation tasks and breast cancer metastasis segmentation task (CAMELYON16 dataset). The code will be available at https://github.com/HongLiuuuuu/WSI-SAM.

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

↑