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

3DSAM-adapter:将 SAM 从 2D 到 3D 进行整体适配以实现可提示的肿瘤分割

3DSAM-adapter: Holistic adaptation of SAM from 2D to 3D for promptable tumor segmentation

The Chinese University of Hong Kong(香港中文大学)

Shizhan Gong, Yuan Zhong, Wenao Ma, Jinpeng Li, Zhao Wang, Jingyang Zhang, Pheng-Ann Heng, Qi Dou

AI总结:

本文提出 3DSAM-adapter,通过轻量级空间适配器将 2D SAM 整体扩展至 3D 体数据,在四个肿瘤分割数据集上验证了点击提示分割的先进性能。

摘要:

尽管 Segment Anything Model(SAM)在通用语义分割上取得了令人瞩目的结果,并在日常图像上表现出强大的泛化能力,但其在医学图像分割上所展示的性能不够精确且不稳定,尤其是在处理涉及小尺寸、不规则形状和低对比度目标的肿瘤分割任务时。值得注意的是,原始 SAM 架构是为 2D 自然图像设计的,因此无法有效地从体素医学数据中提取 3D 空间信息。本文提出了一种新颖的适配方法,用于将 SAM 从 2D 迁移到 3D,以实现可提示的医学图像分割。通过整体设计的架构修改方案,我们将 SAM 转换为支持体数据输入,同时保留其大部分预训练参数以便复用。微调过程以参数高效的方式进行:大多数预训练参数保持冻结,仅引入并调整少量轻量级空间适配器。无论自然数据与医学数据之间的领域差距,以及 2D 与 3D 在空间排列上的差异如何,在自然图像上训练的 transformer 只需进行轻量级适配,即可有效捕捉体素医学图像中的空间模式。我们在四个开源肿瘤分割数据集上进行了实验;仅使用一次点击提示,我们的模型在 4 项任务中的 3 项上优于领域内最先进的医学图像分割模型,具体而言,在肾肿瘤、胰腺肿瘤和结肠癌分割上分别提升 8.25%、29.87% 和 10.11%,并在肝肿瘤分割上取得相近性能。我们还将该适配方法与现有流行适配器进行比较,并观察到其在大多数数据集上的显著性能提升。

英文摘要:

Despite that the segment anything model (SAM) achieved impressive results on general-purpose semantic segmentation with strong generalization ability on daily images, its demonstrated performance on medical image segmentation is less precise and not stable, especially when dealing with tumor segmentation tasks that involve objects of small sizes, irregular shapes, and low contrast. Notably, the original SAM architecture is designed for 2D natural images, therefore would not be able to extract the 3D spatial information from volumetric medical data effectively. In this paper, we propose a novel adaptation method for transferring SAM from 2D to 3D for promptable medical image segmentation. Through a holistically designed scheme for architecture modification, we transfer the SAM to support volumetric inputs while retaining the majority of its pre-trained parameters for reuse. The fine-tuning process is conducted in a parameter-efficient manner, wherein most of the pre-trained parameters remain frozen, and only a few lightweight spatial adapters are introduced and tuned. Regardless of the domain gap between natural and medical data and the disparity in the spatial arrangement between 2D and 3D, the transformer trained on natural images can effectively capture the spatial patterns present in volumetric medical images with only lightweight adaptations. We conduct experiments on four open-source tumor segmentation datasets, and with a single click prompt, our model can outperform domain state-of-the-art medical image segmentation models on 3 out of 4 tasks, specifically by 8.25%, 29.87%, and 10.11% for kidney tumor, pancreas tumor, colon cancer segmentation, and achieve similar performance for liver tumor segmentation. We also compare our adaptation method with existing popular adapters, and observed significant performance improvement on most datasets.

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

↑