Where-OPD:基于合成场景空间引导的多模态大模型策略内自蒸馏 Where-OPD: Spatially Guided On-Policy Self-Distillation of MLLMs with Synthetic Scenes
教师看空间提示、学生只看图,3000张合成计数图自蒸馏出可迁移的真实视觉感知
前置知识
策略内蒸馏(On-Policy Distillation, OPD)
传统蒸馏让学生模仿教师自己生成的轨迹(off-policy),训练分布与推理分布脱节。OPD 则先从学生策略采样轨迹 $y \sim \pi_\theta(\cdot \mid x)$,再由更强教师在学生实际访问的每个状态上给出 token 级分布监督,最小化学生分布 $p_t$ 与教师分布 $q_t$ 之间的散度。因为监督恰好落在推理时会到达的状态上,信号比稀疏的序列级奖励密集得多。
本文整个训练框架就是 OPD:教师在学生自己采样的回答前缀上逐 token 打分,理解这一点是读懂方法部分的前提。
策略内自蒸馏(OPSD)与特权信息
OPSD 不引入外部更强教师,而是把学生自身的一个冻结或 EMA 副本当作教师,只给教师注入学生看不到的特权信息 $x^+$,人为制造信息差。学生从「更有信息的自己」身上学习,推理时却完全自足。核心设计问题是:特权信息选什么形式,决定了教师比学生强在哪里、能蒸馏出什么能力。
本文的全部贡献就是重新设计这个特权信息:从以往「给教师更好的视觉观察」换成「给教师文本空间提示」,这是理解论文动机的关键。
广义 Jensen–Shannon 散度
一种对称的分布差异度量:$D_{JS}(p\|q) = \tfrac{1}{2}D_{KL}(p\|m) + \tfrac{1}{2}D_{KL}(q\|m)$,其中 $m = \tfrac{1}{2}(p+q)$。相比 KL 散度更对称、对教师分布中的低概率 token 更鲁棒。本文使用 $\beta=0.5$ 的广义 JSD,并只在学生 top-100 token 及尾部概率项上计算以节省显存与算力。
它是本文的训练目标(公式 6、7),不熟悉 JSD 就无法理解损失函数如何构造。
GRPO 与稀疏奖励的局限
GRPO 是可验证奖励的强化学习方法:对同一问题采样多个回答,把提取出的答案与真值对比得 0/1 奖励,按组内均值居中计算优势后更新策略。它只告诉模型「最终答案对不对」,属于序列级稀疏信号,无法指出回答中哪个 token 没有正确利用视觉输入。
论文把 GRPO 作为对照基线,用实验证明 token 级自蒸馏信号优于这种稀疏序列级奖励,这是方法选型的核心论据之一。
EMA 教师
指数移动平均教师以 $\bar{\theta} \leftarrow \eta\bar{\theta} + (1-\eta)\theta$ 的方式缓慢跟随学生更新,使教师随训练逐步变强又不漂移过快,是自蒸馏和对比学习中的常用技巧,$\eta$ 越大教师越接近冻结。
论文一个反直觉发现是:在合成数据上冻结教师($\eta=0$)效果最好,任何 EMA 更新都会降低性能,理解 EMA 的含义才能体会这个消融的价值。
程序化场景生成(CLEVR 风格)
先用代码完整指定一个场景——每个物体的类别、属性、坐标都由生成器决定——再渲染成图像,因此场景的全部元数据天然免费可得,无需任何标注。CLEVR 最早用这种思路做组合视觉推理问答,本文把它用作后训练数据的来源。
本文的空间提示之所以能零成本、无标注地大规模生成,正是因为场景由程序生成、坐标已知,这是整个方法的可行性基础。
研究动机
多模态大模型(MLLM)的视觉感知仍然薄弱,尤其是计数、图表理解、文档阅读这类需要在图像中定位多处证据并整合的任务。近期的视觉 OPSD 方法(Vision-OPD、Imagine-OPD、RP-OPSD、S2VOPD 等)无一例外地通过「给教师更好的视觉观察」来制造师生信息差:教师看问题相关区域的裁剪图、放大视图或更高分辨率图像,学生看原图。这类方法的收益高度集中在依赖视觉放大的细粒度任务上:在 Qwen3.5-4B 上 Vision-OPD 在 V* 上提升 8.90 点、ZoomBench 上提升 14.56 点,但在 CountQA 上反而暴跌 10.73 点,BLINK 下降 4.26 点,幻觉相关基准 HallusionBench 和 AMBER 也分别下降 3.57 和 2.48 点。也就是说「看得更清楚」的能力无法均匀迁移到其他视觉任务。此外这些方法还需要人工标注的 grounding 数据或外部模型来定位相关区域,难以规模化。
本文的目标是作者提出换一种思路:特权信息不告诉教师「应该看一张更清楚的图」,而是告诉它「问题相关的证据在图像的哪些位置」。目标是设计一种空间引导的策略内自蒸馏:教师额外获得一段文本提示,指出与问题相关的视觉元素及其坐标;学生只看图像和问题,通过模仿教师的推理分布内化「扫描图像—定位证据—汇总作答」的行为。同时要求整个流程无需人工标注、无需外部教师模型或现成数据集,并验证这种能力能否从训练所用的简单合成场景迁移到真实世界图像以及计数之外的任务。
与已有工作不同的是,本文的独特之处有三层。第一,特权信息的形式从视觉模态换成文本模态:师生看同一张图,差别只在一段列出目标物体位置与数量的文字,这与所有裁剪/放大/分辨率类方法本质不同。第二,数据来源从真实数据集换成程序化生成场景:物体身份、属性、坐标由生成器天然已知,空间提示可以从场景元数据自动模板化生成,彻底绕开了 grounding 标注或外部定位模型的成本。第三,训练任务刻意做得极简(几何图形计数),但押注的假设是:蒸馏出的「寻找并利用分散视觉证据」的通用行为可以跨越视觉分布和任务类型的鸿沟,实现合成到真实的迁移。这借鉴了 CLEVR/VQA 式程序化数据的思想,但把它用在后训练自蒸馏而非预训练或指令微调上。
核心方法
直觉上,计数、读文档、看图表的共同瓶颈是「找到问题相关的视觉元素并把它们整合起来」。如果训练时有一个知道答案线索在哪里的教师,在学生自己生成的回答轨迹上做逐 token 监督,学生就会逐渐学会自己完成定位与整合。技术路线分三步。第一步,程序化生成场景:在 $1024\times1024$ 画布的纯色背景上放置 12–40 个几何图形(7 种形状 × 12 种颜色,半径 14–64 像素,位置互不重叠),场景状态 $z = \{(c_i, r_i, u_i, v_i)\}_{i=1}^{M}, b$ 完整记录每个物体的颜色-形状类别 $c_i$、半径 $r_i$、中心坐标 $(u_i, v_i)$。第二步,自动生成监督数据:对每个场景按目标类别出现次数加权采样一个计数问题 $Q$,相关元素集合 $O_c(z) = \{(u_i,v_i) \mid c_i = c\}$ 的数量 $N_c = |O_c(z)|$ 即答案 $a$,再用固定模板 $h = T(c, O_c(z))$ 把目标类别、每个实例坐标和总数拼成提示。第三步,策略内自蒸馏:学生输入 $x=(I,Q)$,冻结教师输入 $x^+=(I,Q,h)$,两者从同一预训练 checkpoint 初始化,用广义 JS 散度在学生采样的轨迹上做 token 级蒸馏,只更新学生参数;推理时模型只接收 $(I,Q)$,提示完全退场。
与已有 MLLM OPSD 方法的本质区别在于师生信息差的构造位置。Vision-OPD 给教师证据中心裁剪图、Imagine-OPD 给放大视图、RP-OPSD 用分辨率差、S2VOPD 反向给学生强增广的降质图像——它们改变的都是视觉观察本身,因此蒸馏出的优势只对受益于「看得更清」的任务有效。Where-OPD 保持图像完全不变,改用文本空间提示指定「哪里有什么」。一段提示同时携带三重信息:目标物体的类别、每个实例的像素坐标、以及最终计数。其关键假设是:教师因为知道「去哪看」而能在相同图像上产生更好的推理分布,蒸馏迫使学生学会自己执行搜索与整合过程,而不是依赖更清晰的输入,因此提升不局限于细粒度放大类任务。另一个关键设计是提示来自场景元数据,天然精确、零标注成本,绕开了 Vision-OPD 等方法对人工 grounding 标注或外部教师模型的依赖。
方法步骤详情
完整流程如下。步骤一,生成场景:每个场景先采样 2–5 种颜色、2–4 种形状组成的小词汇表,然后在 1024×1024 画布上逐一放置 12–40 个物体,颜色形状取自词汇表、半径均匀取自 14–64 像素、中心均匀采样且拒绝与已有物体重叠,背景从 10 种基色加通道抖动中抽取;渲染得到图像 $I$,同时保留场景状态 $z$。步骤二,构造问题与提示:按类别计数加权采样问题 $Q$(偏好实例多的类别,目标计数范围 1–14、中位数 3),4B 模型配多选题格式(干扰项取真实计数的近似值)、9B 模型用开放问答(因为多选题会让 9B 直接出答案、缺少可供蒸馏的中间 token);由 $z$ 直接算出 $O_c(z)$ 和 $N_c$,模板生成形如 “Scanning for green crosses: found one near (377, 400), found one near (213, 805). Counting: 2 total.” 的提示 $h$。共生成 3000 对训练样本。步骤三,on-policy 采样:每个 prompt 用 vLLM 以随机采样生成 8 条学生轨迹 $y \sim \pi_\theta(\cdot \mid x)$,训练时关闭 thinking。步骤四,教师打分:冻结教师 $\bar{\pi}_\theta$ 在相同前缀 $y_{<t}$ 上以 $x^+=(I,Q,h)$ 计算分布 $q_t = \bar{\pi}_\theta(\cdot \mid x^+, y_{<t})$。步骤五,更新:最小化 $\mathcal{L}_{\text{distill}} = \mathbb{E}_{y \sim \pi_\theta} \left[ \tfrac{1}{|y|} \sum_{t=1}^{|y|} D_{JS}(\tilde{p}_t \| \tilde{q}_t) \right]$,其中 $D_{JS}$ 取 $\beta=0.5$、只在 top $K=100$ token 及尾部概率上计算,教师侧 stop-gradient,梯度只流向学生;学习率 $2\times10^{-6}$、10 步 warmup、全局 batch 96(9B 为 48)、只训 1 个 epoch。步骤六,推理:仅输入 $(I,Q)$,无需任何提示。
技术新颖性
技术新颖性体现在四点。其一,特权信息的「文本化 + 空间化」是全新设计:把 grounding 从一笔需要人工标注或外部模型的成本,转化为合成数据生成的免费副产品,同时保留了「指明证据位置」这一对感知任务最有效的信息。其二,结果证明 3000 张极简几何计数场景足以诱导跨任务、跨分布的感知提升,这种规模的合成到真实迁移在 MLLM 后训练中相当罕见——通常合成数据增益会被分布差距锁死在域内。其三,系统的信号消融揭示了清晰的设计规律:空间坐标本身比答案更有价值(只给坐标不给总数 72.35,只给答案 71.27,二者结合 72.94),而把目标物体裁剪图给教师反而有害(68.76,CountQA 崩到 20.22),直接回应了「特权信息该给什么」这一开放问题。其四,发现冻结教师优于 EMA 教师(72.94 vs 最高 72.57),与自蒸馏社区惯常做法相反,作者将其归因于在合成域上更新教师会引入域漂移。注意力富集分析还提供了机制证据:Where-OPD 在第 15 层对目标物体的注意力富集达 5.79(面积归一化基线为 1.0),显著高于基线模型峰值的 1.15。
实验结果
在 15 个基准、3 个 MLLM 上评测(准确率 %,贪婪解码、thinking 关闭)。Qwen3.5-4B:15 基准平均从 73.86 提至 78.31(+4.45),大幅领先全部基线(Vision-OPD +0.53、S2VOPD +1.40、OPD-V +1.48);ChartQA +7.20(79.32→86.52)、EvoChart +10.11(68.32→78.43)、HallusionBench +10.76(69.40→80.16)、MME-RealWorld +10.64(64.11→74.75)、GQA +6.09、AMBER +4.48、CountQA +2.53(32.00→34.53)、OCRBench +1.93、V* +5.24、CVBench +2.31,15 项中 13 项提升。Qwen3.5-9B:平均 +1.76(75.81→77.57),ChartQA +5.93、EvoChart +6.43、HallB +3.57。Qwen3-VL-4B:+1.01(71.81→72.82),CountQA +2.69。最关键的发现是合成到真实的迁移:尽管只用了 3000 张几何图形计数场景,真实图像基准(CVBench、V*、HR-Bench 4K/8K、ZoomBench、MME-RealWorld、BLINK)的平均提升达 3.23/1.07/1.29 点(三个模型)。逐类别分析(Tab. 4)显示 MME-RW 自动驾驶感知子类暴涨 24.70 点(33.06→57.76)、HallusionBench 数学子类 +27.78、BLINK 计数子类 +9.16。信号消融(Tab. 2)证明增益来自特权信号而非合成数据本身:同数据下 SFT 仅 69.70、GRPO 69.56、无特权信息 OPSD 69.63,而完整空间提示达 72.94。开启 thinking 的最优推理设置下基线 77.06→80.34(+3.28),而 Vision-OPD 仅 +0.45 且在 12 个基准上退化。三次随机种子的平均标准差仅 ±0.20,结果稳定。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| 15 基准平均(Qwen3.5-4B) | 准确率 (%) | 78.31(+4.45) | Qwen3.5-4B 73.86;最强基线 OPD-V 75.34 | 较基线 +4.45,较最强基线 +2.97 |
| EvoChart(图表理解) | 准确率 (%) | 78.43 | 68.32 | +10.11 |
| HallusionBench(幻觉) | 准确率 (%) | 80.16 | 69.40 | +10.76 |
| MME-RealWorld(高分辨率真实场景) | 准确率 (%) | 74.75 | 64.11 | +10.64 |
| ChartQA(图表理解) | 准确率 (%) | 86.52 | 79.32 | +7.20 |
| CountQA(真实世界计数) | 准确率 (%) | 34.53 | 32.00(Vision-OPD 反而降至 21.27) | +2.53 |
| GQA(真实场景推理) | 准确率 (%) | 75.41 | 69.32 | +6.09 |
| 真实图像基准迁移(CVBench/V*/HR-4K/HR-8K/Zoom/MME-RW/BLINK) | 平均准确率提升(点) | +3.23(Qwen3.5-4B) | 各模型基线 | 三模型分别 +3.23 / +1.07 / +1.29 |
| 训练信号消融(7 基准平均) | 准确率 (%) | OPSD+空间提示 72.94 | SFT 69.70 / GRPO 69.56 / 无特权 OPSD 69.63 / 仅答案提示 71.27 | 较 SFT +3.24,较无特权 OPSD +3.31 |
| thinking 最优设置(15 基准平均) | 准确率 (%) | 80.34(+3.28) | Qwen3.5-4B 77.06;Vision-OPD 77.50(仅 +0.45) | 较基线 +3.28,约为 Vision-OPD 增益的 7 倍 |
局限与改进
可以观察到以下局限。其一,细粒度视觉搜索类增益有限:ZoomBench 仅 +0.87(Qwen3.5-4B),远低于 Vision-OPD 的 +14.56,HR-Bench 4K 甚至下降 0.63 点,说明文本位置提示不能替代真正的放大与高分辨率输入,两类特权信息各有盲区。其二,计数本身提升幅度小:CountQA 绝对值仅从 32.00 到 34.53,真实世界计数远未解决;合成场景上的目标计数中位数只有 3,与真实图像的复杂计数分布差距大。其三,存在局部退化:BLINK 多视图推理 -6.01、功能对应 -1.54,Qwen3.5-9B 的 ZoomBench -0.32。其四,训练任务形态单一——只有单类别计数问题,未探索空间关系、属性、多类别组合等更丰富的提示形式;冻结教师最优的结论也是在合成域上得出的,真实数据上未必成立。其五,数据量超过 3000 后收益饱和甚至回退(11000 场景时 Qwen3-VL-4B 的 CountQA 从 27.68 掉到 23.63),规模化路径不明。其六,评测依赖规则验证器加 LLM judge 的三段打分,虽然各方法协议统一,judge 模型未在正文说明,严格复现指标可能有噪声。作者在结论中基本只强调正面结果,未主动讨论上述盲区。
独立分析的弱点
第一,特权信号表达力受限:提示只有「类别 + 像素坐标 + 计数」的固定模板,无法表达空间关系、遮挡、属性比较等结构。改进方向:程序化生成关系型提示(如「红色方块在蓝色圆的左上方」),或引入多类别组合查询。第二,合成到真实的迁移机制完全黑箱:为什么计数行为能迁移到图表理解和自动驾驶感知?论文只有注意力富集的间接证据。改进方向:设计受控实验(改变提示信息量、坐标表示方式)并分析中间表征变化,把「定位—整合」假设做成可证伪的。第三,坐标系错位问题:提示给的是像素坐标,而模型内部使用视觉 token 索引,学生必须自己学会映射;改进方向是直接以视觉 token 网格坐标为提示,或提供 bbox 而非中心点。第四,缺乏通用能力与感知的联合评估:没有报告 MM MU 类通用推理、数学、代码基准,感知增强是否损害语言能力未知;改进方向是加入综合评测并探索数据混合。第五,单 epoch、3000 样本即饱和,说明简单场景的信息量上限很低;改进方向是课程式提升场景复杂度(更密、遮挡、小目标),延长有效训练窗口。第六,教师能力上限等于学生初始化加一段提示,若模板提示本身表达不了某些推理(如需要视角变换),学生永远学不到;可尝试空间提示与更强视觉前端(更高分辨率教师编码器)组合。
未来方向
作者方向的延伸:其一,把程序化场景从计数扩展到相对位置、深度估计、遮挡推理等空间任务,生成更丰富的特权提示,检验迁移边界;其二,将空间提示与视觉特权(放大、高分辨率)组合成混合教师,兼顾广泛迁移与细粒度搜索——本文数据显示两者互补(Where-OPD 赢在广度,Vision-OPD 赢在 ZoomBench);其三,与 RLVR/GRPO 混合训练:OPSD 提供密集 token 级感知信号,可验证奖励保证最终答案正确性,可能优于两者单独使用;其四,在真实图像上用开源 grounding 或检测模型自动产生提示,检验合成场景是否必需;其五,研究 EMA 在合成域失效的机理(域漂移假说需要直接验证,例如在混合真实-合成数据上重复 Fig. 3 实验)。基于本文成果可再延伸:把该方法搬到视频领域(时序定位提示「第 3 秒到第 5 秒之间目标出现两次」);与 V-GIFT 等视觉指令微调方法正交组合进完整预训练-后训练管线;将空间提示用于 3D 场景理解与具身导航;以及从理论层面刻画「位置监督」比「答案监督」更有利于蒸馏的信息论解释。
复现评估
复现难度中低,是这篇论文的显著优点。有利条件:数据完全程序化生成,附录 A.1 给出了生成器的全部细节(1024×1024 画布、10 种背景基色加抖动、7 形状 × 12 颜色、12–40 个物体、半径 14–64 像素、拒绝重叠放置、按计数加权选题、近似值干扰项),无任何外部数据依赖;训练超参完整公开(lr $2\times10^{-6}$、10 步 warmup、全局 batch 96/48、每 prompt 8 个 rollout、1 epoch、$\beta=0.5$、top $K=100$);算力门槛极低——Qwen3.5-4B 只需 4 张 H100 约 1.4 小时,Qwen3-VL-4B 1.2 小时,9B 也只要 5.8 小时,普通实验室甚至单机多卡即可负担。不利条件:依赖 Qwen3.5 与 Qwen3-VL 权重可用性;论文只给出 Project page 占位,未明确承诺代码与场景生成器开源,生成器的精确视觉渲染细节(形状绘制方式等)需要自己实现;基线对比部分依赖作者私下分享的 checkpoint(S2VOPD、Imagine-OPD),第三方想复现完整对比表需要额外跑基线;评测的 LLM judge 具体型号正文未指明,严格对齐数字可能有小偏差。总体而言,核心方法(合成数据 + JSD 蒸馏)结构简单、成本低,复现主结果可行性很高。
论文图表
左侧为方法示意:在合成计数场景上,教师收到「扫描绿色十字:在 (377,400) 附近找到一个、在 (692,619) 附近找到一个、在 (213,805) 附近找到一个,共 3 个」的空间提示并蒸馏给学生;右侧柱状图展示 Where-OPD 相对 Qwen3.5-4B 基线在多个基准上的准确率提升:MME-RW +10.6、EvoChart +10.1、ChartQA +7.2、V* +5.2、HR-Bench 8K +2.7、CVBench +2.3。
一张图概括论文核心卖点:仅用合成计数数据做后训练,却在多种真实图像任务上获得广泛提升,这是全文最想传达的结果。
同 3000 合成场景、同训练步数下不同训练信号与提示的对比(7 基准平均):基线 69.63、SFT 69.70、GRPO 69.56、无特权信息 OPSD 69.63、裁剪图提示 68.76(CountQA 崩至 20.22)、仅答案提示 71.27、仅坐标无总数 72.35、完整空间提示 72.94。
全文最重要的消融:证明增益来自空间特权信号本身而非合成数据,且空间坐标比答案更有价值、视觉裁剪有害,直接回答「特权信息该给什么」。
两个模型上不同 EMA 率的完整逐基准数据:Qwen3.5-4B 冻结 72.94、10⁻⁴ 72.57、10⁻³ 72.14、10⁻² 71.80、10⁻¹ 71.93;Qwen3-VL-4B 冻结 66.86 亦为最优。
支撑「冻结教师优于 EMA」结论的完整数据,排除单点噪声可能,是该反直觉发现的证据基础。
全部模型在 thinking 开启下的 15 基准评测:基线 76.42,Where-OPD 77.37(+0.95,CountQA +7.26),Vision-OPD -0.36,OPD-V -1.88,S2VOPD +0.16。
与 Table 3 互补,给出 thinking 协议下的完整数字,说明本文增益并非依赖关闭 thinking 的评测设置。