← 返回 2026-10-15

VITAL:从视觉运动到精确动作 VITAL: From Visual Motion to Precise Action

Ao Liang, Youquan Liu, Mingxiao Li, Wei Chow, Lingdong Kong, Yue Liao, Dongyue Lu, Yu Yang, Wei Yin, Runnan Chen 📅 2026-10-14 👍 7 2026-11-11 18:30
世界-动作模型 动作分词器 残差强化学习 流匹配 自动驾驶

视觉骨架定主目标、度量残差补精度,一步流匹配生成轨迹,残差RL有界精修

前置知识

世界-动作模型(World-Action Model)

把视频预测(世界模型)与动作生成(策略)耦合的驾驶框架,通常采用混合Transformer结构:动作token通过注意力机制读取视频专家的特征后再解码自车轨迹,代表工作有Drive-WM、DriveLaW、SUV、SimWAM。理想是“模型学会预测世界如何运动,就顺带学会如何行动”。

本文的核心批评正是针对这类模型:两个专家只交换特征、不共享预测目标,动作专家学的仍是独立于视觉动态的轨迹坐标编码,这就是所谓“视觉-动作表示鸿沟”,理解这一背景才能明白VITAL为何要改造动作目标本身。

流匹配(Flow Matching)

一种生成式建模方法:学习速度场 $v_\theta(z_t, t)$,把高斯噪声沿插值路径 $z_t = (1-t)\epsilon + t z^\star$ 输运到数据分布,训练目标是回归目标速度 $u = z^\star - \epsilon$,推理时用欧拉积分从噪声采样到数据,积分步数越少推理越快但误差越大。

VITAL的动作专家就是一个流匹配模型,“单步欧拉采样即可生成高质量轨迹”是其低延迟卖点的技术基础,论文还专门推导了单步积分误差上界来解释为什么一步就够。

动作分词器(Action Tokenizer)

把连续动作序列压缩成紧凑latent码、再解码还原的模块:FAST用离散化、VQ-BeT用向量量化、LAPA/RepWAM从视觉状态转移中推断latent action。tokenizer决定了策略要预测的目标的几何结构和信息容量,直接影响下游规划精度与可预测性。

VITAL的本质贡献就是一个新的动作tokenizer——视觉骨架码加配对度量残差码,全部三个训练阶段(目标构造、模仿学习、残差RL)都围绕这一表示展开,读懂tokenizer设计就读懂了全文。

残差策略学习(Residual Policy Learning)

在已训练好的基础策略之上学习一个修正项,最终动作等于基策略输出加上修正 $\tau_\phi = \tau_{IL} + \Delta$,由Johannink等人在2019年系统化。修正空间通常被约束以保证稳定性和安全性,训练信号来自与环境的奖励交互。

VITAL的残差RL把修正严格限制在tokenizer自带的残差PCA坐标里,并用径向限制器把解码位移限幅在1米/0.2弧度内,零修正时精确还原基策略——这是其“结构保持后训练”区别于一般RL微调的关键。

NAVSIM基准与PDMS指标

NAVSIM是基于nuPlan/OpenScene数据的非反应式驾驶规划仿真基准。v1的PDMS由无责碰撞(NC)、可行驶区域合规(DAC)、自车进度(EP)、碰撞时间(TTC)、舒适度(C)五个子项(多为乘性门控)综合而成;v2的EPDMS扩展为九个子项并修正了打分协议,navhard则用两阶段反应式续接场景考察执行。

论文所有核心数字(90.8 PDMS、91.1 EPDMS、navhard 38.6)都在这套协议上报告,且DAC等乘性门控正是“11.4厘米修正导致PDMS从98.7归零”这一安全案例的机制来源,不理解指标无法解读结果与局限。

研究动机

现有世界-动作模型(DriveLaW、SUV、SimWAM等)用注意力把视频专家和动作专家连接起来,但两者没有共享的预测目标:视频专家预测场景演化,动作专家却回归一个只由轨迹坐标定义的编码,同一段物理运动被两套独立构造的表示描述,形成“视觉-动作表示鸿沟”。作者用一个刻意简化的实验量化这一鸿沟的代价:把问题退化为“给定录制好的当前帧和未来帧、重建自车轨迹”(完全不需要预测未来),在NAVSIM全部18,179个验证场景上,纯视觉重建的平均位移误差(ADE)仍高达123.6 mm、末端误差(FDE)221.6 mm,重建轨迹的高频速度能量占比(HF)仅0.1%——得到的是一条平滑的“运动骨架”,丢失了厘米级度量细节;而在道路边界附近,几厘米的偏差就足以让看似合理的轨迹违反可行驶区域约束(论文后文给出一个11.4 cm修正使PDMS从98.7归零的实例)。

本文的目标是本文的目标是设计一个动作表示,让“视觉运动”成为动作预测的主目标、同时不牺牲执行精度,并且部署时不需要生成未来图像。具体拆成三个子目标:其一在表示层,把视觉可恢复的平滑骨架与视觉恢复不了的度量残差分开编码,通过PCA白化让厘米级小修正获得独立、标准化的学习尺度;其二在预测层,用流匹配从当前图像、自车状态和导航指令联合预测两组码,单步采样即可输出可执行的四秒轨迹;其三在改进层,在冻结基策略与编解码器的前提下,复用残差坐标做有界强化后训练,部署时只附加一次确定性修正,无需奖励评估、候选搜索或额外去噪。

与已有工作不同的是,本文的独特切入角度有三点。第一是诊断先行:作者不是直接提方法,而是先用“录制未来重建”这个比世界-动作建模更简单的任务,把“表示容量”与“预测难度”分离,证明即使未来已知,视觉重建也只剩平滑骨架(HF仅0.1%),说明精度缺口本质上是表示问题而非预测问题。第二是表示构造的差异:与RepWAM、LingBot-VA 2.0、RLA-WM等在视觉特征空间重构视觉过渡的latent action不同,VITAL的残差目标是“视觉解码轨迹骨架与示范轨迹之间的物理坐标度量差”,轨迹监督把骨架锚定在真实世界尺度上。第三是一份表示服务三个阶段:残差不只是重建补丁,它同时充当流匹配的联合预测目标之一和后续有界RL的修正坐标系,把目标构造、模仿学习、策略改进统一在同一套可执行表示之下。

核心方法

直觉上,让“从画面里能看出来的运动”承担轨迹主干,把“视觉说不清的那几厘米”单独装进一个小口袋,并给这个小口袋独立的标准化尺度——主干有视觉依据所以好预测,修正小而精所以有独立量纲。技术路线分三阶段:Stage I 构造目标——冻结的视觉分词器 $E_v$(VGGT-Ω骨干加LoRA,读取0/1/2/3/4秒五帧)产出4个64维运动token,解码成40点@10Hz的轨迹骨架 $\tau^{v^\star}$;对骨架误差 $e^\star = \mathrm{vec}(\tau - \tau^{v^\star})$ 做PCA得统计量 $\mu, U, S$,配对残差码为 $z_r^\star = S^{-1}U^\top(e^\star - \mu)$,每秒区间12个系数,构成4×76的动作接口。Stage II 模仿学习——动作专家用流匹配从部署可见的上下文 $h(c)$ 联合预测两组码,损失包含流匹配项、视频流项和解码轨迹项(残差支路加stop-gradient),未来视频token被因果掩码隔离。Stage III 残差RL——冻结动作生成器与整个codec,训练一个123万参数的小actor,在6维有界修正空间里用NAVSIM打分器做对偶高斯策略梯度,部署时只施加一次均值修正。

核心创新是把动作表示按信息来源拆分:视觉骨架码由录制未来图像导出、承担主预测目标,使动作学习与视觉动态对齐;度量残差码编码骨架重建误差、经PCA系数标准化后给厘米级修正独立的学习尺度。与已有工作的本质区别有三。其一,RepWAM、LingBot-VA 2.0、RLA-WM的latent/residual都在视觉特征空间里重构视觉变化,而VITAL的残差目标是物理坐标下的轨迹度量差,配对后可达亚毫米级复原(ADE 3.11×10⁻⁴ m)。其二,主流少步生成方法(MeanFlow、Consistency Policy、Shortcut Models)靠改变流场或蒸馏实现少步采样,VITAL保留标准流匹配目标、只更换喂给它的动作目标,配对表示天然适配一步预测。其三,残差RL被严格限制在tokenizer自带的残差坐标的6维子空间且限幅1 m/0.2 rad,修正与模仿学习共享同一物理解码映射 $D(z_v, z_r) = D_v(z_v) + \mathrm{unvec}(\mu + M z_r)$,零修正精确还原基策略,这是“结构保持”的含义。

方法步骤详情

第一步,拟合视觉分词器:VGGT-Ω骨干经LoRA适配五帧前视图,时间编码器融合相机特征、register摘要与相对相机运动特征,输出4个64维token;宽256解码器预测车体系速度与偏航率,积分到40点@10Hz轨迹网格,训练后整体冻结。第二步,拟合配对残差:对训练集骨架误差 $e^\star = \mathrm{vec}(\tau - \tau^{v^\star})$ 按秒区间做PCA,扫描秩{2,…,30}后选12维(均值误差0.31 mm),统计量 $\mu, U, S$ 冻结,构成4×76动作接口。第三步,模仿学习:从SimWAM的IL权重出发(继承13万更新预训练),最终阶段2.6万更新、AdamW峰值学习率 $2\times10^{-5}$、batch 64、8×H200;动作专家只经首帧key/value适配器读当前帧特征,因果掩码隔离未来视频token;流匹配权重视觉:残差=1:2,外加解码轨迹损失(残差支路stop-gradient)。第四步,一步推理:单次欧拉更新 $\hat z^{(1)} = \epsilon + v_\theta(\epsilon, 0, h(c))$,FP32解码出4秒轨迹。第五步,残差RL:6个解析扰动模板(纵向/横向 $\rho^p$,$p \in \{2,3,4\}$)投影进残差解码空间并白化得 $B \in \mathbb{R}^{48\times6}$,径向限制器强制1 m/0.2 rad限幅;actor(4528→256→256→6,1.23M参数)读冻结策略隐状态与上下文,对17,013个难场景做对偶高斯策略梯度(每场景8对、$\sigma=1$),6,598个高分anchor做零修正正则,1600次更新共约41万次奖励查询,部署时 $\xi=0$ 只加一次均值修正。

技术新颖性

技术新颖性体现在四个层面。表示层:首次在驾驶动作表示中把“视觉可恢复的运动”与“视觉不可恢复的度量细节”显式分离为双码,并用PCA系数标准化 $S$ 让小修正获得与大运动相当的梯度量级。理论层:给出三元误差分解 $\|\tau - \hat\tau\|_Q \le \varepsilon_{\mathrm{codec}} + L_v\|\hat z_v - z_v^\star\|_F + \|M\|_{2\to Q}\|\hat z_r - z_r^\star\|_2$,把表示容量、视觉预测误差、残差预测误差分开分析,并推导单步欧拉积分误差上界 $\|e_D(z_1) - e_D(z_E)\|_Q \le L_D \sup_t \|a(t)\|_2 / 2$,为“一步够用”提供理论支撑。训练层:证明不改流匹配目标、只换动作目标即可获得一步生成能力,避免了蒸馏(Consistency Policy)或候选集成(DRIFT聚合48个提案)。后训练层:残差RL把干预限制在tokenizer残差坐标的6维有界子空间,基策略、视觉前缀、解码器全部冻结,训练用NAVSIM打分器而部署零奖励开销,与AD-R1需要学习世界模型内部critic的做法形成鲜明对比。

VITAL learns a shared representation for action prediction and refinement. The upper stage constructs visual motion tokens and paired metric residuals from recorded images and trajectories. The lower stage predicts the joint code in one flow step, applies a bounded residual correction, and decodes the executable trajectory. The diagram shows the driving implementation.
Figure 3: VITAL learns a shared representation for action prediction and refinement. The upper stage constructs visual motion tokens and paired metric residuals from recorded images and trajectories. The lower stage predicts the joint code in one flow step, applies a bounded residual correction, and decodes the executable trajectory. The diagram shows the driving implementation.
Clean-code trajectory reconstruction. Visual tokens recover the maneuver, while the paired residual corrects local path error in two turning cases and one near-straight case.
Figure A1: Clean-code trajectory reconstruction. Visual tokens recover the maneuver, while the paired residual corrects local path error in two turning cases and one near-straight case.

实验结果

核心结果分五块。(1) 主基准:NAVSIM v1 navtest上VITAL单步(NFE=1)达90.8 PDMS,十步91.0,延迟149 ms对490 ms(3.29×加速,含图像编码);RL前90.2/90.3,RL后+0.6/+0.7;对比SimWAM(IL)†单步仅66.6、SUV†单步89.7、ReCogDrive 90.8(5 NFE)、SGDrive 91.1、SimWAM(RL) 91.5(10 NFE)。(2) 迁移:v2 navtest修正EPDMS 91.1@1步(91.2@10),超十步SUV的91.0与Metis Top-6的90.3;navhard反应式两阶段38.6@1步(38.1@10),高于SimWAM 37.6、SUV 36.9、GeoWAM 36.6、PerceptDrive 34.5。(3) 重建容量:录制图像+真值残差把ADE从0.12 m降至3.11×10⁻⁴ m、PDMS 92.1→94.7;打乱视觉码ADE暴涨至7.41 m,只打乱残差仅0.19 m,证明骨架定路径、残差补细节。(4) 可预测性探针:仅自车/导航输入时AE的ADE(1.08 m)优于VITAL(1.62 m),加入匹配视觉后反转为VITAL 0.72 m/79.4 PDMS对AE 0.80 m/73.2;打乱图像后VITAL跌至2.20 m/52.3,说明优势来自真实视觉对应。(5) 公平动作学习:同初始化、同数据一轮训练后VITAL 70.1对AE 64.9对原始轨迹55.0 PDMS(十步71.4对66.3),单步+5.1,增益主要来自进度与TTC。定性上(Figure 5)弯道场景VITAL单步100 PDMS、ADE 0.67/FDE 1.82 m,优于SimWAM-RL十步的58.3、1.35/3.77 m。

NAVSIM v1 navtest. "Frame" counts include the current time step; "C/L" denote camera/LiDAR inputs, and "NFE/plans" denotes generator evaluations/candidate counts.
Table 1: NAVSIM v1 navtest. "Frame" counts include the current time step; "C/L" denote camera/LiDAR inputs, and "NFE/plans" denotes generator evaluations/candidate counts.
NAVSIM v2 transfer. One-stage navtest is on the left and two-stage navhard on the right. NFE/plans denotes generator evaluations/candidate counts; dashes mark non-iterative predictions, n.r. unreported values, and AR autoregression.
Table 2: NAVSIM v2 transfer. One-stage navtest is on the left and two-stage navhard on the right. NFE/plans denotes generator evaluations/candidate counts; dashes mark non-iterative predictions, n.r. unreported values, and AR autoregression.
Ablation on recorded-image reconstruction. Residuals use ground truth.
Table 3: Ablation on recorded-image reconstruction. Residuals use ground truth.
Ablation on feature probes. Pairs add matched/shuffled vision to ego/navigation.
Table 4: Ablation on feature probes. Pairs add matched/shuffled vision to ego/navigation.
Ablation on one-step action learning. All targets share action-core initialization.
Table 5: Ablation on one-step action learning. All targets share action-core initialization.
Residual-rank reconstruction. Errors are in millimeters. Bold marks the chosen operating rank.
Table A1: Residual-rank reconstruction. Errors are in millimeters. Bold marks the chosen operating rank.
Temporal energy proportions (%). LF and HF denote low and high frequencies of decoded trajectories.
Table A2: Temporal energy proportions (%). LF and HF denote low and high frequencies of decoded trajectories.
NAVSIM v2 navhard. S1/S2 denote the reactive stages. The last column gives mapping-wise two-stage EPDMS per method.
Table A3: NAVSIM v2 navhard. S1/S2 denote the reactive stages. The last column gives mapping-wise two-stage EPDMS per method.
Planning quality, inference latency, and action learning.
Figure 4: Planning quality, inference latency, and action learning.
A curved-road scene with local trajectory insets. VITAL uses one sampling step, SUV and SimWAM-RL use ten, and DiffusionDrive uses two. ADE and FDE are in meters.
Figure 5: A curved-road scene with local trajectory insets. VITAL uses one sampling step, SUV and SimWAM-RL use ten, and DiffusionDrive uses two. ADE and FDE are in meters.
Additional cross-model comparisons with shared local trajectory enlargements. ADE and FDE are in meters.
Figure A8: Additional cross-model comparisons with shared local trajectory enlargements. ADE and FDE are in meters.
查看结构化数据
任务指标本文基线提升
NAVSIM v1 navtest 闭环规划 PDMS 90.8(NFE=1,单条轨迹);91.0(NFE=10) DiffusionDrive 88.1(2 NFE/20提案);SimWAM(IL)† 66.6@1步;ReCogDrive 90.8(5 NFE);SimWAM(RL) 91.5(10 NFE) 单步即超越多数多步方法,较SimWAM(IL)†单步提升24.2分;距最强SimWAM(RL)@10步差0.7分但只需1/10计算
NAVSIM v2 navtest(修正EPDMS,v1权重直接迁移) EPDMS 91.1(1步);91.2(10步) SUV 91.0(10步);Metis Top-6 90.3(10步/6提案);EponaV2 88.9 较十步SUV +0.1,较Metis Top-6 +0.8,且采样步数为其1/10
NAVSIM v2 navhard 反应式两阶段续接 EPDMS(两阶段映射) 38.6(1步);38.1(10步) SimWAM 37.6(10步);SUV 36.9(10步);GeoWAM 36.6;PerceptDrive 34.5(25 NFE) 较最强基线SimWAM@10步 +1.0,且无需重训练
录制未来图像轨迹重建(18,179验证场景) ADE 3.11×10⁻⁴ m(骨架+真值残差);纯骨架0.12 m 纯视觉骨架重建 0.12 m(123.6 mm) 约390倍精度提升,PDMS同步从92.1升至94.7
单步动作学习公平对比(同初始化、同数据一轮) PDMS 70.1(VITAL目标,ADE 1.16 m / FDE 2.13 m) Trajectory AE 64.9;原始轨迹回归 55.0 较AE +5.2、较原始目标 +15.1;十步时仍领先(71.4 vs 66.3)
端到端推理延迟(RTX 4090,batch 1,含图像编码与残差修正) 中位延迟 149 ms(NFE=1) 490 ms(NFE=10,同一策略) 3.29×加速,PDMS仅损失0.2分(90.8 vs 91.0)

局限与改进

作者承认的局限:其一,有界残差修正只做局部精修,无法挽救错误的高层机动选择——navhard上残差RL改善Stage 1却轻微降低Stage 2 EPDMS,说明场景因自车行为演化、需要重选机动时该机制失效;其二,干净码重建精度不等于条件预测质量,二者必须分开评估(重建0.3 mm不代表部署误差也是0.3 mm);其三,实验全部在NAVSIM仿真内完成,对未见城市、恶劣天气、传感器失效的泛化未被验证;其四,有界修正仍可能越过道路边界——附录C.3给出实例:一次11.40 cm的修正使DAC从1变0,乘性门控把PDMS从98.7打到0.0,且全量配对评估中64个场景在NC/DAC/TTC/DDC至少一项发生回退。我自己的观察:第一,表示质量强依赖视觉分词器与VGGT预训练,单前视相机设定下侧后方来车、遮挡场景的骨架可信度存疑;第二,绝对PDMS并非SOTA(SimWAM(RL)@10步91.5),卖点是单步效率,且跨模型对比中各方法的预训练、输入设定并未完全对齐(作者也坦承这点并用公平对比实验补位);第三,残差RL用非反应式scorer做奖励,与navhard的反应式评估存在分布差异;第四,6维多项式模板修正族表达能力有限,对需要急促、非光滑修正的场景可能不够。

独立分析的弱点

第一个弱点是平均奖励掩盖个体风险:残差RL在平均PDMS上+0.6,但有64个场景在安全分量上回退,最严重者一次11.4 cm修正令PDMS从98.7归零。改进方向是把可行驶区域/碰撞约束做成限制器内的硬投影(对 $B q_\phi$ 施加安全集投影),或在奖励中对DAC类回归施加乘性惩罚而非线性罚项。第二个弱点是修正层级受限:actor只能动6个光滑系数、视觉token冻结,遇到需要换道的续接场景无能为力(Stage 2回退即此因)。改进方向是引入有界的scaffold修正(允许小幅调整视觉token),或设计机动级+度量级双层级actor,并配合反应式world-model critic做在线适配。第三个弱点是骨架的度量性来源单一:图像单目平移被明确当作非度量,尺度完全靠轨迹监督回补,一旦监督分布外的场景(异形路口、陡坡)出现,骨架本身可能失准。改进方向是融合LiDAR深度或多相机几何提供度量锚点,或引入车速先验做尺度校准。第四个弱点是工程门槛高:表示建立在约11亿参数的VGGT-Ω骨干和SimWAM的13万更新预训练之上,小团队难以从头复现。改进方向是研究残差表示对更小骨干的可迁移性,或只微调codec适配已有规划器的输出分布。

未来方向

作者提出的方向:把方法扩展到更广的视觉上下文(多相机、更长时间窗)和更具交互性的规划设定;训练时显式建模周围交通对自车行为的反应(reactive训练)以解决Stage 2回退;对视觉骨架提供更广泛的控制能力并引入显式安全约束,因为有界修正仍可能越界;在恶劣天气、传感器损坏下系统评估视觉骨架的鲁棒性。基于成果可自然延伸的方向:其一,把“骨架+残差”接口推广到机器人操作等具身任务——残差坐标天然适合接触丰富任务中毫米级精修;其二,残差RL的对偶高斯梯度估计器可替换为更样本高效的估计或离线RL,减少40.96万次奖励查询的开销;其三,探索自适应秩/自适应区间划分的残差基,或学习式(非解析模板)修正族以提升修正表达能力;其四,与一致性模型或蒸馏叠加,把“一步流+一次修正”进一步压缩延迟;其五,用世界模型生成的未来替代录制未来做目标构造,在无轨迹标注数据上自监督地学习动作目标,打通VITAL与无动作标注视频数据的结合。

复现评估

开源情况:论文给出了项目页(worldbench.github.io/vital)和GitHub仓库(worldbench/VITAL),附录A极为详尽——所有损失权重(位置1、增量0.25、关键帧0.75、航向0.1等)、残差秩扫描{2,…,30}、奖励乘子自适应规则 $\lambda \leftarrow \mathrm{clip}_{[0,2]}[\lambda + 0.02(\bar c - 0.001)]$、actor结构4528→256→256→6、限制器公式全部公开,文档质量很高。数据全部为公开集:nuPlan、nuScenes、OpenScene(注意nuScenes/OpenScene为CC BY-NC-SA许可,含非商业条款);评测用NAVSIM v1/v2(Apache 2.0);依赖VGGT-Ω(FAIR非商业研究许可,商用受限)、Depth Anything 3、LingBot-VA 2.0及SimWAM公开权重。算力需求:主模型训练为单机8×H200,最终阶段2.6万更新(不含继承的13万更新预训练);残差RL是123万参数小actor、1600次更新、40.96万次奖励查询(scorer可并行但耗时);推理在RTX 4090 batch 1下149 ms。综合评估复现难度为中高:完整复现需要多卡训练预算、SimWAM预训练权重与NAVSIM环境;但只复现诊断实验(重建分析、线性探针、公平动作学习对比)的门槛低得多,这些只需tokenizer拟合与轻量探针训练。

相似论文推荐