ActiveSAM: 图像条件类剪枝加速开放词汇分割

ActiveSAM: Image-Conditional Class Pruning for Fast and Accurate Open-Vocabulary Segmentation

精选理由

研究者用ActiveSAM让SAM 3分割更快更准,比SegEarth-OV3提升1.4的mIoU,速度快5.5倍,还不用重新训练。

AI 摘要

ActiveSAM是一个无需训练、零样本推理的框架,将SAM 3改造为主动词汇分割器。它先通过低分辨率预视估计图像条件下的活跃类别集,再对保留类别进行全分辨率解码。在8个开放词汇语义分割基准上,ActiveSAM平均mIoU比当前最优的SegEarth-OV3高1.4,且在大词汇数据集上速度提升高达5.5倍。该方法无需目标数据集训练或权重更新,在图像损坏鲁棒性测试中也表现最强。

原文 · arXiv cs.LG

ActiveSAM: Image-Conditional Class Pruning for Fast and Accurate Open-Vocabulary Segmentation

Segment Anything Model 3 (SAM 3) provides a strong frozen backbone for concept-prompted segmentation, but applying it directly to open-vocabulary semantic segmentation (OVSS) is inefficient: full-resolution decoding is typically run over the entire dataset vocabulary, whereas each image contains only a small active subset of classes. We introduce ActiveSAM, a training-free, zero-shot inference framework that turns SAM 3 into an active-vocabulary segmenter. ActiveSAM first canonicalizes and expands class prompts, then estimates an image-conditioned active set from a low-resolution presence preview. Only the retained classes are decoded at full resolution, using bucketed prompt multiplexing with the frozen SAM 3 decoder. The preview stage uses only class-presence evidence and skips unnecessary segmentation-head computation, while the final stage applies margin-aware background calibration to suppress low-confidence pixels. ActiveSAM requires no target-dataset training, no weight updates, and no oracle class-presence labels. Across eight OVSS benchmarks, ActiveSAM improves the speed-accuracy tradeoff of training-free open-vocabulary semantic segmentation, outperforming the current state-of-the-art SegEarth-OV3 by approximately +1.4 mIoU on average while running up to 5.5x faster on large-vocabulary datasets. ActiveSAM also demonstrates the strongest robustness under image corruption that simulates real-world distribution shift, making it well-suited for deployment in noisy-input domains such as autonomous driving and embodied AI. Code is available at https://github.com/VILA-Lab/ActiveSAM.