UnSAMv2: Self-Supervised Learning Enables Segment Anything at Any Granularity

TL;DR

UnSAMv2 introduces a self-supervised approach that enables the Segment Anything Model to control segmentation granularity precisely at any scale without human annotations, significantly improving performance across various tasks.

Contribution

It proposes a novel granularity control embedding and a self-supervised learning method that unlocks multi-scale segmentation capabilities in SAM without requiring dense annotations.

Findings

Achieves improved segmentation metrics across 11 benchmarks.

Enables continuous control over segmentation scale.

Uses only 6K unlabeled images with minimal additional parameters.

Abstract

The Segment Anything Model (SAM) family has become a widely adopted vision foundation model, but its ability to control segmentation granularity remains limited. Users often need to refine results manually - by adding more prompts or selecting from pre-generated masks - to achieve the desired level of detail. This process can be ambiguous, as the same prompt may correspond to several plausible masks, and collecting dense annotations across all granularities is prohibitively expensive, making supervised solutions infeasible. To address this limitation, we introduce UnSAMv2, which enables segment anything at any granularity without human annotations. UnSAMv2 extends the divide-and-conquer strategy of UnSAM by discovering abundant mask-granularity pairs and introducing a novel granularity control embedding that enables precise, continuous control over segmentation scale. Remarkably, with only $6$K unlabeled images and $0.02\%$ additional parameters, UnSAMv2 substantially enhances SAM-2, achieving segment anything at any granularity across interactive, whole-image, and video segmentation tasks. Evaluated on over $11$ benchmarks, UnSAMv2 improves $\text{NoC}_{90}$ (5.69 $\rightarrow$ 4.75), 1-IoU (58.0 $\rightarrow$ 73.1), and $\text{AR}_{1000}$ (49.6 $\rightarrow$ 68.3), showing that small amounts of unlabeled data with a granularity-aware self-supervised learning method can unlock the potential of vision foundation models.