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

学习提示 Segment Anything Models

Learning to Prompt Segment Anything Models

Nanyang Technological University(南洋理工大学);SenseTime Research(商汤科技研究院);Carnegie Mellon University(卡内基梅隆大学);Mohamed bin Zayed University of Artificial Intelligence(穆罕默德·本·扎耶德人工智能大学)

Jiaxing Huang, Kai Jiang, Jingyi Zhang, Han Qiu, Lewei Lu, Shijian Lu, Eric Xing

AI总结:

本文提出空间-语义提示学习方法(SSPrompt),通过直接在嵌入空间优化空间和语义提示并利用预训练知识,解决了Segment Anything Models的提示获取问题,在多个数据集上显著提升了分割性能。

摘要:

像 SEEM 和 SAM 这样的 Segment Anything Models 在学习分割一切方面展现了巨大潜力。SAM 的核心设计在于 Promptable Segmentation,它接受手工制作的提示作为输入,并返回预期的分割掩码。SAM 使用两类提示,包括空间提示(例如点)和语义提示(例如文本),它们共同作用以提示 SAM 在下游数据集上分割一切。尽管提示很重要,但如何为 SAM 获取合适的提示在很大程度上仍未被充分探索。在这项工作中,我们检查了 SAM 的架构,并识别出为 SAM 学习有效提示的两个挑战。为此,我们提出了空间-语义提示学习(SSPrompt),它学习有效的语义和空间提示以构建更好的 SAM。具体而言,SSPrompt 引入了空间提示学习和语义提示学习,直接在嵌入空间上优化空间提示和语义提示,并选择性地利用预训练提示编码器中编码的知识。大量实验表明,SSPrompt 在多个广泛采用的数据集上持续实现了卓越的图像分割性能。

英文摘要:

Segment Anything Models (SAMs) like SEEM and SAM have demonstrated great potential in learning to segment anything. The core design of SAMs lies with Promptable Segmentation, which takes a handcrafted prompt as input and returns the expected segmentation mask. SAMs work with two types of prompts including spatial prompts (e.g., points) and semantic prompts (e.g., texts), which work together to prompt SAMs to segment anything on downstream datasets. Despite the important role of prompts, how to acquire suitable prompts for SAMs is largely under-explored. In this work, we examine the architecture of SAMs and identify two challenges for learning effective prompts for SAMs. To this end, we propose spatial-semantic prompt learning (SSPrompt) that learns effective semantic and spatial prompts for better SAMs. Specifically, SSPrompt introduces spatial prompt learning and semantic prompt learning, which optimize spatial prompts and semantic prompts directly over the embedding space and selectively leverage the knowledge encoded in pre-trained prompt encoders. Extensive experiments show that SSPrompt achieves superior image segmentation performance consistently across multiple widely adopted datasets.

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

↑