arXiv:2410.09714cs.CVcs.LG2024-10被引 7

自动提示与掩码校准,让SAM无需人工干预也能精准分割。

AM-SAM: Automated Prompting and Mask Calibration for Segment Anything Model

  • 用双层优化自动生成提示,减少人工标注
  • 冻结主干模型,仅微调解码器,提升特征表达能力
  • 少样本训练下性能超越当前最优方法5%的骰子分数

Segment Anything Model (SAM) 因其强大的泛化能力和优异表现,在语义分割领域广受关注。然而,其仍存在两大局限:(1)严重依赖人工提供的提示(如关键点、边界框或文本),耗时费力;(2)掩码解码器仅使用点积操作,难以准确捕捉特征间相关性,导致表示不准确。现有解决方案如微调 SAM 常需重训大量参数,消耗巨大算力。为此,我们提出基于双层优化框架的 AM-SAM,实现自动提示生成,避免人工参与,并在早期训练阶段即获得良好效果,加速收敛。同时,冻结 SAM 主体,仅通过低秩适应(LoRA)修改掩码解码器,引入更先进的特征关联建模机制,超越简单点积。实验表明,AM-SAM 在分割精度上媲美甚至优于人工与默认提示。尤其在人体分割数据集上,使用4个样本的少样本训练集时,骰子分数较 SOTA 方法高出5%,凸显其在语义分割任务中的优势。

原文摘要 · Abstract (English)

Segment Anything Model (SAM) has gained significant recognition in the field of semantic segmentation due to its versatile capabilities and impressive performance. Despite its success, SAM faces two primary limitations: (1) it relies heavily on meticulous human-provided prompts like key points, bounding boxes or text messages, which is labor-intensive; (2) the mask decoder's feature representation is sometimes inaccurate, as it solely employs dot product operations at the end of mask decoder, which inadequately captures the necessary correlations for precise segmentation. Current solutions to these problems such as fine-tuning SAM often require retraining a large number of parameters, which needs huge amount of time and computing resources. To address these limitations, we propose an automated prompting and mask calibration method called AM-SAM based on a bi-level optimization framework. Our approach automatically generates prompts for an input image, eliminating the need for human involvement with a good performance in early training epochs, achieving faster convergence. Additionally, we freeze the main part of SAM, and modify the mask decoder with Low-Rank Adaptation (LoRA), enhancing the mask decoder's feature representation by incorporating advanced techniques that go beyond simple dot product operations to more accurately capture and utilize feature correlations. Our experimental results demonstrate that AM-SAM achieves significantly accurate segmentation, matching or exceeding the effectiveness of human-generated and default prompts. Notably, on the body segmentation dataset, our method yields a 5% higher dice score with a 4-example few-shot training set compared to the SOTA method, underscoring its superiority in semantic segmentation tasks.

语义分割自动提示少样本学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。