ROBOTICS · VLA · WORLD MODEL

Intercepting the Future

Latent-Space Predictive World Model for Dynamic VLA Manipulation ——「AHEAD」
Shahram Najam Syed*, Arthur Jakobsson*, Haoran Hao*, Jeffrey Ichnowski · Robotics Institute, Carnegie Mellon University(*Equal contribution)

VLA 模型在静态场景里泛化得不错,但物体一旦动起来就失效:它们把当前观测映射成动作,默认「观测到执行」之间场景静止不动,于是在任何非平凡速度下,延迟都超过了可用于抓取的时间窗。AHEAD 是一个 predict-then-act 外挂:给冻结的 VLA 接上一个运动感知的 latent world model,在 VLA 自己的 feature space 里把 patch tokens 向前预测 K 步,再把预测出的未来 tokens 喂给原封不动的 action decoder。只增加 4.9M 参数,不重训 7B OpenVLA。

arXiv 2606.02486(2026-06-01) CoRL 2026 preprint 模板 frozen OpenVLA 7B + 4.9M 外挂 📄 arXiv:2606.02486 PDF
VLAworld modeldynamic manipulationflow matchingRAFT optical flowadaptive horizonOpenVLA潜空间预测动态抓取predict-then-act

01 Motivation

论文开篇引了 Wayne Gretzky 那句话:“I skate to where the puck is going to be, not where it has been.” 这正是当前 VLA 缺的东西——它们只会滑向 puck 刚才在的地方。

“VLAs map the current observation to an action and assume the scene is stationary between observation and execution, so at any non-trivial object speed the resulting latency exceeds the time available to grasp.”

作者指出,之前有两条路线试图补这个缺口,但都没打到点上:

于是论文把问题重新表述为三个没被回答的问题:场景里哪些部分需要预测、要预测多远、以及在什么延迟下完成。AHEAD 的三条贡献正好对应——冻结 VLA 的 predict-then-act 外挂;沿空间与时间两个轴的自适应算力分配;以及显式的 kinematic conditioning(把 velocity/acceleration 解析式地在 rollout 步间传播,从而把 world model 从 constant-velocity 扩展到 acceleration regime,“without learning second-order physics from data”)。

AHEAD 在四个真实动态操作任务上的效果
Figure 1(hero). AHEAD 在四个真实动态操作任务上的表现。半透明叠影表示物体轨迹与机械臂的中间位姿,实心轮廓是最终操作姿态。上排:匀速传送带运输(moving object / moving box 两种变体);下排:抛体拦截(paddle hit)与抛体接取(catch)。
4.9M新增参数(外挂在冻结的 7B OpenVLA 上)
79–97%20 个动态仿真场景成功率(最强 baseline 31–58%)
19/30真机抛体接取(所有 baseline 均为 0/30)
≈158 ms端到端单步延迟(预算 Δtmax ≈ 200 ms)

Problem Statement 的两条 desiderata

论文把「有效的 predict-then-act loop」拆成两条必须满足的性质:

Scope(作者自己划的边界):本工作针对 dynamic scenes——物体在任务执行期间独立于机器人而运动,包括人为推动、传送带运输、接触后滚动。方法假设物体运动在预测 horizon 内可由 constant acceleration 良好近似(含 constant velocity 这个零加速度特例)。高度随机的 post-collision 动力学违反该假设,论文把它作为 stress test 评测,并明确声明报告的是 graceful degradation 而非 competitive performance。

02 Method

每个时刻 t,冻结的 VLA vision encoder φ 把 RGB 观测 ot 映射为 N 个 patch tokens vt ∈ ℝN×d。标准 VLA 直接把 vt 送进 action decoder,动作执行时它已经过期了。AHEAD 改为预测 v̂t+K(未来 K 步的 patch tokens),送给未改动的 action decoder:at = π(v̂t+K, eℓ)。

三个观察塑造了整条流水线:(1) 操作场景里大多数 patch 不动、不值得花算力,只有任务相关且在动的 patch 需要预测;(2) 合适的 horizon K 取决于场景——线性运动允许长 lookahead,chaotic 运动不允许;(3) 整个预测必须在 200 ms 量级的反应式操作延迟预算内完成。

AHEAD 架构图
Figure 2(architecture). 左侧 Preprocessing:三帧 → RAFT flow maps → 16×16 patch average pool → 有限差分得加速度 A(2D) = (v₂−v₁)/Δt;VLA Encoder 出 patch token(图中以 PCA overlay 显示),与运动描述子拼接后经 Language-guided Attention(配合指令 “Pick up the yellow duck”)得到 Masked Tokens(从 N 个中选 |S| 个)。中部 Dynamic World Model:conditional flow matching 的 ODE(dz/dτ = vθ(zτ, τ, [zt, Vk, A]))经 Euler ODE Solver 得 zt+k 及其 μ/σ,右侧 kinematic update Vk = V₀ + A·k·Δt 逐步更新条件,构成 Autoregressive Rollout。右侧 VLA re-injection:Decoder 出 predicted patch tokens t+K,经 Cross-Modal Projector 与 Llama 2 7B backbone 送入 Action head。火焰图标标记可训练组件(合计 ~4.9M 参数),雪花图标标记从 OpenVLA 冻结的组件。

① Language-and-Motion Saliency(空间上的算力分配)

RAFT 在三帧连续观测间计算 optical flow,经 patch-level pooling 与 finite-differencing 转成每 token 的 velocity V 与 acceleration A;把它们拼到 patch token 上得到 motion-enriched tokens ṽt ∈ ℝN×(d+4)。一个可学的 cross-attention 层从 ṽt 注意到指令 embedding eℓ,产生逐 token 的 language relevance 分数 M̃ ∈ [0,1]N;速度幅值超过标定阈值 τflow 的 token 会被抬到一个 relevance floor:

Mi = max( M̃i, αmotion · 𝟙[‖Vi‖ > τflow] )

选中集合 S = {i : Mi > 0.2},实验中典型只选出 N=196 个 token 里的 30–60 个。未选中的 patch 原样透传,不参与预测。

② Flow-Matching World Model + Kinematic Conditioning

world model 在 VLA feature space 而非 pixel space 里运作,理由是「对齐 action decoder 的输入分布,并避免昂贵的重建」。它用 conditional flow matching——论文的理由很实在:flow matching “produces high-quality samples in a few Euler steps against 100+ for diffusion”,从而把 world model 卡在每步延迟预算内。并行抽 S 个样本还免费同时给出均值预测和 sample-variance 形式的不确定度估计。

一个 4-layer transformer encoder 把选中的 motion-enriched tokens 压成紧凑 latent zt,dynamics model 自回归地向前滚:zt+k ~ p(zt+k | zt+k−1, Vk(S), A(S))。关键在于不让 world model 从数据里学完整物理,而是把 velocity 条件按 constant-acceleration kinematics 解析式地在 rollout 步间传播:

Vk = V0 + A · k · Δt

其中 V₀ 是 RAFT 给的初始速度、A 是有限差分得到的加速度。这一步捕捉了 gravitational pull、friction decay 这类加速度场景中的主要轨迹曲率。最后 MLP decoder 重建预测的 patch tokens,拼回完整 token grid(未选中的 patch 原样透传)后交给冻结的 action decoder。

③ Adaptive Horizon Halting(时间上的算力分配)

每步抽的 S 个轨迹样本提供了场景级不确定度估计 ūt+k(选中 token 上、样本间方差的平均)。rollout 在第一个满足 ūt+K > τu 的步 K 处停止,否则滚到 Kmax。于是线性、低不确定度的运动滚得久,chaotic 或 post-collision 的运动早停。实验中 Kmax = 10,τu 取训练集上 ū 的 90th percentile;实际 realized horizon 平均为 3–5 步。

④ 训练

两阶段:先在 manipulation video 语料上 pretrain,在 VLA feature space 里建立物体动力学的先验(论文的论证是——只要特征由 φ 抽出,任何含 manipulation 相关运动的视频都提供有用信号);语料涵盖六个公开来源,包括第一人称厨房操作(≈50K 序列)、hand-object interactions(174 actions,≈220K)、diverse robot manipulation(86 tasks,≈76K)、WidowX tabletop(≈60K)、仿真 Franka(≈100K)以及 OXE 里的 xArm 5/6/7。随后在 ≈200 条 in-domain xArm 7 轨迹上 fine-tune 以弥合 domain gap。dynamics model 用 conditional flow matching 目标;encoder/decoder 用 rollout horizon 上的 MSE 重建;feature alignment 层单独训练。

延迟拆解

在 NVIDIA H100 80GB、224×224 输入、bfloat16 下的单步端到端延迟(action head 用 OpenVLA-OFT 式 parallel decoding + action chunking):

ComponentLatency
RAFT optical flow (×2)≈ 20 ms
Motion pooling and concatenation≈ 2 ms
World model (S = 5 flow matching samples)≈ 40 ms
Decoder and feature alignment≈ 6 ms
VLA forward pass (parallel decoding)≈ 70 ms
Framework and I/O overhead≈ 20 ms
Total end-to-end≈ 158 ms

论文注明:各组件数值包含当前 PyTorch eager-mode 执行与 Python rollout 循环带来的框架开销,用 torch.compile、CUDA graphs 编译后还能进一步降低。

03 Experiments

仿真:自建 MuJoCo 环境、Franka Emika Panda 臂,20 个动态场景,覆盖四类运动——constant-velocity transport(conveyor / beam / pole push + red cup)、gravity-driven(rolling ball)、reactive contact(air hockey、ballistic catching)、chaotic post-collision(multi-deflection pinball、occlusion deflection、plinko drop)。真机:UFactory xArm 7 + 固定第三人称 RGB(224×224)+ 平行夹爪,五个任务。

Baselines(六个):Open-loop VLA、Retargeting VLA(closed-loop)、VLA + Fast Replan(OpenVLA-OFT,带 action chunking)、Realtime ACT、Streaming Diffusion Policy,以及 DreamVLA——一个基于 world model 的 VLA,先预测 dynamic regions 与 depth 再生成动作。协议:仿真每格为 5 个训练 seed × 每 seed 100 rollouts(共 500 rollouts)的 mean ± standard error;真机每任务 30 次尝试。

代表性仿真场景
代表性仿真场景(拼图). 左上 Air Hockey (AH);右上 Ballistic Catch (BC);左下 Occlusion (Occ)——物体需先穿过遮挡板;右下 Plinko Drop——球穿过钉阵下落,是全文动力学最混乱的 stress test 场景。每个场景都以随机化的物体位置、运动参数与光照程序化生成。

匀速与加速/减速两种 regime(Table 1)

任务成功率(%),Conv. = conveyor,Roll (g.) = 重力驱动滚球:

MethodConstant velocityAcceleration / deceleration
Conv.BeamPoleRollConv.BeamPoleRoll (g.)
Open-loop VLA0.0 ± 0.00.0 ± 0.00.0 ± 0.00.0 ± 0.00.0 ± 0.00.4 ± 0.60.0 ± 0.00.0 ± 0.0
Retargeting VLA13.7 ± 3.822.3 ± 4.013.3 ± 3.27.7 ± 2.91.0 ± 1.410.7 ± 3.11.0 ± 1.40.6 ± 1.1
VLA + Fast Replan24.0 ± 4.632.7 ± 4.222.3 ± 3.716.7 ± 4.022.3 ± 4.231.3 ± 4.621.7 ± 4.116.3 ± 3.7
Realtime ACT51.7 ± 5.160.3 ± 4.748.7 ± 4.937.0 ± 5.331.7 ± 4.641.0 ± 4.940.3 ± 4.931.0 ± 4.6
Streaming DP40.7 ± 5.250.3 ± 4.839.3 ± 4.528.7 ± 4.621.0 ± 4.129.7 ± 4.629.0 ± 4.519.7 ± 4.0
DreamVLA58.3 ± 4.747.7 ± 5.157.3 ± 5.046.0 ± 5.240.3 ± 4.938.7 ± 4.948.0 ± 5.030.7 ± 4.6
AHEAD (ours)97.3 ± 1.591.7 ± 2.895.0 ± 2.390.0 ± 3.187.7 ± 3.391.0 ± 3.993.7 ± 6.189.0 ± 0.9

差距在加速 regime 最大:最强 baseline(DreamVLA)落在 30–48%,AHEAD 保持在 87% 以上;且 AHEAD 跨 regime 的性能波动在 10 个百分点以内,而 baseline 大幅退化。作者检查失败模式后指出,没有显式预测的 baseline(Retargeting VLA、VLA + Fast Replan、Realtime ACT)最常见的失败是抓向物体刚才在的地方——夹爪合在运动物体身后的空气里;显式 kinematic update 让 AHEAD 转而瞄准预测的 intercept point。

对物体速度的鲁棒性(Table 2)

conveyor + red cup 场景下随传送带速度扫描的成功率(%):

Method0 cm/s10 cm/s20 cm/s30 cm/s40 cm/s
Open-loop VLA96.4 ± 1.90.0 ± 0.00.0 ± 0.00.0 ± 0.00.0 ± 0.0
Retargeting VLA96.8 ± 1.840.6 ± 4.931.2 ± 4.422.4 ± 4.114.6 ± 3.4
Realtime ACT97.2 ± 1.678.4 ± 3.964.8 ± 4.451.4 ± 4.641.0 ± 4.7
Streaming Diffusion Policy95.4 ± 2.131.6 ± 4.427.2 ± 4.022.4 ± 4.118.6 ± 3.7
DreamVLA96.8 ± 1.879.2 ± 3.870.4 ± 4.360.2 ± 4.447.2 ± 4.1
AHEAD (ours)97.6 ± 1.597.4 ± 1.797.0 ± 1.896.6 ± 2.095.4 ± 2.2

这张表里论文做了一个很干净的论证:静态(0 cm/s)时所有方法都在 95–98% 打平,说明感知与抓取能力本身是匹配的;差距只在引入独立物体运动后才拉开,因此可归因于 predictive lookahead,而非 perception 或 action decoder 的混淆差异。作者也如实交代了 40 cm/s 处 2.2 个点的残差来自 flow regime 高速端的 RAFT 误差,并说 motion-estimator ablation 支持这一解释。

复杂场景(Table 3)

八个复杂场景成功率(%):AH = Air Hockey,BC = Ballistic Catch,MO = Multi-Object,PiB = Place in Box,MB = Multi-Balls,Occ = Occlusion,Mid = Mid-flight deflection,L+MO = Language + Multi-Object。

MethodAHBCMOPiBMBOccMidL+MO
Open-loop VLA0.00.00.00.00.00.00.00.0
Retargeting VLA10.6 ± 3.00.00.00.00.00.00.00.0
VLA + Fast Replan0.00.031.6 ± 4.422.4 ± 4.10.00.00.00.0
Realtime ACT31.2 ± 4.422.4 ± 4.140.6 ± 4.941.2 ± 4.822.4 ± 4.111.4 ± 3.060.2 ± 4.431.6 ± 4.4
Streaming DP11.4 ± 3.041.6 ± 4.841.4 ± 4.622.4 ± 4.10.00.041.6 ± 4.830.4 ± 4.3
DreamVLA31.2 ± 4.449.1 ± 9.143.5 ± 4.541.2 ± 4.840.6 ± 4.90.049.4 ± 4.745.2 ± 4.8
AHEAD (ours)94.6 ± 2.387.4 ± 3.295.4 ± 2.295.8 ± 2.087.2 ± 3.379.4 ± 3.894.4 ± 2.495.6 ± 2.1

论文认为 Occ 那一列是「预测」区别于「反应」的最干净证据:除 Realtime ACT(11.4%)外,每个 baseline 在 Occ 上都是恰好 0%——因为接住一个已经消失在遮挡物后面的物体,需要预测它会从哪里重新出现,而只看最新一帧的 closed-loop policy 无解。AHEAD 达到 79.4%。

真机实验(Table 4)

UFactory xArm 7。左半:五个动态操作任务(30 次尝试);右半:传送带速度敏感性(每个速度点 10 次)。

MethodTasks (/30)Speed sensitivity, cm/s (/10)
Static obj.,
mov. box
Static box,
mov. obj.
Paddle
hit
Stop
ball
Catch
proj.
0510152025
Open-loop VLA0/300/300/300/300/3010/100/100/100/100/100/10
Retargeting VLA0/300/300/300/300/3010/104/100/100/100/100/10
VLA + Fast Replan0/300/300/300/300/30——————
Realtime ACT9/305/309/302/300/3010/108/100/100/100/100/10
Streaming Diffusion Policy3/303/300/300/300/3010/100/103/100/100/100/10
DreamVLA12/309/301/3012/300/3010/102/100/100/100/100/10
AHEAD (ours)30/3029/3023/3030/3019/3010/1010/1010/107/106/102/10

论文诚实地分析了 sim→real 的落差:平滑运动任务迁移得很紧(conveyor:97.3% sim、30/30 real),接触物理任务则退化(paddle hitting 94.6% sim → 23/30 real;projectile catching 87.4% sim → 19/30 real),归因于 MuJoCo 碰撞模型、夹爪柔顺性与相机快门行为的限制。11 次失败的抛体接取被逐条拆开:5 次是关节限位保护在几何上正确的 intercept 点前停住了手臂(说明预测目标原则上可达,但从当前构型不可达);剩下 6 次分别是末端执行器在球飞过后才刚好到位、以及球接触时从网中弹出。速度扫描也显示 AHEAD 是唯一在 5 cm/s 以上超过随机水平的方法——不过注意它自身在 25 cm/s 处也降到 2/10。

Ablations(Table 5)

各消融配置的平均成功率(%)。motion estimator / spatial masking / fixed horizon 在 constant-velocity 组的四个场景上评测,velocity model 在 acceleration/deceleration 组上评测。

Motion estimator

RAFT (ours)93.5 ± 1.8
CoTracker384.0 ± 2.7
Farnebäck68.9 ± 4.5
Frame differencing38.5 ± 3.4
No velocity input0.0 ± 0.0

Velocity model

Kinematic + accel (ours)90.7 ± 1.1
Kinematic only82.5 ± 3.6
Learned dynamics78.7 ± 7.9
Accel-token only78.5 ± 5.8
Constant Vk = V071.7 ± 10.4
Exponential decay65.3 ± 11.0

Spatial masking

Language + motion (ours)93.8 ± 1.2
Full prediction91.2 ± 2.7
Motion only84.0 ± 1.5
Language only77.9 ± 3.9
Uniform random (30%)46.4 ± 2.5

Fixed horizon K

K = 5 (= adaptive)93.8 ± 1.2
K = 892.9 ± 1.1
K = 1286.3 ± 3.0
K = 379.5 ± 1.8
K = 260.2 ± 1.9
K = 150.0 ± 4.0

几点值得注意(含对方法不那么有利的读数):No velocity input 直接归零,说明运动条件是这套方法的命门;Kinematic + accel 相比 Learned dynamics(78.7)高出 12 个点,支持「解析式传播优于让模型学二阶物理」的主张。但 Language + motion masking 相对 Full prediction 只高 2.6 个点(93.8 vs 91.2,且误差带重叠)——masking 的主要收益其实在算力而非精度上;而 adaptive horizon 与最好的固定 K = 5 打成完全平手(同为 93.8 ± 1.2),它的价值在于免去逐场景调 K,而非超越调好的固定值。

Stress tests:混沌动力学下的退化

论文额外在三个超出方法适用范围的场景上做了压力测试(按 time-to-exit 窗口参数化)。在最长的 1600 ms 反应窗下,AHEAD 取得 multiple deflection 76.4%、occlusion deflection 68.2%、Plinko drop 48.6%。Plinko 是全文最低的绝对性能,作者明确说这与「Plinko 的 post-collision 动力学本质上混沌、违反低阶运动学假设」一致——每次撞钉后的轨迹对接触位置与角度高度敏感,有限差分的加速度对下一次碰撞之后的步几乎不提供预测信号。即便如此,相对优势保持甚至放大:1600 ms 处 Plinko 上是最强 baseline 的 2.7×(48.6% vs 17.8%),multi-deflection 2.1×,occlusion deflection 2.5×。作者的措辞是:“The claim is graceful degradation under chaotic dynamics, not competitive absolute performance.”

04 Limitations

Note: 本文有明确的 “Limitations and Future Work” 小节,以下五条均为作者自述(stated),非推断。
① constant-acceleration 假设在混沌 post-collision 动力学下失效

“The constant-acceleration kinematic update breaks down under chaotic post-collision dynamics.” Plinko drop 在最长反应窗下降至 48.6%,正是低阶运动学假设失效的 regime。作者提出的自然扩展是更高阶的运动模型或学习式的 post-collision dynamics。

② RAFT 光流只捕捉 image-plane 运动,深度方向运动缺失

“RAFT-derived flow captures only image-plane motion, so scenes with significant depth motion would benefit from depth-augmented flow or multi-view inputs.” 即朝向/远离相机的运动分量拿不到,需要深度增强的 flow 或多视角输入。

③ 真机评测规模有限,跨平台迁移未验证

“The physical evaluation uses ≈ 200 in-lab xArm 7 trajectories, and transfer to substantially different platforms (bimanual systems, humanoids) is untested.”

④ 不确定度估计只覆盖 dynamics model,未涵盖运动估计本身

“The S trajectory samples capture dynamics-model uncertainty but share a single RAFT velocity estimate; propagating velocity uncertainty explicitly would improve calibration when motion estimation itself is unreliable.” 也就是说,当 RAFT 自身不可靠时(例如前面提到的高速端),adaptive halting 所依赖的不确定度信号是失准的。

⑤ 把底层 VLA 当作固定组件,未做联合优化

“AHEAD also treats the underlying VLA as fixed; a faster VLA backbone would directly benefit the predict-then-act wrapper, and we leave the joint optimization of backbone and world model to future work.” 从延迟拆解也能看出这点——VLA forward pass 的 ≈70 ms 占了 ≈158 ms 总预算的近一半。

一句话总结

论文自己给的定位是:真正的贡献不止于数字,而是“an architectural template for adding predictive lookahead to pretrained VLAs without modifying or retraining the underlying model”——一个 4.9M 参数的外挂,让冻结的 7B VLA 学会对着未来动作。