InstructSAM:通过任意指令分割任意实例的统一框架 InstructSAM: Segment Any Instance with Any Instructions
用显式查询接口让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的组合也是工程上的有效决策。
实验结果
在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点。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| 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),但由于权重已开源,仅做推理或下游微调可在单卡完成,验证门槛很低。
论文图表
图1用并排对比展示三条指令驱动分割路线:(a) SAM3接收短名词短语,经Text Encoder和Mask Decoder并行生成多个候选掩码,但仅支持简单概念级提示;(b) 常见MLLM方案让LLM通过[SEG]等特殊token直接解码掩码,缺点是会产生语义级掩码或不一致的多实例结果;(c) InstructSAM在LLM内部维护一组并行可学习的实例查询,输出多个带分数的掩码,支持复杂推理、多实例预测且无重复预测。
这是论文的核心动机图,一图概括了三条路线各自的瓶颈和InstructSAM的差异化设计,是理解'motivation'和'method'关键思想的最佳起点。
表6扫描查询数 $K\in\{10,50,200\}$,推理时间从1.1秒增至2.1秒,但Inst2Seg mAP仅在31.1–31.5间波动,Inst2Seg单目标mAP稳定在52.0–52.6,ReasonSeg gIoU稳定在62.3–62.9。
指导实际部署时的查询数选择,证明小查询库已足够,是工程上的关键参考。
表9对两阶段训练的阶段一做消融:去掉对齐阶段后gRefCOCO val mAP从57.3跌至41.3(-16.0)、Inst2Seg mAP从31.5跌至8.1(-23.4)、ReasonSeg val cIoU从65.0跌至15.9(-49.1),证明对齐阶段对指令级和推理级分割都至关重要。
说明两阶段训练的合理性,是理解训练策略不可替代的证据。
附录图A.1用饼图分别展示阶段一(250万样本)和阶段二(50万样本)的数据组成,阶段一以类别级和短语级对齐数据为主,阶段二加入指令级推理分割数据和部分短语级数据。
辅助理解两阶段训练的数据配比与课程设计。
附录表A.1列出两阶段全部超参:batch size(128/64)、epoch(1/1)、各模块学习率、AdamW参数(β=(0.9,0.999))、weight decay 0、warmup 0.03、LoRA rank(64,与正文256有冲突需以代码为准)。
复现训练的核心参考表,是评估可复现性的关键信息。