← 返回 2026-05-26

InstructSAM:通过任意指令分割任意实例的统一框架 InstructSAM: Segment Any Instance with Any Instructions

Yuqian Yuan, Wentong Li, Zhaocheng Li, Yutong Lin, Juncheng Li, Siliang Tang, Jun Xiao, Yueting Zhuang, Wenqiao Zhang 📅 2026-05-25 👍 18 2026-07-13 08:36
DETR式集合预测 SAM3 多模态大模型 实例分割 指令理解

用显式查询接口让VLM和SAM3协同处理复杂指令的多实例分割。

前置知识

SAM系列与提示式分割

SAM把分割建模为"提示→掩码"任务,通过点/框/粗掩码驱动掩码解码器。SAM2扩展到视频,SAM3进一步支持开放世界概念级多实例分割(用短语如"交通锥"检索所有同类实例)。

InstructSAM建立在SAM3的检测器查询空间之上,需要理解SAM3如何从短语提示生成多实例掩码,才能理解本工作为何仍以SAM3为分割器。

DETR式集合预测与二部匹配

DETR将检测视为集合预测问题,输出一组固定数量的对象查询,通过匈牙利算法做预测与真值的二部匹配实现端到端一对一监督。本文的并行实例槽位本质即沿用此范式。

InstructSAM的查询库、匹配监督、存在性损失都直接源自DETR思想,是理解其训练目标和如何避免重复预测的关键。

混合注意力机制

在Transformer中通过自定义注意力掩码让不同token遵守不同注意力规则。本文对文本token保留因果注意力(自回归生成),对实例查询token允许全局双向注意力以互相通信。

这是InstructSAM在保持LLM语言建模能力的同时让实例槽位互相协调的核心机制。

多模态大语言模型(MLLM)

以大语言模型为核心、融合视觉编码器与投影器的统一模型,可同时处理图像与文本指令并以自回归方式生成回答。本文使用Qwen3-VL作为指令理解与查询注入的主干。

InstructSAM将实例查询作为特殊token注入MLLM的一次前向中,让模型同时完成语言理解和实例槽位的上下文化。

研究动机

SAM3虽然把提示式分割扩展到了开放世界的概念级多实例场景(用"交通锥"这样的短语一次找出所有同类目标),但用户的真实意图远不止孤立的类别名词,常常涉及属性("小杯子")、空间约束("靠左的那个")、关系("笔记本旁边")、排除("除前面那个以外")和计数("最大的两个")等组合性指令,需要模型联合做语义解析、视觉推理和实例级定位。当前两条主流路线均有明显缺陷:路线一是VLM(如Qwen-VL/Gemini)将复杂指令改写为概念级短语,反复调用SAM3生成候选掩码再用启发式过滤,但过程缓慢且脆弱,改写会丢弃细粒度约束、迭代过滤会累积误差;路线二是LLM附加特殊[SEG]token,把隐藏态解码成掩码(LISA/Sa2VA),但[SEG]本身缺乏实例绑定机制,多[SEG]token常坍缩为重复掩码(如LISA++-7B在Inst2Seg上mAP仅2.2),且自回归生成速度随实例数线性下降。

本文的目标是本文目标是构建一个统一的端到端框架,让2B规模模型在任意复杂自由形式指令下都能高效、稳健地完成多实例分割,既不依赖多轮agent管线,也无需LLM逐token"说"出掩码。具体来说,作者希望以单一前向的并行预测形式同时获得:(1) 对组合性指令(属性、空间关系、排除、计数、隐式意图)的稳健理解;(2) 对多个目标实例的明确区分(实例判别性);(3) 远快于agent管线的推理效率(实测1.1秒 vs 29.6秒,约27倍加速)。最终目标是让单一2B模型在实例级Inst2Seg、语义级ReasonSeg、指代级gRefCOCO/GSEval/RoboRefIt四大类基准上都达到或超越此前SOTA。

与已有工作不同的是,本文的独特切入角度是引入"显式的推理到实例查询接口":把一组可学习的并行查询当作"实例槽位"注入MLLM,让LLM利用通用推理能力理解复杂指令,再把上下文化后的查询通过轻量MLP投影到SAM3的检测器查询空间,由SAM3在一次前向中解码出多实例掩码。这种设计既保留了SAM3强大的开放世界分割器,又把LLM定位为推理器而非掩码生成器,避免了[SEG]token的非实例判别性和多轮agent的低效,从而填补了"复杂指令理解"与"实例级分割"之间的关键空白。

核心方法

InstructSAM的直觉是:与其让LLM自回归地"说"出掩码token,不如让它把复杂指令"翻译"成一组并行的实例假设,再交给SAM3去落地。具体来说,作者在Qwen3-VL这一多模态LLM中插入一个由 $K=10$ 个可学习查询组成的"实例槽位库",当LLM看到触发token $$ 时把这组查询作为特殊token插入多模态序列,让它们通过自注意力和视觉/文本上下文互相协调;输出端用一个MLP把这组经过LLM条件化的查询投影到SAM3检测器期望的嵌入空间,SAM3据此在单次前向中同时给出多个实例掩码及置信度。训练分两阶段:阶段一对齐LLM查询空间与SAM3(250万样本),阶段二注入推理式指令知识(50万样本),配合DETR式二部匹配与BCE+Dice损失端到端优化。

和已有方法的本质区别在于"显式集合结构接口":LISA系列用[SEG]token做"嵌入即掩码",共享符号没有实例绑定,结果会重复或坍缩(Inst2Seg mAP仅1.9–2.2);SAM3-Agent用VLM改写指令再反复调用SAM3,间接且慢(推理29.6秒)。本文用并行可学习查询替换自回归掩码token,配合混合注意力让查询之间可以全局通信,再用MLP投影到SAM3的查询空间,使LLM只负责"推理与上下文化"、SAM3只负责"定位与解码",任务解耦带来三个新能力——实例判别性(查询一一对应不同实例)、单前向效率(1.1秒)和对组合指令的稳健性。

方法步骤详情

方法分四步:(1) 编码:图像编码器输出 $V$,指令分词为 $T$,引入 $K=10$ 个可学习查询 $Q=\{q_k\}$ 作为并行实例槽位。(2) 注入:LLM遇到 $$ 时,把短语 $T_{phrase}$、触发token与查询库拼成 $X=[V;T;T_{phrase};;q_1;\dots;q_K]$,前向取 $z_k=H[q_k]$。(3) 混合注意力:文本用因果,查询在视觉/文本/查询间全局双向通信。(4) 解码:MLP将 $\{z_k\}$ 投影为SAM3掩码查询 $\{\tilde{z}_k\}$,送入融合编码器、检测器、分割头与分数头,输出 $Y=\{(M_i,s_i)\}_{i=1}^{N}$。训练损失 $\mathcal{L}=\lambda_{text}\mathcal{L}_{text}+\lambda_{seg}\mathcal{L}_{seg}+\lambda_{presence}\mathcal{L}_{presence}$,配DETR式二部匹配与BCE+Dice。

技术新颖性

技术新颖性体现在三点。一是把"实例槽位"作为一等公民注入MLLM单次前向,避免[SEG]token的非实例判别性和多轮agent的语义损失,是首个把"显式集合预测"和"指令推理"统一到同一前向中的方案;二是混合注意力设计同时保留了LLM的因果语言建模能力和实例查询的双向通信能力,是该架构得以成立的关键trick;三是构建了Inst2Seg这一高质量指令级实例分割基准(986张图像、3328条人工核验指令),并提出配套的四阶段数据引擎(QA生成、目标合并、SAM2掩码标注、过滤),其中过滤阶段被消融证明为关键(去掉后Inst2Seg mAP从31.5跌到11.9)。两阶段训练(250万样本对齐+50万推理微调)和Qwen3-VL-2B+SAM3的组合也是工程上的有效决策。

Overview of the InstructSAM framework
Figure 2: Overview of the InstructSAM framework
Examples of the proposed Inst2Seg benchmark
Figure 3: Examples of the proposed Inst2Seg benchmark

实验结果

在Inst2Seg基准上InstructSAM-2B取得mAP 31.5/gIoU 60.4(单目标52.6/66.8、多目标22.2/44.0、无目标gIoU 74.3),大幅领先所有端到端基线(LISA-7B仅1.9/28.6、LISA++-7B 2.2/25.8、SA2VA-8B 9.4/53.9、X-SAM-3.8B 11.0/36.6)。在ReasonSeg上cIoU 65.0/gIoU 62.5,长指令子集提升6.9个cIoU。在gRefCOCO val mAP 57.3/cIoU 68.3,比GSVA-7B在cIoU上提升+6.6。GSEval零样本gIoU 64.1超EVF-SAM 62.6。RoboRefIt分布外testB 74.4远超SA2VA-4B 34.0。推理仅1.1秒,比SAM3-Agent-Qwen3-VL-2B快27倍。消融显示查询库对Inst2Seg mAP贡献最大(去掉从31.5降至20.1)、混合注意力对ReasonSeg cIoU贡献最大(从65.0降至52.4)、阶段一对齐让Inst2Seg mAP提升23.4点。

Comparison of Inst2Seg benchmark with existing referring image segmentation benchmarks
Table 1: Comparison of Inst2Seg benchmark with existing referring image segmentation benchmarks
Results on reasoning-based instruction segmentation benchmarks
Table 2: Results on reasoning-based instruction segmentation benchmarks
Results on phrase-level Referring Expression Segmentation benchmarks
Table 3: Results on phrase-level Referring Expression Segmentation benchmarks
Results on RoboRefIt benchmark
Table 4: Results on RoboRefIt benchmark
Ablation study of key design choices in InstructSAM
Table 5: Ablation study of key design choices in InstructSAM
Ablation study on phrase conditioning and LLM-conditioned queries
Table 7: Ablation study on phrase conditioning and LLM-conditioned queries
Effectiveness of the data engine
Table 8: Effectiveness of the data engine
Inference-time comparison on Inst2Seg benchmark
Table 10: Inference-time comparison on Inst2Seg benchmark
Qualitative comparison with SAM3-Agent-Qwen3-VL-2B and SA2VA-4B
Figure 4: Qualitative comparison with SAM3-Agent-Qwen3-VL-2B and SA2VA-4B
查看结构化数据
任务指标本文基线提升
Inst2Seg 多实例推理分割(实例级mAP) mAP 31.5 (2B参数) LISA-7B 1.9, LISA++-7B 2.2, SA2VA-4B 8.2, X-SAM-3.8B 11.0, SAM3-Agent-Qwen3-VL-2B 29.7 相对最强端到端基线X-SAM提升+20.5 mAP;与同规模agent管线SAM3-Agent-Qwen3-VL-2B相比提升+1.8 mAP且推理快27倍
Inst2Seg 多目标子集 mAP 22.2 (单目标52.6) SA2VA-8B 1.4 (单目标35.5), X-SAM-3.8B 2.0 (单目标33.4) 多目标mAP提升约15-21个点,显示并行查询在多实例枚举上的明显优势
ReasonSeg 验证集 cIoU / gIoU 65.0 / 62.5 X-SAM-3.8B 32.9 / 56.6, SA2VA-4B 60.0 / 59.2, LISA++-7B 68.1 / 64.2 相对SA2VA-4B提升cIoU +5.0;gIoU优于大多数2B级模型
gRefCOCO 验证集 mAP / cIoU 57.3 / 68.3 GSVA-7B — / 61.7, PixelLM-7B 33.9 / 51.2, SA2VA-4B 26.1 / 42.3 mAP超越最强多实例基线PixelLM-7B +23.4;cIoU相对GSVA-7B +6.6
GSEval 全集(零样本) gIoU 64.1 EVF-SAM 62.6, InstructSeg 52.5, GSVA-7B 48.6 相对此前SOTA EVF-SAM提升+1.5
RoboRefIt 分布外testB IoU 74.4 RefTR-r50 61.5, SA2VA-4B 34.0, LISA-7B 28.7 相对RefTR提升+12.9,相对SA2VA-4B提升+40.4,体现强泛化
Inst2Seg 推理效率 Infer Time (s) 1.1秒 SAM3-Agent-Qwen3-VL-2B 29.6秒 快约27倍,单前向输出多实例掩码

局限与改进

作者在讨论章节明确两点局限。一是当前版本仅支持图像输入,未涉及视频场景;视频中的多实例交互、时序对应和帧级掩码一致性使数据标注和自动生成更困难,也更容易产生幻觉指令。二是将分割数据集与现有大规模对话式指令联合训练仍具挑战,朴素联合训练会损害某些分割基准的指标,需要设计更原则性的协同训练策略和数据平衡方案。我自己的额外观察:(1) 模型规模仅2B,对长尾概念和极复杂句法(嵌套从句、多重排除)的上限仍未知;(2) Inst2Seg无目标gIoU仅在零样本设置下评估,缺少该子集的训练数据可能限制真实部署表现;(3) Inst2Seg基准只有3328条指令,规模有限且偏向英文,可能无法全面反映跨语言/跨文化指令的鲁棒性;(4) 推理阶段仍需先生成短语 $T_{phrase}$ 再做分割,存在短语生成错误向下游传播的风险(虽然消融显示去掉短语只损失约2个点,但生产环境中短语错误的复合影响仍待评估)。

独立分析的弱点

独立分析有三个弱点。第一,查询数 $K=10$ 是一个较强的先验,对密集场景(每张图>10个目标)的能力上限未充分验证;表6显示 $K=200$ 时性能几乎不增但延迟翻倍,意味着架构对"目标数远大于 $K$"的极端情况缺乏鲁棒方案(如两阶段提议-精化、或动态停止条件)。第二,MLLM权重用LoRA(rank 256)冻结微调而SAM3的视觉编码器完全冻结,限制了模型对新概念或罕见类别的在线适应能力;可以探索更强的adapter或基于检索的开放词汇扩展。第三,对无目标指令的处理本质上是"所有查询都判为不存在",缺乏显式的拒识机制和可解释性,部署时若需置信度阈值或反例拒识仍需额外工程。第四,数据引擎高度依赖Gemini 3 Flash生成QA和SAM2生成掩码,整个标注管线存在错误传播风险(虽然过滤阶段证明至关重要),对中文/小语种指令同样未验证。改进方向:动态查询数、显式拒识头、多语种数据合成、检索增强的实例编码。

未来方向

作者提出的方向主要是视频扩展和对话/分割协同训练。基于本文成果还可以延伸:(1) 把并行查询接口推广到视频帧间共享槽位,加入时序记忆模块获得"视频InstructSAM",解决跨帧实例对应;(2) 引入规划/链式推理监督,让LLM在$$之前显式生成中间推理步骤("先找桌子→再找桌上物品→排除食物"),使分割过程可解释;(3) 探索查询库与RAG/工具调用的结合,让罕见实体通过外部知识库补充;(4) 把混合注意力思想扩展到3D点云或医疗影像等多模态分割;(5) 把Inst2Seg基准扩展到更多语言、更长指令、含OCR/数学推理的复合任务,并构建公评估协议。

复现评估

复现友好性较高。代码、模型权重和数据集均已开源:GitHub https://github.com/DCDmllm/InstructSAM,模型权重 https://huggingface.co/CircleRadon/InstructSAM-2B/,数据集 https://huggingface.co/datasets/CircleRadon/Inst2Seg。主干使用Qwen3-VL-2B,分割器使用SAM3的预训练权重,依赖都可通过HuggingFace直接获取。训练细节较为透明:阶段一2.5M样本/batch 128/lr 5e-6,阶段二0.5M样本/batch 64/lr 2e-6,均AdamW(β=(0.9, 0.999))、warmup 0.03、weight decay 0、LoRA rank 256(一处文档给出rank 64,附录A.1中存在不一致需要以官方代码为准)、各1 epoch,所有超参数已列表A.1详尽给出。复现难度中等偏上:完整复现两阶段训练需较大算力(建议至少8×A100 80G),但由于权重已开源,仅做推理或下游微调可在单卡完成,验证门槛很低。