精读笔记
Problem Setting
ProLaViT 处理的不是一般 VQA,而是 MLLM 在视觉中心推理中如何进行多步、可递进的视觉状态更新。关键矛盾是:模型需要保留低层视觉几何和局部外观,但现有 reasoning 接口主要是语言 token;如果转向图像级中间结果,推理成本又不可接受;如果只做 latent prediction,则一步到位的表示压力太大,容易把目标区域、背景和任务相关线索混在一起。
以前方法卡在两个地方。Textual CoT 可以提供逻辑结构,但视觉 grounding 被离散语言瓶颈破坏,坐标、边界、局部纹理等信息会被粗粒度语义替代。显式 visual CoT 或 tool-use 能保留视觉状态,但推理时要反复调用生成器、分割器、深度模型或视觉编码器,系统复杂且慢。单步 latent 方法看似折中,但缺少中间状态约束,容易退化为一次性 feature regression。
Motivation
作者的核心观察是:复杂视觉推理并不一定需要生成像素,但需要一个“逐步变化的视觉状态”。如果模型只输出文本,它没有足够通道表达细粒度视觉变换;如果模型只预测一个 latent,它没有结构化路径来分摊定位、聚焦、验证等子问题。
因此真正缺的是一种训练时可监督、推理时低成本、并且和原生视觉表示对齐的中间视觉状态机制。ProLaViT 的动机不是发明新的视觉操作,而是把可程序化视觉操作转化为 latent space 中的监督轨迹,让模型在训练时看见“应该如何从全局视图走到任务相关视图”,推理时则只在隐藏空间中复用这种轨迹。
Core Idea
论文真正的核心思想是:把视觉推理建模为 structured latent trajectory,而不是 text trace 或 single latent target。每个 latent thought 对应一个程序生成的辅助视觉视图在 native vision encoder 下的特征。模型学习的不是“解释为什么答案对”,而是“在内部表示中走过一条视觉变换路径”。这改变了信息流:视觉状态不再被压缩成语言,而是在连续特征空间中被逐步重写。
这个设计引入了很强的 inductive bias:视觉推理应当是 coarse-to-fine 或 hypothesize-critique-verify 的有序过程。相比 prior 的本质区别不是“多了 latent token”,而是这些 token 被绑定到一条有因果顺序的视觉轨迹上,并通过同源视觉编码器监督。它比显式 tool-use 更 scalable,因为推理时不重复调用工具;比文本 CoT 更保真,因为中间监督仍在视觉 embedding space;比单步 latent 更稳定,因为任务被分解到多个状态。
Method
方法层面最关键的是三件事。
第一,Progressive Latent Visual Thought 解决单步 latent 容量过载的问题。它把视觉任务拆成固定数量的 latent states,使模型不必一次性从原图跳到最终判断,而是经过定位、聚焦、隔离或假设、批判、验证等状态变化。核心变化是把 reasoning burden 从一个隐变量分摊到一条有序轨迹。
第二,Endogenous Self-Distillation 解决中间状态没有标注的问题。作者用程序合成辅助图像,再用模型自己的冻结视觉编码器提取 teacher features。这个机制的必要性在于避免外部专家特征空间和 MLLM 内部视觉空间不兼容。它不是单纯蒸馏,而是把可执行视觉操作映射回模型原生视觉 manifold。
第三,Distance-Weighted Diversity Loss 解决 latent chain 坍缩的问题。如果不约束,多步 token 可能学成相似表示,progressive chain 只剩形式。均匀 diversity 又会破坏相邻步骤共享上下文。距离加权约束的作用是:近邻步骤可相似,远距离步骤必须明显不同,从而给 latent trajectory 一个粗略拓扑结构。
Key Insight / Why It Works
这篇最可能有效的部分是 representation alignment 加 structured latent bias 的组合,而不是某个具体 loss。程序合成轨迹提供了高精度、低噪声的中间监督;native encoder 保证 teacher space 和 student 所处视觉语言空间一致;progressive token chain 降低了单步 feature reconstruction 难度。三者合在一起,模型学到的是一种可复用的视觉状态转移模板。
Distance-weighted diversity loss 是合理但更像辅助机制。它解决的是 latent collapse 这个实际训练病灶,但核心增益未必来自该 loss 本身,而是来自“中间视觉轨迹可监督”这一设定。表中 margin loss 有时还会伤害性能,说明简单拉开 latent states 并不等价于更好推理;真正重要的是保留相邻状态连续性,同时让远端状态承担不同视觉功能。
这套方法本质上不是 test-time planning,也不是模型自主发现视觉操作。它更接近 offline program supervision + latent internalization。所谓 reasoning 很大程度上可能是把训练时见过的操作轨迹模式压进模型参数,推理时以 latent tokens 复现。这个判断并不贬低方法:很多有效的 MLLM reasoning 本来就是把外部过程内化。但它意味着泛化上限强依赖合成轨迹的覆盖范围。
最值得注意的是 endogenous distillation 的判断。外部专家看似更强,但如果其 feature space 与 MLLM downstream decision space 不一致,蒸馏反而引入 alignment gap。用 native encoder 做 teacher 在能力上未必最强,但在闭环一致性上更优。这是一个可迁移 insight:对于 latent supervision,teacher 的 task alignment 和 manifold consistency 往往比 teacher 单点感知能力更重要。
需要警惕的是,实验中的增益可能同时来自数据、curriculum、格式约束和 latent 表示。文中虽然做了文本轨迹 SFT、teacher ablation、ordering ablation,但还不足以完全排除“程序合成数据覆盖了 benchmark 模式”或“固定轨迹模板适配了评测分布”的可能。增益来源不清,尤其是 ChartQA 和 Jigsaw 这类任务,可能较强受合成数据风格影响。
Relation To Prior Work
ProLaViT 位于 visual CoT、latent reasoning、tool distillation 三条线的交叉处。和 textual CoT 的差异在于中间推理状态不再离散化为自然语言;和显式 visual CoT 的差异在于训练时用辅助视图监督、推理时不生成图像;和 LVR / Mirage 这类 latent 方法的差异在于它不把 latent reasoning 当黑盒连续 token,而是绑定到有顺序、有操作语义的视觉轨迹。
看似新的部分中,programmatic synthesis、teacher feature distillation、special latent tokens 都不是全新思想;它们分别来自 synthetic reasoning data、feature distillation 和 continuous CoT。实质创新在于把这些组件组织成一个闭环:程序生成中间视觉状态,native encoder 把它们投到模型自身视觉空间,LLM latent tokens 学习复现该轨迹,diversity loss 维持轨迹拓扑。
因此它不是一次范式革命,而是一次相当务实的系统重组。真正新增的信息是:对于视觉推理,latent chain 需要被中间视觉状态锚定;仅仅增加连续 token 不够,必须给这些 token 一个结构化的监督路径。
Dataset / Evaluation
评测覆盖了视觉搜索、拼图、图表、空间感知等 perception-intensive 场景,基本对应作者声称的“复杂视觉推理”范围。结果支持 progressive latent chain 在这些 benchmark 上优于文本 CoT、单步 latent 和部分外部 tool distillation。尤其是 teacher selection、step ordering 和 explicit tool-use 的对比,确实在验证核心机制,而不是只报 SOTA。
但 evaluation 仍有明显边界。训练数据来自 Chart Refocus、Jigsaw Assembly、Visual Search 三类程序化合成轨迹,而评测也集中在类似视觉操作可定义的任务上。它验证的是“在可程序化视觉操作族内,latent internalization 是否有效”,而不是开放世界视觉推理能力。真实 deployment 中,模型需要自己决定操作类型、步骤数、失败回退和长期状态维护;这些能力文中未充分说明。
此外,benchmark 是否真正验证了 reasoning 仍存疑。很多任务可能通过学到特定视觉变换模板或数据分布 shortcut 获益,而不是形成通用规划能力。文中没有足够 evidence 证明 latent thoughts 在 out-of-distribution 操作组合、未见任务类型、或更长 reasoning horizon 上仍然可组合。
Limitation
方法成立的首要前提是:目标任务可以被预先拆成少数几类程序化视觉操作,并且这些操作生成的中间视图能覆盖推理所需状态。这对视觉搜索、图表局部重读、拼图验证是合理的,但对开放式物理推理、因果事件理解、视频长程规划、多主体交互未必成立。
第二,固定 K-step chain 是强 inductive bias,也是上限。它能稳定训练,但不具备动态规划能力。真实视觉推理往往需要条件分支、循环检查和不确定性驱动的操作选择;ProLaViT 的链条更多是离线定义的轨迹模板,而不是 inference-time 自主 planner。
第三,核心能力可能主要来自数据覆盖。程序合成数据把视觉操作、区域、顺序和答案关系都结构化好了,模型学到的可能是这些轨迹分布的压缩。若 benchmark 与合成轨迹高度同构,提升会显著;若遇到未见操作组合,泛化是否真实存在文中未充分说明。
第四,所谓 interpretability 也要谨慎。latent similarity matrix 只能说明步骤表示被拉开,不等于模型真的执行了可解释视觉操作。辅助视图监督给了 latent token 语义锚,但推理时这些 latent state 是否因果决定答案,而不是旁路 correlated feature,仍需要 intervention 级别实验。
第五,增益归因不完全清楚。curriculum、LoRA 训练、合成数据、多阶段格式训练、latent token 数量、teacher 选择和 diversity loss 同时变化。虽然 ablation 有帮助,但还不能精确回答:如果只用更强数据和普通 SFT,或只用 native encoder feature auxiliary loss,不做 progressive chain,会差多少。
Takeaway
- 第一,视觉推理中的 continuous CoT 不应只是“把文字 CoT 换成 latent token”,而应有可监督的中间视觉状态;否则很容易坍缩成无结构 embedding。
- 第二,latent supervision 的 teacher 不一定越强越好;与 base MLLM 表示空间一致的 endogenous teacher 可能比外部 expert 更有效。
- 这一点对多模态 agent、tool distillation、video latent planning 都值得迁移。
- 第三,程序化合成轨迹是当前让模型 internalize visual operations 的高性价比路线,但它的泛化上限取决于操作语言和任务覆盖。
一句话总结
ProLaViT 是一篇把程序化视觉操作轨迹蒸馏进 MLLM latent space 的工作,真正贡献在于用同源视觉表示和结构化 latent chain 将 visual CoT 从昂贵的显式工具调用推进到可训练、低成本的内部化视觉状态转移。
