← 返回 2026-04-21

OneVL:基于视觉-语言解释的单步潜在推理与规划 OneVL: One-Step Latent Reasoning and Planning with Vision-Language Explanation

Jinghui Lu, Jiayi Guan, Zhijian Huang, Jinlong Li, Guang Li, Lingdong Kong, Yingyan Li, Han Wang, Shaoqing Xu, Yuechen Luo, Fang Li, Chenxu Dang, Junli Wang, Tao Xu, Jing Wu, Jianhua Wu, Xiaoshuai Hao, Wen Zhang, Tianyi Jiang, Lingfeng Zhang, Lei Zhou, Yingbo Tang, Jie Wang, Yinfeng Gao, Xizhou Bu, Haochen Tian, Yihang Qiu, Feiyang Jia, Lin Liu, Yigu Ge, Hanbing Li, Yuannan Shen, Jianwei Cui, Hongwei Xie, Bing Wang, Haiyang Sun, Jingwei Zhao, Jiahui Huang, Pei Liu, Zeyu Zhu, Yuncheng Jiang, Zibin Guo, Chuhong Gong, Hanchao Leng, Kun Ma, Naiyang Wang, Guang Chen, Kuiyuan Yang, Hangjun Ye, Long Chen 📅 2026-04-20 👍 96 2026-07-13 08:36
世界模型 思维链压缩 潜在推理 端到端规划 自动驾驶 视觉-语言-动作模型

用世界模型辅助的视觉-语言双解码器监督潜在 CoT,让自动驾驶规划在答案级延迟下超越显式 CoT。

前置知识

Vision-Language-Action 模型 (VLA)

VLA 是把视觉编码器、多模态投影器和大型语言模型主干串联起来、最后直接输出动作(轨迹路点或控制信号)的统一模型。它继承了 VLM 的场景理解能力,并把语言推理与控制决策融合到一个端到端可训练的框架中,OneVL 主干就是 Qwen3-VL-4B-Instruct。

理解 VLA 的结构是看懂 OneVL 设计的入口:VLM 提供上下文推理能力,潜在 token 嵌在 assistant 响应中,端到端损失同时监督语言、视觉和轨迹输出。

Chain-of-Thought (CoT) 推理

CoT 让模型在给出最终答案前先生成自然语言推理步骤,迫使模型在 next-token 监督下压缩理解为结构化、可泛化的表征。在自动驾驶中体现为先描述场景再输出轨迹,可显著降低规划错误。

显式 AR CoT 是 OneVL 的性能上界(NAVSIM 上 88.29 PDM-score),也是论文要压缩、但又不能丢失的目标对象。理解 CoT 才有助于体会压缩驱动泛化这一中心假设。

Latent CoT / 隐式推理

把显式 CoT 替换为连续潜在向量(hidden states),例如 COCONUT 的课程学习、CODI 的自蒸馏、SIM-CoT 的辅助解码器。其目标是在保留推理能力的同时去掉逐 token 生成的开销。

OneVL 直接对标 COCONUT/CODI/SIM-CoT,指出它们在自动驾驶上全部失效(PDM-score 比 AR Answer 还低),原因是纯语言潜在表征压缩的是世界的符号抽象而非物理因果动态。

世界模型 (World Model)

学习环境状态转移的模型,给定当前观察和动作预测下一时刻状态。OneVL 的视觉辅助解码器本质上就是短期未来帧预测器(0.5s 和 1.0s 后的图像),充当世界模型辅助监督。

世界模型是 OneVL 的关键创新:它把潜在空间推向必须编码因果动态的方向,使得压缩目标不再是抽象语言,而是可以直接被未来帧重建损失验证的物理结构。

预填充 (Prefill) 推理

Transformer 把 prompt 一次性并行编码(prefill 阶段),再逐 token 解码答案。OneVL 把潜在 token 全部塞进 prefill 阶段,省去逐个生成它们的 AR 开销。

Prefill 是 OneVL 实现 4.46s 与 AR Answer 4.49s 几乎相同延迟的工程基础,也是它能宣称以答案级延迟完成 CoT 推理的核心机制。

压缩视角的智能 (Compression View of Intelligence)

Delétang 等人提出的假说:在 next-token 监督下,模型必须把对世界的理解压缩进有限表征中,越紧的压缩越能保留可泛化的因果结构。OneVL 的双解码器正是为了让压缩目标同时通过语言和视觉两路验证。

这是论文的中心理论支柱,所有设计选择(双模态辅助解码器、三阶段训练、world model 监督)都服务于让潜在空间承载真正可泛化的结构。

研究动机

在 VLA 自动驾驶中,显式 Chain-of-Thought 推理被证明能显著提升轨迹质量,例如在 NAVSIM 上 AR CoT+Answer 比 AR Answer 提升 0.80 PDM-score(88.29 vs 87.47)。但代价是推理延迟成比例增加:在 NAVSIM 上从 4.49s 膨胀到 6.58s,ROADWork 上从 4.74s 暴涨到 10.74s,Impromptu 上从 4.24s 升到 6.84s。在车规级实时系统里这是不可接受的。已有的 latent CoT 路线(COCONUT、CODI、SIM-CoT)正是为了消除这一延迟开销,但在自动驾驶上集体翻车:在 NAVSIM 上 COCONUT 仅 84.84、CODI 83.92、SIM-CoT 84.21,全部低于 AR Answer 的 87.47;在 ROADWork 上同样 COCONUT 15.44 / CODI 16.45 / SIM-CoT 16.49 的 ADE 都比 AR Answer 的 15.98 更差。作者认为这并非工程实现问题,而是设计假设出了问题:纯语言的潜在空间压缩的是世界的符号抽象(语义标签、关系描述),并不包含决定未来轨迹的时空因果动态。

本文的目标是论文的核心目标是设计一个 latent CoT 框架,既保留显式 CoT 带来的泛化能力,又把推理延迟压到 answer-only 水平,并在四个不同自动驾驶基准上同时做到快、准、可解释。具体地,OneVL 要在 NAVSIM 上拿到 88.84 PDM-score(超过 AR CoT 88.29 和此前 SOTA AdaThinkDrive 86.20 / LaST-VLA 87.30),并保持 4.46s 延迟(与 AR Answer 4.49s 几乎相同,是 AR CoT 6.58s 的约 1/1.5)。同时还要在 ROADWork、Impromptu、Alpamayo-R1 三个独立基准上同时取得 SOTA 或可比结果。

与已有工作不同的是,现有 latent CoT 工作的瓶颈是它们用语言描述作为压缩目标。作者观察到,自然语言对驾驶场景的描述本质上是抽象的——它编码语义标签和实体关系,但不编码位置、速度、加速度、交通参与者交互这类决定未来轨迹的几何因果结构。OneVL 的独特切入角度是引入一个视觉世界模型辅助解码器,让视觉潜在 token 必须能解码出 0.5s 和 1.0s 之后的未来帧视觉 token。这个未来帧预测目标是具体的、物理因果的、几何上紧贴轨迹预测的,从而把潜在空间从符号抽象强制推向场景因果动态。同时,论文设计了 Prefill 推理,让潜在 token 在 inference 时被一次性并行塞进 prompt,绕过 latent CoT 仍需逐 token 生成的剩余开销。配套的三阶段训练流程解决了主模型、视觉解码器、语言解码器优化目标冲突的问题,消融实验显示缺少三阶段训练会让 PDM-score 从 88.84 崩到 67.13。

核心方法

OneVL 的核心直觉是:压缩驱动泛化,但压缩目标必须同时通过语言和视觉两条路验证。具体技术路线是:在 Qwen3-VL-4B-Instruct 这个 VLA 主干的 assistant 响应里,插入两类特殊潜在 token——视觉潜在 token(用 4 段 $\langle|\text{latent-vis}|\rangle$ 表示)和语言潜在 token(用 2 段 $\langle|\text{latent}|\rangle$ 表示),用 delimiter 包裹;训练时主 VLM 仍按 next-token 预测损失 $\mathcal{L}_c$ 监督,但这些潜在位置上的 hidden state 会被抽出来,分别送入视觉辅助解码器和语言辅助解码器,分别用未来帧视觉 token 和 CoT 文本作为监督目标;推理时把两个辅助解码器扔掉,潜在 token 全部 prefilled 进 prompt,单次并行通过主模型,再 AR 解码轨迹,整个过程的延迟与 AR Answer 几乎相同。整体框架由一个预备自监督预训练和三个正式训练阶段组成,确保三组件(主 VLM、视觉解码器、语言解码器)从不同起点平稳对齐。

OneVL 的核心创新是把世界模型内嵌为潜在 token 的辅助监督器,而不像 DriveVLA-W0 等工作那样把世界模型作为独立表示学习栈或数据生成器。COCONUT/CODI/SIM-CoT 等 latent CoT 方法只压缩语言描述,得到的潜在空间只能解释语义意图;OneVL 通过预测未来 0.5s 和 1.0s 帧的 IBQ 离散视觉 token,强制视觉潜在 token 编码道路几何演化、agent 轨迹、emerging hazard 等时空因果结构。第二个本质区别是 Prefill 推理:prior latent CoT 的隐藏状态仍按 AR 顺序逐个生成,OneVL 把这些潜在 token 的具体词表身份固定下来、一次性塞进 prefill 阶段,从而真正实现一个并行前向 = 一次推理。第三个本质区别是三阶段训练:先让主 VLM 在轨迹任务上预热 latent 位置,再冻结主 VLM 单独训练两个辅助解码器,最后才解冻所有参数联合优化,这避免了视觉解码器在 latent 还无信息时被迫预测未来帧导致的梯度爆炸。

方法步骤详情

具体步骤分四步。第一步是预备的视觉辅助解码器自监督预训练(13,040 个 optimizer step、global batch size 256、lr=1e-4):暂时不接入主 VLM,只让视觉解码器从当前帧 ViT embedding 预测 +0.5s 和 +1.0s 的未来视觉 token,损失为 $\mathcal{L}_p = -\sum_{t=1}^{|T^v_y|} \log P_{D_v}(T^v_{y,t} | V, T^v_{y,<t})$,让解码器先学到强大的视觉动态先验。第二步是 Stage 0 主模型预热(2 个 epoch、lr=4e-5、batch 64):在 assistant 响应里插入 $\langle|\text{start-latent-vis}|\rangle$ + 4 个 $\langle|\text{latent-vis}|\rangle$ + $\end$ + $\langle|\text{start-latent}|\rangle$ + 2 个 $\langle|\text{latent}|\rangle$ + $\end$ + 轨迹答案,训练 ViT、LLM、Aligner 在 $\mathcal{L}_c$ 下学习让这些 latent 位置产生有意义的 hidden state。第三步是 Stage 1 辅助解码器预热(1 个 epoch、lr=1e-4):冻结主 VLM,只训练 $D_l$ 和 $D_v$,分别最小化 $\mathcal{L}_l = -\sum_{i=1}^{|T^l_y|} \log P_{D_l}(T^l_{y,i} | Z_l, T^l_{y,<i})$ 和 $\mathcal{L}_v = -\sum_{t=1}^{|T^v_y|} \log P_{D_v}(T^v_{y,t} | Z_v, T^v_{y,<t})$,其中 $Z_l = [W_l(V), W_l(H_l)]$、$Z_v = [W_v(V), W_v(H_v)]$,使解码器在稳定的 latent 分布上对齐。第四步是 Stage 2 端到端联合微调(5 个 epoch、lr=1e-4、batch 64、$\lambda_l=1.0$、$\lambda_v=0.1$):解冻所有参数,最小化 $\mathcal{L} = \mathcal{L}_c + \lambda_l \mathcal{L}_l + \lambda_v \mathcal{L}_v$,让 latent 在三个目标之间形成良性循环。视觉 token 化采用 Emu3.5 IBQ tokenizer,码本大小 131,072,离线生成训练目标。

技术新颖性

技术新颖性主要体现在四个层面。第一,把世界模型监督从独立数据生成器/模拟器转译为潜在 bottleneck 的训练期验证器:传统世界模型用未来帧做数据增广或闭环评估,OneVL 让未来帧视觉 token 重建成为潜在空间是否学到因果动态的直接判据。第二,把压缩驱动泛化这一智能假说落到双模态解码器设计上:语言解码器保证语义意图,视觉解码器保证物理因果,两者共同保证潜在空间既不过于抽象(仅语言)也不过于具体(仅像素),而是承载可泛化的共享因果结构。第三,工程上的 Prefill 机制:把 latent token 的具体词身份(empirically chosen, e.g. 35 个视觉 token + 20 个语言 token)固定下来,作为 prompt 一部分并行处理,绕开了 prior latent CoT 仍 AR 逐 token 生成的关键瓶颈。第四,三阶段训练作为可复用优化范式:消融显示直接联合优化会导致梯度范数从 0.28 暴增到 378.22、PDM-score 跌 21.71 分,证实了 staged curriculum 的必要性,这一发现对其他多模块多目标协同训练都有借鉴意义。

Comparison of CoT paradigms. (a) Explicit CoT (b) Implicit CoT (Existing Work) (c) OneVL (Ours).
Figure 2: Comparison of CoT paradigms. (a) Explicit CoT (b) Implicit CoT (Existing Work) (c) OneVL (Ours).
OneVL architecture. An image and structured text prompt (ego state, command, historical trajectory) are fed into the VLM.
Figure 3: OneVL architecture. An image and structured text prompt (ego state, command, historical trajectory) are fed into the VLM.
Visual CoT under full training vs. an overfitting visual auxiliary decoder.
Figure 8: Visual CoT under full training vs. an overfitting visual auxiliary decoder.

实验结果

实验在四个基准上一致表明 OneVL 是首个超越显式 CoT 的 latent CoT 方法。NAVSIM(Table 1)上 OneVL 拿到 88.84 PDM-score,超过 AdaThinkDrive 86.20 (+2.64) 和 LaST-VLA 87.30 (+1.54),同时比 AR CoT+Answer 88.29 高 0.55 分;延迟 4.46s 几乎与 AR Answer 的 4.49s 持平,仅为 AR CoT+Answer (6.58s) 的 67.8%。ROADWork(Table 2)上 OneVL 取得 12.49 ADE / 28.80 FDE(像素),比之前 SOTA YNet 22.68/80.78 下降近一半,比 AR CoT+Answer 13.18/29.98 也明显更好;延迟 4.71s 仍与 AR Answer 4.74s 同档。Impromptu(Tables 3-4)上 OneVL ADE 1.34m / FDE 3.70m,比 Impromptu VLA 1.60/4.28 显著提升;L2 误差分时段 1s 0.13 / 2s 0.48 / 3s 1.18 / 4s 2.25,平均 1.01 优于 AR CoT 1.09。Alpamayo-R1(Table 5)上 OneVL ADE 2.62m 优于 Cosmos-Reason 2.86m,FDE 7.53m 略输 7.42m(作者解释 Cosmos-Reason 用了 RL 后训练)。解释质量(Table 6)显示 OneVL 的语言辅助解码器在 Meta Action Acc 上拿到 71.00(SIM-CoT 67.20、AR CoT 73.20),STS 78.26、LLM Judge 79.13,平均 76.13 vs SIM-CoT 74.06。消融(Table 7)拆出视觉解码器贡献 +0.87(87.97→88.84)、语言解码器贡献 +0.31(88.53→88.84)、三阶段训练贡献 +21.71(67.13→88.84),后者证明没有 staged train 模型直接崩溃。Figure 8 进一步显示,没有三阶段训练时视觉解码器解出的 T+1/T+2 帧与输入场景毫无关系,CoT 也退化为模板化错误输出。Figure 1 把四个基准上的准确率 vs 延迟拉到同一图上,OneVL 是唯一同时落在左下高准确率、低延迟区间的点。

Performance comparisons on the NAVSIM benchmark.
Table 1: Performance comparisons on the NAVSIM benchmark.
Performance comparisons on the ROADWork benchmark.
Table 2: Performance comparisons on the ROADWork benchmark.
Performance comparisons on the Impromptu benchmark (ADE/FDE).
Table 3: Performance comparisons on the Impromptu benchmark (ADE/FDE).
Performance comparisons on the Impromptu benchmark (Traj. Pred. L2 Error).
Table 4: Performance comparisons on the Impromptu benchmark (Traj. Pred. L2 Error).
Performance comparisons on the Alpamayo-R1 benchmark.
Table 5: Performance comparisons on the Alpamayo-R1 benchmark.
Text CoT quality on the NAVSIM test set.
Table 6: Text CoT quality on the NAVSIM test set.
Ablation study on each component in OneVL.
Table 7: Ablation study on each component in OneVL.
Accuracy-latency trade-off for real-time deployment on NAVSIM.
Table 8: Accuracy-latency trade-off for real-time deployment on NAVSIM.
Training hyperparameters for OneVL.
Table 9: Training hyperparameters for OneVL.
Accuracy and efficiency comparison across four benchmarks. Existing latent CoT methods underperform explicit CoT. OneVL is the first to surpass it while matching answer-only prediction latency.
Figure 1: Accuracy and efficiency comparison across four benchmarks. Existing latent CoT methods underperform explicit CoT. OneVL is the first to surpass it while matching answer-only prediction latency.
Visualizations of prediction on NAVSIM.
Figure 4: Visualizations of prediction on NAVSIM.
Visualizations of prediction on ROADWork.
Figure 5: Visualizations of prediction on ROADWork.
Visualizations of prediction on Impromptu.
Figure 6: Visualizations of prediction on Impromptu.
Visualizations of prediction on Alpamayo-R1.
Figure 7: Visualizations of prediction on Alpamayo-R1.
查看结构化数据
任务指标本文基线提升
NAVSIM 规划 (PDM-score) PDM-score (越高越好) 88.84 AdaThinkDrive 86.20 / LaST-VLA 87.30 / AR CoT+Answer 88.29 相比此前 SOTA LaST-VLA +1.54;相比 AR CoT+Answer +0.55;视觉解码器贡献 +0.87,语言解码器贡献 +0.31
NAVSIM 推理延迟 平均延迟(秒,越低越好) 4.46s AR Answer 4.49s / AR CoT+Answer 6.58s Prefill 后与 AR Answer 几乎相同,是 AR CoT 的 67.8%
ROADWork 规划 (ADE/FDE) ADE / FDE (像素,越低越好) 12.49 / 28.80 YNet 22.68/80.78;AR CoT+Answer 13.18/29.98 相比 YNet ADE 降低 10.19 像素、FDE 降低 51.98 像素;相比 AR CoT ADE 0.69、FDE 1.18
Impromptu 规划 (ADE/FDE) ADE / FDE (米,越低越好) 1.34 / 3.70 Impromptu VLA 1.60/4.28;AR CoT+Answer 1.42/3.96 相比 Impromptu VLA ADE -0.26m、FDE -0.58m;相比 AR CoT ADE -0.08m、FDE -0.26m
Impromptu 轨迹 L2 误差 1s/2s/3s/4s L2 + 平均 (米) 0.13 / 0.48 / 1.18 / 2.25, Avg 1.01 AR CoT+Answer 0.13/0.51/1.27/2.44, Avg 1.09 平均 L2 降低 0.08m;3s/4s 长程预测提升最明显
Alpamayo-R1 规划 (ADE/FDE) ADE / FDE (米,越低越好) 2.62 / 7.53 Cosmos-Reason 2.86/7.42;AR CoT+Answer 2.99/8.54 ADE -0.24 vs Cosmos-Reason;FDE 略输 0.11(Cosmos-Reason 用了 RL 后训练)
CoT 解释质量 (NAVSIM) Meta Action Acc / STS / LLM Judge 71.00 / 78.26 / 79.13 SIM-CoT 67.20/76.25/78.73;AR CoT+Answer 73.20/79.75/81.86 相比 SIM-CoT 三个指标全面 +3.8/+2.01/+0.40;与 AR CoT 差距收敛到 2.20/1.49/2.73
MLP 头部部署延迟 (NAVSIM) PDM-score / 延迟 (秒) 86.83 / 0.24s (≈4.16 Hz) AR CoT+Answer – / 4.46s 延迟仅为 AR 模型的 5.4%,并保持 86.83 PDM-score

局限与改进

作者承认的局限性主要有三点。第一,训练时需要同时容纳三个 4B 模型实例(主 VLM、视觉解码器、语言解码器),整体显存需求约为基线的 3×,仅靠 DeepSpeed ZeRO-2 缓解,对基础设施仍有较高门槛。第二,视觉/语言潜在 token 数量($C_v=4$、$C_t=2$)以及对应的 35/20 个词表身份是基于经验选择的,论文未给出 token 数量与表征能力之间的系统消融。第三,轨迹 token 本身仍以自回归方式生成,即便 Prefill 省掉了 latent CoT 的 AR 开销,剩余 AR 解码仍是车规部署的主要延迟来源。基于这些论文结果,独立观察还可补充两点:第四,超越显式 CoT 的优势在不同基准并不一致——Alpamayo-R1 FDE 上 OneVL 7.53m 反输 Cosmos-Reason 7.42m,说明 world model 监督的收益与数据集/标注噪声/RL 后训练都有关。第五,Prefill 即并行依赖于具体的 transformer 内核实现(FlashAttention 等),在不同推理框架(TensorRT-LLM、ONNX Runtime)下能否保持 4.46s≈4.49s 的等价延迟还需独立验证。

独立分析的弱点

从独立视角,OneVL 主要有四个可改进的弱点。第一个是视觉解码器目标分辨率受限于 IBQ tokenizer 的 512×512 输入,未来帧的细节(远处行人、信号灯颜色)很可能在 token 化时已损失,这解释了为何 1.0s 远处预测的视觉解释仍然较粗,并限制了对快速突变场景(如突然窜出的非机动车)的反应。改进方向是用更高分辨率 tokenizer 或金字塔式多尺度 token 化。第二个是双模态解码器只在训练时存在,意味着训练-推理分布的 latent 编码方式存在隐式不一致(推理时没有解码器反馈信号),长期看可以加入解码器蒸馏回主模型的机制,让推理时的输出也对齐训练时的语义。第三个是 CoT 标签质量参差——NAVSIM 用 AdaThinkDrive 标注、ROADWork 走自建 pipeline、Alpamayo-R1 用官方 checkpoint 自跑、Impromptu 用原数据集 Q&A 加 root-cause 增强,标签噪声在各基准上分布不同,这可以解释 Alpamayo-R1 FDE 上 OneVL 反输的原因;改进方向是统一多源 CoT 标签协议或加入 CoT 标签去偏模块。第四个是 staged training 的三阶段需要精细的超参(学习率 4e-5→1e-4、batch 64),迁移到其他 VLA 主干(如 LLaVA、InternVL)时需要重做 sweep,缺乏一键迁移能力。

未来方向

作者明确点出三条未来方向。第一,轨迹 AR 解码仍是主要延迟瓶颈,应当探索并行或非自回归的轨迹解码(例如 flow matching、扩散式轨迹生成),把 OneVL 的 0.24s MLP 部署路径推到端到端。第二,把视觉世界模型解码器扩展到多相机输入,实现 360° 未来场景预测,这能进一步强化对侧方切入车辆、盲区行人的因果建模。第三,把双模态解释(视觉 + 文本)作为人机交互界面和 RL 训练信号——驾驶员可看到未来帧和 CoT,系统可把视觉/语言解释作为 reward shaping 的高阶信号。除作者提出的方向外,还可以延伸出几条思路:用 OneVL 的 latent bottleneck 作为多任务预训练 backbone,复用到预测、感知、问答等任务;引入更长的未来帧预测(2s、3s)测试 world model 监督的边际收益是否饱和;以及探索更紧凑的潜在 token 设计(如 dynamic token pruning)。

复现评估

复现性整体较好但有一定门槛。论文明确给出 GitHub 仓库 https://github.com/xiaomi-research/OneVL 和项目页 https://Xiaomi-Embodied-Intelligence.github.io/OneVL,所有 CoT 标注的构建细节在 Appendix A.3(NAVSIM/ROADWork/Impromptu/Alpamayo-R1 四套独立 pipeline)和 A.4(LLM-as-Judge 提示词)公开。训练超参列在 Table 9:预备预训练 13,040 steps + global batch 256,Stage 0/1/2 分别 2/1/5 epoch、batch 64,$\lambda_l=1.0$、$\lambda_v=0.1$,统一 BF16 + ZeRO-2 + AdamW + Cosine。数据规模方面,NAVSIM/ROADWork/Impromptu/Alpamayo-R1 各自规模未在主文给出,但 Alpamayo-R1 是从可用视频片段里 subsample 700 例子作 test。算力方面 4B 规模 + 三组件 + 13k 步预备训练估计需要 32-64 张 A100/H100 几天到一周。复现难度中等偏上:潜在 token 数量、词表身份(35/20)是经验性的,三阶段顺序和超参都需重做 sweep;视觉 tokenizer 的离线生成也需要复现 IBQ 流程。