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

SU-SAM:在表现不佳场景中适配 Segment Anything Model 的简单统一框架

SU-SAM: A Simple Unified Framework for Adapting Segment Anything Model in Underperformed Scenes

Shanghai Jiao Tong University(上海交通大学);La Trobe University(拉筹伯大学)

Yiran Song, Qianyu Zhou, Xuequan Lu, Zhiwen Shao, Lizhuang Ma

AI总结:

针对SAM在专门数据上泛化能力不足的问题,提出简单统一的参数高效微调框架SU-SAM,通过四种变体结构显著提升模型在表现不佳场景下的适应性,在多个下游任务上取得优异性能。

摘要:

Segment anything model (SAM) 在常见视觉场景中展现出了卓越的泛化能力,但在理解专门数据方面能力不足。近期,一些方法将参数高效技术与任务特定设计相结合,以在特定任务上微调 SAM。然而,这些方法严重依赖手工制作、复杂且任务特定的设计以及前/后处理,才能在下游任务中达到可接受的性能。因此,这严重限制了其在其他下游任务上的泛化能力。为了解决这一问题,我们提出了一个简单且统一的框架,即 SU-SAM,该框架能够利用参数高效技术轻松高效地微调 SAM 模型,同时保持对各种下游任务的出色泛化能力。SU-SAM 不需要任何任务特定的设计,旨在显著提升类 SAM 模型对表现不佳场景的适应性。具体而言,我们将不同方法的参数高效模块抽象为框架中的基本设计元素。此外,我们提出了 SU-SAM 的四种变体,即串联、并联、混合和 LoRA 结构。我们在九个数据集和六个下游任务上进行了全面实验以验证 SU-SAM 的有效性,包括医学图像分割、伪装目标检测、显著目标分割、表面缺陷分割、复杂目标形状和阴影掩码。我们的实验结果表明,与最先进的方法相比,SU-SAM 取得了具有竞争力或更优的准确率。此外,我们提供了深入分析,强调了 SU-SAM 内不同参数高效设计的有效性。另外,我们提出了一个通用模型和基准,展示了 SU-SAM 同时跨越所有多样化数据集的泛化能力。

英文摘要:

Segment anything model (SAM) has demonstrated excellent generalizability in common vision scenarios, yet falling short of the ability to understand specialized data. Recently, several methods have combined parameter-efficient techniques with task-specific designs to fine-tune SAM on particular tasks. However, these methods heavily rely on handcraft, complicated, and task-specific designs, and pre/post-processing to achieve acceptable performances on downstream tasks. As a result, this severely restricts generalizability to other downstream tasks. To address this issue, we present a simple and unified framework, namely SU-SAM, that can easily and efficiently fine-tune the SAM model with parameter-efficient techniques while maintaining excellent generalizability toward various downstream tasks. SU-SAM does not require any task-specific designs and aims to improve the adaptability of SAM-like models significantly toward underperformed scenes. Concretely, we abstract parameter-efficient modules of different methods into basic design elements in our framework. Besides, we propose four variants of SU-SAM, i.e., series, parallel, mixed, and LoRA structures. Comprehensive experiments on nine datasets and six downstream tasks to verify the effectiveness of SU-SAM, including medical image segmentation, camouflage object detection, salient object segmentation, surface defect segmentation, complex object shapes, and shadow masking. Our experimental results demonstrate that SU-SAM achieves competitive or superior accuracy compared to state-of-the-art methods. Furthermore, we provide in-depth analyses highlighting the effectiveness of different parameter-efficient designs within SU-SAM. In addition, we propose a generalized model and benchmark, showcasing SU-SAM's generalizability across all diverse datasets simultaneously.

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

↑