UniT:迈向人类到人形机器人策略学习与世界建模的统一物理语言 UniT: Toward a Unified Physical Language for Human-to-Humanoid Policy Learning and World Modeling
通过视觉锚定的三支路交叉重建,将异构人类与机器人动作投影到共享离散潜在空间,统一服务策略与世界模型
前置知识
视觉-语言-动作模型(VLA)
VLA 是把预训练视觉-语言大模型(VLM)作为感知与语义中枢,再在末端接一个动作生成头(典型如 flow matching 或扩散)输出机器人控制的整体架构。Qwen2.5-VL、GR00T、π0、Octo 都属于这一谱系。VLA 的训练目标通常是在 VLM 的 token 输出端做动作回归或离散动作 token 分类。
UniT 的策略版本 VLA-UniT 完全建立在 GR00T n1.5 + Qwen2.5-VL 的 VLA 骨架上,只是把动作回归的目标换成了 UniT token 预测加 flow matching 头;读懂它需要知道 VLA 如何把语义指令映射到连续动作。
离散潜在动作与 RQ-VAE
离散潜在动作是把一段连续动作序列编码成有限码本上的整数索引序列的表征方法。RQ-VAE(Residual Quantization VAE)通过多层残差量化逐级逼近连续特征,可以在同一码本中容纳粗粒度意图与细粒度动作细节,被 FAST、VQ-BeT 等行为克隆工作广泛使用。
UniT 的核心是用一个共享 RQ-VAE 码本同时量化视觉、动作、融合三支路的特征,从而把异构本体压到同一离散词汇表上;理解 RQ-VAE 才能读懂它怎么把人类腕部 21 自由度动作和机器人 50 自由度动作放在同一个 token 空间里。
Flow Matching(流匹配)
Flow matching 是一种生成式建模目标:在样本和噪声之间定义一条连续插值路径 $A^\tau = \tau A + (1-\tau)\epsilon$,让神经网络直接回归这条路径的速度场 $V_\theta(A^\tau | x, \tau)$,相比 DDPM 训练更稳定、采样步数更少,是当前主流 VLA 动作头(π0、GR00T)和视频世界模型(Cosmos)的事实标准。
UniT 在策略侧(VLA-UniT)和世界模型侧(WM-UniT)都依赖 flow matching 来生成动作和未来帧潜在;公式 (5) 和 (6) 直接复用了这一范式,理解它才能看明白 UniT 是如何与现有 VLA / 视频扩散栈对接的。
跨本体对齐(Cross-Embodiment Alignment)
不同机器人形态的关节数、控制模式、动作参数化差异巨大,直接在原始动作空间联合训练会让策略走捷径学到本体特异模式。跨本体对齐的目标是在表征层面找到与具体硬件解耦的共享意图空间,让人类的腕部动作和人形机器人的关节角被映射到同一表征。
这是 UniT 整篇论文的核心动机;文中用 t-SNE(图 7)展示 raw action 空间里人类和机器人样本完全分离、UniT 空间里高度重叠,正是跨本体对齐的视觉证据。
动作条件视频生成(Action-Conditioned World Model)
给定当前观测与动作信号,让视频生成模型自回归或一次性生成未来若干帧,用于机器人仿真、规划或数据增广。Cosmos Predict、IRASim、Ctrl-World 等是该方向代表。UniT 用统一 token 替换原始动作作为条件,让一个世界模型能同时建模人和人形机器人的动力学。
WM-UniT 直接基于 Cosmos Predict 2.5 改写条件信号;理解这个范式才能体会为什么把动作换成共享潜在特征对跨本体视频生成意义重大。
研究动机
人形机器人基础模型的扩展被两条互相纠缠的瓶颈死死卡住:一方面是高质量机器人示教数据极度稀缺,采集一条高质量人形机器人轨迹需要数倍于夹爪机械臂的成本;另一方面是人类第一视角(egocentric)视频数量庞大(EgoDex 就有 27,419 条 basic_pick_place 轨迹),却无法直接喂给机器人,因为人类手腕 21 自由度、手指动作与人形机器人 50 维关节空间在自由度、参数化、控制范式上完全不同。已有工作基本沿两条路走,但都有硬伤:retargeting 路线(EgoVLA、In-n-On)靠 IK 把人类手腕姿态映射到机器人关节,这种 case-by-case 的运动学求解难以规模化、且常常与原始视频对不上;而 latent action 路线又分两支,纯动作自编码器(VQ-BeT、FAST)只用本体感知信号做重建,跨本体时分布严重错位;纯视觉 latent action(Moto、LAPA、UniVLA)只从像素推断意图,容易缠入纹理、光照等低层外观因素,丢失精细动作细节。两者解耦的双编码器(METIS、XR-1)又没有强制对齐,仍然停留在两套词表。
本文的目标是本文提出 UniT(Unified Latent Action Tokenizer via Visual Anchoring),目标是在表征层面建立一种与具体硬件解耦的「统一物理语言」,让异构的人类视频和人形机器人示教投影到同一离散潜在 token 空间,从而可以同时支撑策略学习(VLA-UniT)和世界建模(WM-UniT)两条范式下的规模化人类知识蒸馏。具体而言,作者希望在 RoboCasa GR1 仿真和 IRON-R01-1.11 真实人形机器人上同时证明三点:数据效率提升、OOD 泛化增强、以及跨任务零样本迁移(stacking 这种机器人数据完全没有的任务)。
与已有工作不同的是,本文的独特切入角度是「以视觉为锚定」。作者的核心观察是:人和人形机器人的运动学结构虽然不同,但他们动作引起的视觉后果(物体被拿起、被放下、被推倒)是高度可类比的。因此 UniT 设计了视觉-动作-融合三支路结构,用共享 RQ-VAE 码本强制所有支路量化到同一离散空间,并用交叉重建(cross-reconstruction)双向约束——动作特征必须能重建视觉过渡、视觉特征必须能重建动作——这样既把动作锚定到物理后果(剔除不可视觉化的运动学噪声),又剥离外观混淆(保留与物理运动相关的视觉信息),最终得到的 token 只保留跨本体共享的「物理意图」。这套机制与传统 latent action 的根本区别在于:它不是把视觉或动作单独压成 token,而是显式让两个模态互相翻译,从而在 token 层和下游模型的中间表征层(图 7 的 t-SNE)都产生跨本体的对齐,这是之前所有 latent action 工作都没有显式做到的。
核心方法
UniT 的整体思路可以用一个直觉来理解:人和人形机器人虽然关节数量不同,但「握住杯子放下」这个意图在视觉层面是完全一致的——所以可以用视觉当锚,把所有本体的运动学压进同一个离散 token 词表。技术上,UniT 是一个由三支路编码器组成的潜在动作 tokenizer:视觉支路把 DINOv2 提取的相邻两帧特征 $(\phi(o_t), \phi(o_{t+k}))$ 通过反向动力学建模压缩成潜在动作 token;动作支路把 padding 到统一长度后的本体状态-动作块 $(s_t, a_{t:t+k})$ 通过本体专属 MLP 压成 token;融合支路把前两者的特征再融合一次。三路连续特征送入同一个 RQ-VAE 共享码本 $\mathcal{C}$ 量化,并通过视觉解码器 $D_v$(作为 forward dynamics model)和动作解码器 $D_a$ 实施双向交叉重建:每个 token 既要被解码出未来帧特征 $\hat{f}_{t+k}$(动作→视觉),又要被解码出原始动作块 $\hat{a}_{t:t+k}$(视觉→动作)。下游有两种用法:VLA-UniT 把融合支路的离散 token 作为 VLM 的预测目标,用一个轻量 flow matching 头再生成本体专属动作;WM-UniT 把动作支路的连续预量化特征作为视频世界模型的跨注意力条件,直接驱动跨本体视频生成。
UniT 的核心创新点是把「跨重建(cross-reconstruction)」从单纯的训练目标升级为一种跨本体对齐机制。已有 latent action 工作的重建都是单方向、单模态:纯动作自编码器只重建动作(导致跨本体分布错位),纯视觉 latent action 只重建未来帧(导致缠入外观),METIS / XR-1 虽然双编码但只在 KL 层面拉近距离、不强制两个模态互译。而 UniT 强制每个支路量化出的 token 同时承担「预测视觉后果」和「还原原始动作」两个任务,这等价于施加了一个跨模态互信息下界:能让视觉还原的 token 一定是和物理结果强相关的,能让动作还原的 token 一定保留精细运动细节,两者交集就是真正的物理意图。该机制还附带两个意料之外的好处:一是把噪声人类动作(比如动作捕捉中的标注抖动)变成视觉锚定的去噪过程(视觉无法被伪造的细节会反向纠正动作重建),二是迫使下游 VLM 的中间表征自然向跨本体对齐(图 7b、7c 的 t-SNE 重叠)。
方法步骤详情
UniT 的训练和部署可以拆成以下几步。第一步是输入预处理:每个样本是 $(o_t, o_{t+k}, s_t, a_{t:t+k})$ 的四元组,其中 $o$ 是原始像素、$a$ 是本体动作序列;人类和人形机器人动作先 padding 到统一最大长度(如人形 50 维),再分别通过本体专属 MLP 投影。第二步是三支路编码:视觉支路 $E_v$ 冻结 DINOv2 提取特征后用 transformer encoder + learnable queries 总结成 latent $z_v$;动作支路 $E_a$ 把投影后的状态-动作块编码成 $z_a$;融合支路 $E_m$ 把前两者特征拼起来再编码成 $z_m$。第三步是共享 RQ-VAE 量化:$\hat{z}_i = RQ(z_i; \mathcal{C}),\ i\in\{v,a,m\}$,三层残差量化同时精炼粗粒度意图与细粒度动作,所有支路共享码本 $\mathcal{C}$。第四步是交叉重建:每个 $\hat{z}_i$ 都通过共享视觉解码器 $D_v$(以当前帧特征 $f_t$ 为条件)生成 $\hat{f}_{t+k}^{(i)}$,并通过本体专属动作解码器 $D_a$(以当前状态 $s_t$ 为条件)生成 $\hat{a}_{t:t+k}^{(i)}$。第五步是损失聚合:$\mathcal{L} = \sum_{i\in\{v,a,m\}}[\lambda_v \mathcal{L}_{cos}(\hat{f}_{t+k}^{(i)}, f_{t+k}) + \lambda_a \mathcal{L}_{act}(\hat{a}_{t:t+k}^{(i)}, a_{t:t+k})] + \mathcal{L}_{RQ}$,对三支路同时监督视觉余弦相似和动作重建。VLA-UniT 在此基础上冻结 tokenizer,把 $c_t = \text{UniT}(\cdot)$ 当作 VLM 的监督标签(交叉熵预测 token),同时用 flow matching 头从 VLM 最后一层特征生成本体专属动作;WM-UniT 冻结 tokenizer,取动作支路连续特征 $\tilde{z}_a^t$ 经 MLP 投影后通过跨注意力注入 Cosmos Predict 2.5,flow matching 训练未来帧潜在。
技术新颖性
UniT 的技术新颖性主要体现在三个方面。其一是「视觉作为跨本体锚」的设定:之前 latent action 工作要么完全依赖本体感知,要么完全依赖像素,没有把视觉当作对齐枢纽;本文显式利用「物理后果跨本体一致」这一先验,将视觉从输入模态升格为对齐锚点。其二是「交叉重建式互信息下界」:用动作↔视觉的双向解码强制两个模态必须互相可预测,这是 KL 拉近(XR-1)和单向重建(Villa-X)都做不到的强约束;它在数学上等价于最小化 $I(z; \text{non-physical})$,最大化 $I(z; \text{physical intent})$。其实验证据是 $\sigma=0.2$ 的动作噪声下 UniT 仅退化 1.7×、动作 tokenizer 退化 2.7×、FAST 退化 10.7×。其三是「token 接口同时服务 VLA 和世界模型」:作者并没有把 UniT 当成策略专属的预处理,而是把它设计成双用途——融合支路 token 给 VLA 当预测目标、动作支路连续特征给视频世界模型当条件信号,让一个表征同时支撑两条范式,并在 t-SNE(图 7)上展示两种下游模型内部表征都被诱导到跨本体对齐,这是已有工作从未在同一框架下验证的。
实验结果
UniT 在四类实验中被系统验证。Q1 表征层面:图 7 的 t-SNE 显示 raw action 空间中人类(蓝)和人形机器人(黄)样本完全分离,而 UniT token 嵌入空间中两者高度重叠,VLA 视觉-语言隐藏状态和 WM 跨注意力输出都从「分离」变成「交织」。图 8 的去噪实验进一步证明,$\sigma=0.2$ 的高斯噪声注入后 FAST 退化 10.7×、动作 tokenizer 退化 2.7×、UniT 仅退化 1.7×,相对误差曲线在所有 $\sigma$ 下 UniT 都最低,说明视觉锚定确实提供了去噪正则化。Q2 策略层面:图 9 全量 RoboCasa GR1 24 个任务上 VLA-UniT 整体成功率 66.7%,其中 Pick & Place 67.3%、Articulated 64.7%,相对此前最优 FLARE 55.0% 提升 11.7 个百分点,相对同架构但无 UniT 的 GR00T-Qwen2.5 基线 47.8% 提升 18.9 个百分点。图 10 左展示数据效率:仅用 10% 数据(每任务 100 条)时 VLA-UniT 已达 45.5%,逼近全量 GR00T 的 47.8%;右图显示 few-shot + EgoDex 人类数据共训练后,in-domain 从 45.5% 升到 50.0%(Pick & Place 41.7%→49.4%),OOD 平均 34.7%→38.5%(Unseen Appearance 37.0%→42.7%、Unseen Combinations 40.7%→43.0%、Unseen Object Types 29.0%→33.0%)。真实 IRON-R01-1.11 人形机器人上(图 11),Pick & Place 70%→78%、Pouring 35%→75%;五项 OOD 维度均显著提升:Geometry 23.3%→63.3%、Distractor 26.7%→60.0%、Background 35.0%→65.0%、Target 33.3%→73.3%、Combinational 10%→70%。图 12 展示关键的零样本 stacking 迁移:GR00T 基线和 VLA-UniT w/o Cross-Recon 都是 0%、无人类数据 VLA-UniT 10%、加人类数据后达 60%,并涌现出腰部旋转和头部调整视角等上半身协调模式。Q3 世界模型层面:表 1 的 DROID 单本体可控生成上 WM-UniT 在 PSNR/SSIM/LPIPS/EPE 上均优于 Raw Action 和 WM-Action;EgoDex+RoboCasa 共训练时 WM-UniT 在 EgoDex 上 PSNR 28.06 vs Raw 24.84,RoboCasa-GR1 上 17.66 vs 13.45,FVD 166.50 vs 237.13。表 2 显示人类数据预训练后再微调 humanoid,相比无预训练 PSNR 从 16.34 提到 18.06、EPE 0.478→0.446。表 3 的 Gemini-3-Pro 评测显示跨本体条件一致性上 WM-UniT 在 robot→human 方向 3.84 vs 2.92、human→robot 3.27 vs 2.95,语义/时序/几何三个维度全面领先。Q4 设计消融(图 15):OOD 平均 VLA-UniT 49.9% > VLA-Vision 45.2% > VLA-Action 42.1% > VLA-UniT w/o Cross-Recon 30.3%;In-domain Pick & Place VLA-UniT 66.8% > VLA-Villa 63.1%(单向 V2A 重建),说明双向交叉重建和双模态同时使用都是关键。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| RoboCasa GR1 Tabletop Overall (24 tasks) | Success Rate (%) | 66.7 | FLARE 55.0 / GR00T-Qwen2.5 47.8 | +11.7 vs FLARE; +18.9 vs GR00T |
| RoboCasa GR1 Pick & Place (n=900) | Success Rate (%) | 67.3 | FLARE ~55; GR00T-Qwen2.5 ~47 | +12.3 vs FLARE |
| RoboCasa GR1 Articulated (n=300) | Success Rate (%) | 64.7 | FLARE 47.6; GR00T-Qwen2.5 ~47 | +17.1 vs FLARE |
| RoboCasa GR1 Few-Shot (10% data, n=1200) | Success Rate (%) | 45.5 (45.5+human → 50.0) | GR00T-Qwen2.5 ~21.3 (full) / similar few-shot gap | 约 10× 数据效率(10% 数据 ≈ GR00T 全量水平) |
| Real IRON-R01-1.11 Pick & Place | Success Rate (%) | 78 (with human), 70 (robot-only) | GR00T-Qwen2.5 30 | +48 (with human) vs GR00T |
| Real IRON-R01-1.11 Pouring (Bimanual) | Success Rate (%) | 75 (with human), 35 (robot-only) | GR00T-Qwen2.5 5 | +70 (with human) vs GR00T |
| Real-world OOD Geometry Generalization | Success Rate (%) | 63.3 | GR00T-Qwen2.5 23.3 | +40 |
| Real-world OOD Distractor Generalization | Success Rate (%) | 60.0 | GR00T-Qwen2.5 26.7 | +33.3 |
| Real-world OOD Combinational (instruction disambiguation) | Success Rate (%) | 70 | GR00T-Qwen2.5 10 | +60 |
| Real-world Zero-Shot Stacking (unseen task) | Success Rate (%) | 60 (with human), 10 (no human) | GR00T / VLA-UniT w/o Cross-Recon 0 | +60 vs both baselines |
| DROID World Modeling Controllability | EPE ↓ | 0.2588 | Raw Action 0.2662; WM-Action 0.2593 | -0.0074 EPE vs Raw Action |
| EgoDex+RoboCasa World Modeling (human subset) | PSNR ↑ / FVD ↓ | 28.06 / 130.87 | Raw Action 24.84 / 171.37 | +3.22 PSNR, -40.5 FVD |
| Cross-Embodiment Conditioning (Robot→Human, Overall) | Gemini-3-Pro 1-5 score | 3.84 | Raw Action 2.92 | +0.92 |
| Tokenizer Noise Robustness (σ=0.2) | MSE_noisy / MSE_clean ↓ | 1.7× | FAST 10.7×; Action Tokenizer 2.7× | 6.3× 比 FAST 更鲁棒,1.6× 比 Action Tokenizer 更鲁棒 |
局限与改进
作者在文中相对克制地讨论局限,但综合上下文可以归纳出几点。第一个限制是当前 UniT 仅在 Pick & Place 类短程任务上验证,Pouring 也只是双机械臂配合倒水的简单动作;论文承认「扩展到灵巧上半身操控」是未来方向,但并未在论文中给出真正涉及多指关节复杂操作的实验,这意味着其声称的「dexterous humanoid」扩展性目前还是纸面上的。第二是 EgoDex 与 RoboCasa 共训练时,作者明确指出需要在共训练后「再 fine-tune 纯机器人数据」才能发挥效果(Fig. 10 right 显示 co-train → fine-tune 是标准流程),这暗示 UniT 的 token 空间虽统一但人类数据仍可能带入本体特异的分布偏差,需要一个显式校准步骤。第三是真实机器人实验的样本量较小(每任务 120 条),OOD 每场景只评估 30 次,统计置信区间较宽,作者并未报告置信区间或显著性检验。第四是 VLA-UniT 严重依赖预训练好的 VLA 骨干(GR00T n1.5 + Qwen2.5-VL / Qwen3-VL),其训练算力和数据量没有详细披露,论文主框架更像是「统一 token 接口」的插件式方案而非端到端的大模型;真实工业场景的扩展成本需要外部验证。最后,论文中世界模型的可控性指标 EPE 用 optical flow 估计,缺少人手或机器人末端真值跟踪,其精度本身有误差传递问题;且 Gemini-3-Pro 自动评分虽然在论文中被作为定量指标使用,本质上仍是 MLLM 主观打分,与人类评分的一致性需要单独验证。
独立分析的弱点
独立审视论文,至少有几个明显弱点。第一是 ablation 过于干净,缺一个明显的对照组:VLA-Action 在 OOD 上 42.1% 反而比 VLA-Vision 的 45.2% 低,看起来「视觉」比「动作」更重要,但作者把这点解释为动作分布错位,缺乏一个「视觉+动作但只用其中一路做 token」的对照,无法干净拆解「视觉带来的增益」和「动作带来的增益」分别多大。改进方向是在 tokenizer 训练时就分两路监督,再冻结其中一支用于 VLA。第二是 Zero-shot stacking 从 0% 跳到 60% 的飞跃完全依赖人类数据,但作者并没有跨数据集验证(只用 EgoDex basic_pick_place),如果换成不同来源的人类数据(如 Epic-Kitchens、Something-Something)会不会同样有效未验证。改进方向是补充跨人类数据集的消融。第三是 WM-UniT 的世界模型 backbone 是 Cosmos Predict 2.5 这种已经极强的视频基础模型,在 DROID 这种单本体场景下提升幅度很小(PSNR 21.02→21.32、EPE 0.2662→0.2588),换到更弱的基线才更能说明 UniT 条件的价值。改进方向是在自训练的 baseline 上做对比。第四是 cross-reconstruction 损失中超参数 $\lambda_v, \lambda_a$ 的敏感度、码本大小、视觉-动作编码器容量比例等关键超参都没有系统消融,而这些直接决定 UniT 是否真的「视觉锚定」而非「动作主导」。改进方向是补一个 hyperparameter sensitivity 实验。
未来方向
作者明确点出了三个未来方向。第一是利用 UniT 视觉支路只需要观测对、不需要配对动作标注的特性,去吸收海量的无标注互联网视频(YouTube、Epic-Kitchens),把「物理先验」扩展到人类所有日常任务,而不仅限于 EgoDex 这一类。第二是把 UniT 的统一 token 空间变成策略与世界模型的闭环:策略提出 latent action,世界模型基于此 roll out 出未来视觉帧,再把想象的 rollout 当作 RL reward 或 test-time planning 的搜索目标,作者称之为「shared token space 内自演化」。第三是把 UniT 推广到上半身协调和灵巧操控(dexterous manipulation),从 EgoDex basic_pick_place 这种短程任务扩展到多阶段、长视野、双手灵巧操作。基于结果可延伸的方向还包括:把 UniT 接到 RL post-training 上做 latent-action PPO;把 UniT 作为不同 VLA 之间的协议层,让 GR00T、π0、Octo 共享同一 token 词表做迁移学习;以及利用 UniT 的去噪特性($\sigma=0.2$ 仍 1.7×)反向清洗大规模人类动作数据集。
复现评估
从可复现性角度,UniT 处于「方法清楚但工业级细节较厚」的水平。代码方面,作者给出了项目页 https://xpeng-robotics.github.io/unit/ 并承诺开源,但截至论文 v1 没有附带 GitHub 链接或具体仓库;Tokenizer 训练、VLA-UniT 训练、WM-UniT 训练三部分均依赖其内部数据规模(proprietary 32k 机器人轨迹 + 30k EgoDex),其中 32k 是 XPENG 自有数据,外部研究者无法访问;RoboCasa GR1 的 24k 全量轨迹是公开可获取的,可以复用仿真评估。训练算力方面,论文没有披露具体 GPU 类型和总训练小时数,但 GR00T-Qwen2.5 基线本身就需要数百 GPU 小时,UniT 在此基础上还要先训练一个 tri-branch tokenizer 再做 VLA 共训练,叠加后算力门槛很高。难度方面,对一家具备 GR00T / Cosmos 复现经验的研究组来说,UniT 的复现中等偏难:架构清晰、损失公式简洁(公式 2-6)、消融完善,主要难点是 DINOv2 冻结 + Qwen-VL 全参微调 + RQ-VAE 三路同训的工程稳定性,以及在 DROID / EgoDex / RoboCasa 三套数据 pipeline 之间切换的数据治理。对于普通研究组,可以优先复现 tokenizer 的 t-SNE 对齐(图 7a)和去噪实验(图 8)这两个核心论据,这两个只需要 EgoDex 公开数据和相对普通的算力。
论文图表
对比四种 latent action 架构:(a) Action-Only 只重建动作 $Z_a$、无视觉锚;(b) Vision-Only 只从像素推断 $Z_v$、缠入外观;(c) Decoupled 用独立编码器得到 $Z_v, Z_a$ 但不对齐;(d) UniT 通过 cross-reconstruction 把三支路压到统一 $Z_{unified}$,并标出 Misalignment 警告。
这张图把论文的核心对比和动机一次性铺开,是 method 部分必须展示的「问题定义图」。
五种真实 OOD 场景的指令与示例图:(a) Geometry 用不同形状杯子替代 bowl;(b) Target 给出新的放置位置(basket/plate);(c) Distractor 加入干扰物;(d) Visual 改变桌面纹理;(e) Combinational 多个 seen object 通过指令消歧。
真实 OOD 评估的任务分布图,直接对应图 11 右的实验结果。