← 论文解读

LoopVL:共享参数怎样在循环中重新读取视觉证据

视觉语言模型增加有效深度,是否必须增加同等规模的参数?LoopVL 让低层与高层模块循环更新多模态状态。本文从计算预算、训练和推理循环的匹配、视觉注意力变化出发,分析参数复用的收益,以及为什么更多循环不自动带来更好答案。

作者:Zhe Qian、Ziyang Gong、Zhongxing Xu、Hehan Li、Zhonghua Wang、Fei Luo、Mingxuan Wang、Xue Yang、Shiwei liu、Yanbiao Ma、Junchi Yan、Jungong Han
单位: Renmin University of China、Baidu、Shanghai Jiao Tong University、Monash University、ELLIS Institute Tübingen、Max Planck Institute for Intelligent Systems、TierFlow Team、Tsinghua University

1. Motivation:视觉推理的深度一定要靠更多参数吗?

看一张复杂图像,可能先定位目标,再根据问题重新检查文字、数量和空间关系。常规 Transformer 为不同深度存储不同参数,扩大有效计算通常也扩大模型。LoopVL 研究另一种选择:反复应用相同模块,让视觉语言状态继续变化。

它并不是让模型多写几段文字思考,而是在骨干内部循环。论文同时训练循环架构并检查中间状态,尝试说明共享层在不同轮次怎样改变读取视觉证据的方式。

LoopVL 通过三次低层模块与一次高层模块组成循环,并重复两轮

图 1. 模块循环与模型循环,以及 L/H 模块内部结构。图源:原论文 Figure 2。

2. Insight:共享参数节省存储,计算仍有成本

2.1 状态变化让同一层承担不同读取行为

相同变换作用在不同隐状态上,可以读取不同区域。第一轮没有关注的文字,第二轮可能因为问题状态已被整理而获得更高权重。论文用对齐的层和位置观察这种跨轮变化,称明显的视觉注意力重分配为 Visual Aha Moment。

这是注意力行为的命名,不能直接解释为人类式顿悟。注意力集中、位置排序改变和答案正确,是相关但不同的观察;一张热图也不足以确定某层永久承担某种语义功能。

2.2 推理循环应当与训练匹配

反复调用共享层可以增加有效深度,但模型没有自动学会任意轮数的状态更新。过早读出可能尚未形成合法答案,额外循环也可能使状态偏离训练分布。LoopVL 的消融恰好表明,多算几轮并不一定提高成绩。

3. Method:L/H 如何更新统一多模态状态?

3.1 两个模块,八次调用

冻结的 Penguin Vision Encoder 生成视觉特征,两层 GELU projector 将其映射到语言空间。L 和 H 分别包含 16 层 Transformer,两个模块各自共享参数,但 L 与 H 彼此不共享。

默认 H2L3 的顺序是:

(L→L→L→H)→(L→L→L→H).(L\to L\to L\to H)\to(L\to L\to L\to H).

一轮前向执行 128 次 Transformer 层调用,只存储两套共 32 层的参数。L 状态在较稳定的 H 条件下反复更新,H 再整合 L 的结果,成为下一轮的条件;最终从 H 状态生成答案。

图像、指令和头部 token 使用双向可见的 PrefixLM 前缀,回答只能读取前缀和已生成回答。视觉位置采用二维空间 RoPE;每轮开始还通过问题相关的门控重新注入初始视觉 anchor,帮助状态更新保留输入依据。

3.2 语言骨干从头训练,视觉编码器不是

语言预训练使用 75B tokens,随后进行视觉语言对齐、49B tokens 多模态中训练、16.81B tokens SFT 和视觉强化学习。对齐阶段只更新视觉接口,后续更新语言模型与接口,视觉编码器保持冻结。

默认训练使用选择性的反向传播,逐步把梯度图扩展到后面五次模块调用;更早调用仍做前向计算,但不参与该反向图。这个安排影响训练成本,所以不能从 128 次前向调用直接推算完整训练 FLOPs。

3.3 保留初始视觉和持续更新视觉是两件事

Anchor 提供固定输入参照,循环隐状态则持续处理它。论文还训练去掉 anchor 与 gate 的模型,以及全循环传播梯度的模型,跨轮注意力集中变化仍存在。这说明所观察现象不完全依赖某个门控或默认梯度路径。

4. Experiments:收益来自怎样的计算配置?

4.1 相同 token 预算下的架构对照

原论文 Table 1 将所有模型放在 0.14T 训练 token 预算下,使用相同 Transformer 块设计:

模型 训练 FLOPs(102110^{21}) MMStar RealWorldQA ChartQA
Transformer-VL 1B 0.92 55.33 55.29 51.12
Transformer-VL 4B Deep 2.89 61.33 66.54 75.24
Transformer-VL 4B Wide 2.99 60.47 66.01 75.56
LoopVL 2.47 63.47 70.98 74.52

LoopVL 全面超过单次执行的 1B 对照,但训练计算约为其 2.7 倍。相对 4B 模型,它在 MMStar 和 RealWorldQA 更高,ChartQA 略低,VMCBench 与 AI2D 也没有全面胜出。证据支持参数复用的竞争力,不能概括为在相同计算下无条件优于扩大模型。

4.2 训练时组织顺序有影响

Table 3 的各变体从初始化独立训练。H1L3 和 H2L1 都执行 64 次层调用,但 MMStar 分别为 58.13 和 60.80,ChartQA 为 60.40 和 65.12。相同展开深度不等于相同更新顺序,循环如何分配给 L/H 本身影响能力。

4.3 推理时增加循环可能退化

Table 4 固定 H2L3 checkpoint,改变推理顺序:

推理配置 层调用数 MMStar RealWorldQA
H2L1 64 29.40 12.81
H2L2 96 51.33 58.95
H2L3 128 63.47 70.98
H3L3 192 58.20 60.13
H4L3 256 24.00 14.51

提前从单轮读出还会大量产生不可解析响应。因此,这组结果描述的是固定 checkpoint 对推理调度的敏感性,不能用来否定独立训练的单轮模型,更不能宣称推理深度已经能够任意扩展。

4.4 注意力诊断提供机制线索

第二轮的视觉注意力熵下降、Gini 集中度上升,部分初轮低排序位置进入第二轮高排序集合。与此同时,视觉隐状态仍持续更新,最终回答分布在最后 H 调用中快速接近终态。

这些诊断区分总视觉注意力质量与图内分布:总质量减少,也可能同时更集中。热图中的参考框是人工标注,不是模型预测框;选定案例不能估计现象出现频率,也不能单独证明集中位置提供了答案所需的因果证据。

5. Conclusion

LoopVL 展示了共享参数可以支撑更深的多模态计算,并通过不断更新的状态改变同一层的视觉读取行为。最有价值的对照,是把参数、训练 token、FLOPs 和执行顺序共同列出来,才能判断模型究竟节省了什么。

后续可以用同计算训练的架构对照,结合区域替换和反事实图像检查循环中的证据变化,再学习可停止的更新机制。如果要动态分配循环,训练阶段也需要覆盖这些调度,不能只在推理时简单重复已经训练好的模块。

References

— END —浏览论文解读 →