ICLR 2025 · arXiv:2412.10345 · 2D 轨迹视觉提示增强 VLA 时空意识
TraceVLA 用 CoTracker 从历史图像中跟踪活跃点,把末端/物体历史运动轨迹画回 RGB 图像,再和原图一起输入 OpenVLA。它证明:把历史压缩成 2D visual trace,比直接拼接多帧更适合 VLA 使用。
| 设计 | 细节 |
|---|---|
| 点跟踪 | CoTracker;起始帧划分 K×K 网格,K=40。 |
| 活跃点选择 | 累计相邻帧位移,超过阈值视为 active point。 |
| 轨迹采样 | 随机采样 M=5 条 active trajectories。 |
| 输入格式 | 原图 + visual trace 图,中间插入 learnable separator token。 |
| trace dropout | 训练时以概率 α 用原图替换 trace 图,提升 CoTracker 失败时的鲁棒性。 |
tracks = cotracker(history_frames, grid=40)
motion = sum_norm(tracks[:,1:] - tracks[:,:-1])
active = tracks[motion > threshold]
trace = sample(active, M=5)
trace_img = overlay_polyline(current_img, trace)
action = traceVLA(images=[current_img, trace_img], text=trace_prompt + instruction)| Benchmark/平台 | 内容 | 目的 |
|---|---|---|
| SimplerEnv Google Robot | 131/137 多环境配置,含 visual matching 与 variant aggregation | 验证环境变体、相机、光照、背景、干扰物下的鲁棒性 |
| WidowX-250 真机 | 固定第三视角 256×256 RGB;8 个真实任务;每任务 30 demos finetune | 验证真实机器人和 unseen tasks |
| LIBERO | Spatial/Object/Goal/Long 四 suite,多任务混合训练 | 验证空间关系、物体、目标和长程任务收益 |
| 方法 | Spatial | Object | Goal | Long | Avg |
|---|---|---|---|---|---|
| OpenVLA finetuned | 82.6±0.4 | 83.8±0.6 | 70.4±0.5 | 45.7±0.6 | 70.6±0.4 |
| TraceVLA finetuned | 84.6±0.2 | 85.2±0.4 | 75.1±0.3 | 54.1±1.0 | 74.8±0.4 |