arXiv:2410.06964cs.CV2024-10NeurIPS被引 53

用图分析优化少样本语义分割,提升精度与效率。

Bridge the Points: Graph-based Few-shot Segment Anything Semantically

  • 通过图结构动态选择正负提示点,利用背景信息增强区分能力。
  • 基于掩码覆盖关系聚类点,实现粒度对齐,提升分割一致性。
  • 无需额外超参数,推理速度快,适合自动化场景应用。

近年来大规模预训练技术显著提升了视觉基础模型的能力,尤其是分割一切模型(SAM),能根据点或框提示生成精确掩码。现有研究将SAM扩展至少样本语义分割(FSS),聚焦于SAM的自动提示生成。然而,这些方法在提示选择上表现不佳,需针对不同场景调整超参数,且因过度使用SAM导致单次推理时间过长,效率低下、自动化能力有限。为此,本文提出一种基于图分析的简单而高效的方法。首先,正负对齐模块动态选择点提示,尤其挖掘背景作为负参考的潜力;其次,点-掩码聚类模块基于掩码对点的覆盖关系构建有向图,对弱连通分量进行分解,聚合点以形成自然簇;最后,通过正向与过拟合门控机制,结合图对齐的粒度信息,聚合高置信度掩码并剔除误检,减少对额外超参数和冗余掩码生成的依赖。在标准FSS、单样本部件分割及跨域FSS数据集上的广泛实验表明,该方法在保持高效的同时超越现有通用模型,在COCO-20i上达到58.7% mIoU,LVIS-92i上达35.2%。代码已公开于https://andyzaq.github.io/GF-SAM/。

原文摘要 · Abstract (English)

The recent advancements in large-scale pre-training techniques have significantly enhanced the capabilities of vision foundation models, notably the Segment Anything Model (SAM), which can generate precise masks based on point and box prompts. Recent studies extend SAM to Few-shot Semantic Segmentation (FSS), focusing on prompt generation for SAM-based automatic semantic segmentation. However, these methods struggle with selecting suitable prompts, require specific hyperparameter settings for different scenarios, and experience prolonged one-shot inference times due to the overuse of SAM, resulting in low efficiency and limited automation ability. To address these issues, we propose a simple yet effective approach based on graph analysis. In particular, a Positive-Negative Alignment module dynamically selects the point prompts for generating masks, especially uncovering the potential of the background context as the negative reference. Another subsequent Point-Mask Clustering module aligns the granularity of masks and selected points as a directed graph, based on mask coverage over points. These points are then aggregated by decomposing the weakly connected components of the directed graph in an efficient manner, constructing distinct natural clusters. Finally, the positive and overshooting gating, benefiting from graph-based granularity alignment, aggregate high-confident masks and filter out the false-positive masks for final prediction, reducing the usage of additional hyperparameters and redundant mask generation. Extensive experimental analysis across standard FSS, One-shot Part Segmentation, and Cross Domain FSS datasets validate the effectiveness and efficiency of the proposed approach, surpassing state-of-the-art generalist models with a mIoU of 58.7% on COCO-20i and 35.2% on LVIS-92i. The code is available in https://andyzaq.github.io/GF-SAM/.

少样本分割图神经网络SAM高效推理

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