用少量参考图自动分割遥感图像中的未知类别,无需手动标注提示。
ViRefSAM: Visual Reference-Guided Segment Anything Model for Remote Sensing Segmentation
- 通过参考图像提取语义线索,自动生成目标图像的提示信息。
- 在三个少样本基准上实现优于现有方法的分割精度,仅需几幅参考图。
- 适合遥感领域中缺乏标注数据的新类别分割任务。
段落任何模型(SAM)凭借其提示驱动范式,在通用分割任务中展现出强大泛化能力。然而,将其应用于遥感(RS)图像仍面临两大挑战:首先,为每张图像手动构建精确提示(如点或框)成本高且效率低,尤其在密集小目标或空间碎片分布场景下;其次,SAM缺乏领域适应性,因其主要在自然图像上预训练,难以捕捉遥感特有的语义和空间特征,对新类或未见类分割效果差。为此,受少样本学习启发,我们提出ViRefSAM,一种利用少量包含特定类别的标注参考图像引导SAM的新框架。无需人工提示,即可实现跨遥感图像的类别一致对象自动分割。具体而言,ViRefSAM在保持SAM原始架构不变的基础上引入两个关键组件:(1) 视觉上下文提示编码器,从参考图像中提取类别特异性语义线索,并通过与目标图像的上下文交互生成对象感知提示;(2) 动态目标对齐适配器,集成于SAM的图像编码器中,通过注入类别特异性语义缓解领域差距,使SAM能动态聚焦于任务相关区域。在iSAID-5$^i$、LoveDA-2$^i$和COCO-20$^i$三个少样本分割基准上的大量实验表明,ViRefSAM仅需少量参考图像即可准确自动分割未见类别,且在多个数据集上持续优于现有少样本分割方法。
原文摘要 · Abstract (English)
The Segment Anything Model (SAM), with its prompt-driven paradigm, exhibits strong generalization in generic segmentation tasks. However, applying SAM to remote sensing (RS) images still faces two major challenges. First, manually constructing precise prompts for each image (e.g., points or boxes) is labor-intensive and inefficient, especially in RS scenarios with dense small objects or spatially fragmented distributions. Second, SAM lacks domain adaptability, as it is pre-trained primarily on natural images and struggles to capture RS-specific semantics and spatial characteristics, especially when segmenting novel or unseen classes. To address these issues, inspired by few-shot learning, we propose ViRefSAM, a novel framework that guides SAM utilizing only a few annotated reference images that contain class-specific objects. Without requiring manual prompts, ViRefSAM enables automatic segmentation of class-consistent objects across RS images. Specifically, ViRefSAM introduces two key components while keeping SAM's original architecture intact: (1) a Visual Contextual Prompt Encoder that extracts class-specific semantic clues from reference images and generates object-aware prompts via contextual interaction with target images; and (2) a Dynamic Target Alignment Adapter, integrated into SAM's image encoder, which mitigates the domain gap by injecting class-specific semantics into target image features, enabling SAM to dynamically focus on task-relevant regions. Extensive experiments on three few-shot segmentation benchmarks, including iSAID-5$^i$, LoveDA-2$^i$, and COCO-20$^i$, demonstrate that ViRefSAM enables accurate and automatic segmentation of unseen classes by leveraging only a few reference images and consistently outperforms existing few-shot segmentation methods across diverse datasets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。