机器人 · Robotics · arXiv 2506.22242

4D-VLA

Spatiotemporal Vision-Language-Action Pretraining with Cross-Scene Calibration
Jiahui Zhang, Yurui Chen, Yueming Xu, Ze Huang, Yanpeng Zhou, Yu-Jie Yuan, Xinyue Cai, Guowei Huang, Xingyue Quan, Hang Xu, Li Zhang† · 复旦大学数据科学学院 & 华为诺亚方舟实验室

4D-VLA 将 RGB-D 深度信息与历史时序帧融入 VLA 预训练,通过空间对齐 token 和记忆库采样(Memory Bank Sampling)解决多机器人数据集联合训练时出现的"坐标系混乱"与"状态混乱"问题,在 LIBERO 仿真平台和真实操作实验中均大幅超越 OpenVLA 基线。

2025 · 复旦大学 LIBERO Avg +12.1pp vs OpenVLA 真实场景 Avg 85.63% vs 27.70% 📄 arXiv:2506.22242 GitHub 代码
VLA 机器人预训练 RGB-D 4D表征 spatiotemporal reasoning memory bank sampling 多视角泛化 coordinate alignment imitation learning MV-Bench

01 动机

大规模机器人预训练的关键障碍:当把多个异构机器人数据集混合训练时,单帧 RGB 图像缺乏充足的空间与时序上下文,导致 action 分布极度分散,模型难以收敛。

坐标系混乱(Coordinate System Chaos)

动作定义在机器人坐标系中,而视觉输入缺乏足够的空间上下文。论文指出:"if the image does not fully capture the robot's body, it becomes challenging to infer the robot's exact position and orientation." 不同相机外参与内参使得同一动作在不同数据集中的视觉表征截然不同。

状态混乱(State Chaos)

单帧图像缺少必要的时序与上下文线索以消除动作歧义。论文指出这包括"symmetric trajectories——where it is difficult to infer the direction of motion",以及"visually similar observations correspond to entirely different actions"的情形,使得模型对当前运动方向无法判断。

"We identify two primary factors contributing to incomplete input: coordinate system chaos and state chaos, both of which severely limit the training efficiency achievable with diverse robotic datasets."
teaser
图 1:左侧展示坐标系混乱问题——不同数据集的相机视角与机器人摆放导致同一动作在视觉上呈现截然不同的分布;右侧对比 OpenVLA 与 4D-VLA 的整体性能,后者在 LIBERO 和真实场景均取得显著提升。
88.6%LIBERO Avg(±0.3)
+12.1ppvs OpenVLA (76.5%)
85.63%真实操作 Full Model
81.0%MV-Bench In-View Avg

02 方法

4D-VLA 以顺序 RGB-D 输入为基础,构建两类核心机制:① 空间感知视觉 token(Spatial Vision Token)将深度信息提升至世界坐标,实现跨场景坐标系对齐;② 记忆库采样(Memory Bank Sampling, MBS)从时序窗口中自适应挑选 k=5 帧历史信息,既捕捉关键状态变化,又避免冗余。

pipeline
图 2:4D-VLA 整体 pipeline。视觉编码器提取 RGB 特征图;深度图经由相机内外参反投影至世界坐标,通过可学习位置嵌入 ℰₛ 编码后与视觉特征做逐元素加法,形成 spatial vision token eˢᵀ;记忆库从 n=20 帧的时序窗口中选取 k=5 帧历史 token;拼接文本 token 后输入 LLM 骨干,最终接 MLP action head 输出预测动作。

空间感知视觉 Token(Spatial Vision Token)

给定 RGB 图像 I ∈ ℝ³ˣʰˣʷ,视觉编码器 ℰ 提取特征图 fᵥ = ℰ(I)。深度图 D 通过相机外参 [R|T] 和内参 K 反投影得到世界坐标点云 Pₘ,再经可学习位置嵌入 ℰₛ 编码并与视觉特征做逐元素加法:

eˢᵀ = P(ℰ(I) + ℰₛ(Pₘ))

此设计使 token 同时携带外观与精确三维位置信息,消除坐标系混乱。

记忆库采样(Memory Bank Sampling, MBS)

从长度为 n=20 的历史帧序列中自适应选取 k=5 帧:算法顺序遍历帧组,维护一个相似度队列,保证每帧与已选帧的相似度低于当前队列最大值,从而避免冗余、保留关键状态转变。历史 token 与时序偏移通过可学习编码 ℰₜ(t-j) 注入相对位置信息,最终与当前帧合并:

𝒳 = ⋃ᵢ∈ℋ [eᵢᵀ | eᵢˢᵀ] ∪ {eᵗᵉˣᵗ}

Action head 采用 MLP,在延迟(160.0 ms)与成功率(86.6%)之间取得最优平衡,相比 Autoregressive(0.6 FPS)和 Diffusion head 均有优势。

MV-Bench 多视角评测基准

论文提出 MV-Bench——一个覆盖 270° 前向视角范围内 6 个训练视角的仿真评测集,同时评估 In-View(已见视角)与 Cross-View(偏移 Δ15°/Δ30°)的泛化能力,填补了现有 VLA 多视角评测的空白。

MV-Bench
图 3:MV-Bench 6 个训练相机视角的设置(覆盖 0°、60°、120°、270°、300°、330°),用于评估 VLA 的空间感知与新视角泛化能力。

03 实验

评测平台:LIBERO 仿真基准(4 个子任务)、MV-Bench 多视角仿真评测、真实 Franka 机器臂操作(4 项任务 × 多视角泛化)。基线方法:OpenVLA。所有数字均源自论文原文。

Table 1 · LIBERO 仿真基准

方法LIBERO-SpatialLIBERO-ObjectLIBERO-GoalLIBERO-LongAvg
OpenVLA84.7±0.988.4±0.879.2±1.053.7±1.376.5±0.6
4D-VLA(Ours)88.9±0.595.2±0.390.9±0.479.1±1.288.6±0.3

Table 2 · MV-Bench 多视角仿真

方法 / 视角类型60°120°270°300°330°Avg
OpenVLA In-View57.450.050.643.553.857.852.2
4D-VLA In-View83.287.079.570.275.890.281.0
OpenVLA Cross-View (Δ15°)50.5
4D-VLA Cross-View (Δ15°)73.8

Table 3 · 真实机器臂操作

配置Task 1Task 2Task 3Task 4Avg
OpenVLA45.0022.5030.0013.3327.70
Base VLA35.0020.005.002.6715.67
+Pretraining60.0060.0040.0028.0047.00
+Pretraining+Coord75.0060.0085.0034.6763.67
+Pretraining+Hist80.0077.5070.0036.0065.88
Full Model(Ours)90.0082.5090.0080.0085.63
results
图 5:真实场景多视角泛化测试——标准视角(In-View)与新视角(Δ15°、Δ25°、Δ45° 偏移)下的成功率对比。4D-VLA 在所有偏移角度均显著优于 OpenVLA(平均 55% vs 18%),体现出空间感知预训练带来的跨视角泛化能力。

消融实验(Ablations)

Table 5 对时序编码方式进行系统消融:可学习相对位置编码 + Concat 融合达到最高成功率 75.6%;使用绝对位置编码则完全失效(0.0%);去掉所有历史编码降至 63.0%。

Table 8 对帧采样方法进行对比(LIBERO-Spatial):MBS(0.866)优于单帧(0.738)、Adaptive Pooling(0.604)、Grid Pooling(0.620)和 Q-Former(0.556),同时延迟 160.0 ms 与显存占用 8682.9 MB 均处于合理水平。

ablation
图 6:历史帧窗口大小与帧采样效率的对比分析。随时序窗口 n 增大,成功率先升后趋于稳定;MBS 在 n=20、k=5 时取得最佳权衡。

04 局限性

说明:以下第 1 条为论文作者在正文中明确陈述的局限;第 2、3 条为基于方法设计推断(inferred)的潜在局限,未在论文中直接点明。
依赖 RGB-D 硬件(stated)

论文明确指出:"A limitation of our approach is its reliance on RGB-D input, which introduces hardware restriction." 相比纯 RGB 方法,4D-VLA 需要深度传感器,限制了在无深度摄像头场景下的直接部署。

计算与延迟开销(inferred)

Memory Bank Sampling 引入 k=5 历史帧的 spatial token,相比单帧推理延迟从 76.5 ms 增加至 160.0 ms(约 2.1×)。在对实时性要求极高的场景(如高速操作)下可能存在瓶颈。

相机标定精度依赖(inferred)

空间感知 token 的构造依赖精确的相机内参 K 和外参 [R|T] 将深度图反投影至世界坐标。若标定存在误差或外参在运行中漂移,空间对齐效果会受损,可能影响在标定精度受限场景下的稳定性。