让SAM3只处理图像中实际存在的类别,大幅提速且更准。
ActiveSAM: Image-Conditional Class Pruning for Fast and Accurate Open-Vocabulary Segmentation

- 先用低分辨率预览判断图像中哪些类别存在,再只对这些类别做全分辨率分割。
- 在8个基准上平均提升1.4 mIoU,大词汇量数据集最快快5.5倍。
- 无需训练、不依赖标签,适合自动驾驶等噪声环境部署。
Segment Anything Model 3(SAM 3)为概念提示分割提供了强大冻结主干,但直接用于开放词汇语义分割(OVSS)效率低下:通常需对整个数据集词汇表进行全分辨率解码,而每张图像仅涉及少量活跃类别。本文提出ActiveSAM,一种无需训练的零样本推理框架,使SAM 3变为动态词汇分割器。ActiveSAM首先标准化并扩展类别提示,然后通过低分辨率存在性预览估计图像相关的活跃类别集合。仅保留的类别在全分辨率下进行解码,采用桶式提示复用与冻结的SAM 3解码器。预览阶段仅使用类别存在性证据,跳过不必要的分割头计算;最终阶段应用边缘感知背景校准以抑制低置信度像素。ActiveSAM无需目标数据集训练、权重更新或类存在性真值标签。在8个OVSS基准上,其速度-准确率权衡优于现有最优方法SegEarth-OV3,平均提升约1.4 mIoU,大词汇量数据集上最快达5.5倍加速。同时在模拟真实分布偏移的图像退化下表现最强鲁棒性,适用于自动驾驶和具身智能等噪声输入场景。代码已公开于https://github.com/VILA-Lab/ActiveSAM。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。