几何动作模型:把几何基础模型改造为统一的机器人感知—预测—动作骨干 Geometric Action Model for Robot Policy Learning
把3D几何基础模型拆分后插入因果时序预测器,单次前向同时输出未来几何与动作
前置知识
几何基础模型(GFM)
如 DA3-Giant、VGGT 这类前馈 Transformer,接受一张或多张 RGB 图像 $\{I_v\}_{v=1}^V$,按 patch 划分后逐视图构建 token 序列(含一个 camera token 加 $P$ 个 patch token),通过 $M$ 个 Transformer block(含 frame-wise 和 global 两种 attention 模式)隐式建立跨视图关系,最终用 DPT 头输出逐像素深度、3D 点图以及相机内外参,本质上是一次性端到端的 3D 重建网络。
本文的所有做法都建立在 GFM 已经训练好、可以直接拿出来用的 3D 先验之上;如果不了解 GFM 的 token 结构和多 view attention 模式,就看不懂为什么要在 split layer 处插入因果预测器。
Vision-Language-Action 模型(VLA)
把预训练视觉-语言模型改造成机器人策略:用文本指令和图像观测作为输入,自回归地或并行地解码机器人动作 token(关节或末端位姿增量),代表工作包括 OpenVLA、$\pi_0$、$\pi_0.5$ 等;动作通常以 action chunk 的形式一次预测未来 $C$ 步再开环执行。
VLA 是本文最重要的对照基线,理解 VLA 才能明白本文为何强调 GFM 共享骨干对 2D-only 视觉先验的超越。
因果(Block-Causal)Transformer 预测器
在自回归训练时通过 block-causal attention mask 保证每个时间步的 token 只能 attend 到当前及历史 token(包括语言 token、本体感知 token、动作历史 token 和当前观测 token),从而在没有未来信息泄漏的条件下预测未来帧的 latent 表示,本质上就是把 LLM 中的 next-token prediction 思路移植到 3D latent 空间。
这是 GAM 的核心模块:它让 GFM 获得「以语言和动作为条件的时序世界建模」能力,同时不需要重训整个 backbone。
动作块(Action Chunk)与本体感知(Proprioception)
动作块指策略一次预测未来 $C$ 步动作(如 $C=8$)并以开环方式全部执行再重新观察;本体感知 $s_t \in \mathbb{R}^{d_s}$ 描述机器人当前关节配置与末端位姿(本文 $d_s=7$),通常被作为条件 token 喂回策略;动作空间为末端增量位姿 + 夹爪,维度 $d_a=7$。
理解 GAM 的输入输出维度($H$ 个历史观测、$C$ 步动作块、$d_g=1024$ 宽度的 predictor token)才能在脑中把整张架构图跑通。
研究动机
通用机器人策略需要同时理解语言指令、视觉外观、场景几何、机器人自身状态和物理动力学,但当前两大主流路线各有明显短板。Vision-Language-Action(VLA)模型以 2D 图像为先验,在面对「深度/尺度/遮挡」这类空间歧义时必须靠 action decoder 自己反推 3D 结构,在 LIBERO-Plus 这类 OOD 评测上显著掉点——例如 OpenVLA-OFT(7B)在 LIBERO 上有 97.1% 的成功率,到了 LIBERO-Plus 整体跌至 69.6%($\downarrow$27.5),相机扰动子项只有 56.4%;$\pi_0$-FAST、RIPT-VLA、UniVLA、NORA 等也在不同维度上跌 20–60 个百分点。Video World Action Model(WAM)例如 Cosmos-Policy 借助视频扩散把未来帧预测和动作联合训练,虽然原始 LIBERO 能冲到 98.5%,但视频骨干只在 2D 像素空间学习,对几何推理帮助有限,且需要多步去采样,单次推理高达 382.4 ms。已有的几何感知 VLA(如 $\pi_0.5$ + Spatial Forcing、$\pi_0.5$ + ROCKET)只是把 GFM 当成静态的『蒸馏特征来源』贴在一个 VLA 外壳上——GFM 的多层几何结构没有真正被用来时序化预测,这就是 GAM 要填补的空白。
本文的目标是论文要构建一个真正『以几何为基础』的通用操作策略:用预训练好的 GFM(比如 DA3-Giant,作为唯一共享骨干)让模型在保留几何先验的前提下同时承担三件事——感知当前多视角观测、预测未来时刻的 3D 几何 latent、解码出机器人动作 chunk,最终在仿真和真机两个维度上同时证明 GAM 在『精度、鲁棒性、推理速度、模型大小』四象限均优于 VLA、WAM、geometry-aware VLA 三大主流路径。具体目标包括:在 LIBERO 4 个 suite(500 次 rollout)上保持 ≥97% 的高成功率;在 LIBERO-Plus 7 类扰动维度上整体成功率 ≥85%、相机扰动场景 ≥80%;在 RoboCasa-Kitchen 24 个任务上超越 67.1% 的当前 SOTA;在 4 个真机任务上同时在 ID 与 OOD 相机扰动下保持领先;推理延迟压到 10 ms 以内、单步参数量控制在 1.5B 以内,从而让 GFM 驱动的策略既能在 GPU 服务器上做大规模离线训练,也能在真机控制回路里以 ≥100 Hz 闭环运行。
与已有工作不同的是,既有 VLA(OpenVLA、π0、π0.5、π0-FAST、UniVLA、NORA 等)以 2D 视觉-语言先验为核心,在 LIBERO 上能拿到 87–98% 的成功率,但一旦换成 LIBERO-Plus 的相机/光照/背景扰动就平均掉 30–60 个百分点;既有 WAM(Cosmos-Policy、Fast-WAM、WorldVLA)借助视频扩散把未来帧预测与动作联合训练,几何推理能力受限于 2D 视频骨干,且需要数十步去采样,单次推理 ≥380 ms;既有 geometry-aware VLA(Spatial Forcing、ROCKET)虽然在 VLA 上挂了 GFM 蒸馏,但只把它当作静态特征源,没有把 GFM 的多层结构用于时序预测和动作解码。三类方法都没有让 GFM 自身参与未来状态建模;GAM 的独特切入角度是:把 GFM 在某一中间层($L_s$)物理上『劈开』,浅层当编码器、深层当解码器,在中间插入一个轻量因果预测器,让未来几何 latent 和未来动作 token 共享同一段自回归序列、同一次 backbone 前向,由此把『世界建模』能力直接嵌入几何表征空间,而不是外挂一个语义或视频模型——这从根本上把『在哪里』和『怎么动』在特征层面耦合起来。
核心方法
直觉上,GAM 把一个已经会做 3D 重建的 Transformer(DA3-Giant,40 个 block)当作『乐高积木』,在第 12 层用一把『虚拟剪刀』把它裁成两段。前 13 个 block(block 0–12)被当作 observation encoder,只负责把当前窗口 $H$ 个时间步的多视角 RGB 转成几何 latent;最后 28 个 block(block 13–39)被当作 decoder,搭配一个 8M 参数的 action head 和原来就有的 DPT depth head,既解动作也解未来深度。技术路线上:在裁开处再插入一个 12 层、宽度 $d_g=1024$ 的 causal future predictor,它把语言 token、本体感知 token、动作历史 token 和几何 latent 拼成一段 block-causal 序列,每一步同时预测未来几何 latent $\tilde{Z}^{(L_s)}_{t'+1}$ 和一个未来动作 token $\tilde{a}_{t'}$;这个 $\tilde{a}_{t'}$ 被复制 $V$ 次贴回各视图,再丢进剩下的 GFM block 里走完一次前向,就同时得到未来深度图和最终 action chunk $\hat{a}_t$。整张网络只剩 1.4B 参数、983.2M 可训练,全部用 64 张 GH200 训约 96 小时。
与已有方法的本质区别是把 GFM 从『静态特征提取器』升级为『时序世界模型的共享骨干』。Video WAM 把未来预测放在 2D 像素 latent 空间、用扩散采样解动作;geometry-aware VLA 把 GFM 当外挂蒸馏源;GAM 则把动作 token 和未来几何 token 塞进同一段自回归序列、由同一组 GFM block 共同解码,让『在哪里』和『怎么动』在特征层就被显式绑死,而不是各自用各自的 head。
方法步骤详情
(1)切分骨干:选取 split layer $L_s=12$,令 $E_{\le L_s}=f^{(L_s)}\circ\cdots\circ f^{(1)}$,$D_{>L_s}=f^{(M)}\circ\cdots\circ f^{(L_s+1)}$;前 13 个 block 冻结不动以保留几何先验,block 13–39 与 predictor 一起训练。(2)观测编码:对窗口内每个时刻 $t'\in\{t-H+1,\ldots,t\}$,先按 GFM 原生 patch embedding 得到 $Z^{(0)}_{t'}\in\mathbb{R}^{V(1+P)\times d}$,再过 $E_{\le L_s}$ 得到每个时刻的几何 latent $Z^{(L_s)}_{t'}$。(3)构造 predictor 输入:把语言 token $L_\ell$(冻结 T5 编码)、本体感知 token $p_{t'}=\psi_s(s_{t'})$、上一动作 token $q_{t'}=\psi_a(a_{t'-1})$ 与几何 latent 拼成 $X=[L_\ell;U_{t'-H+1};\ldots;U_{t'}]$,其中 $U_{t'}=[p_{t'};q_{t'};Z^{(L_s)}_{t'}]$。(4)因果预测:用 block-causal self-attention 的 12 层因果 Transformer $g_\phi$ 处理 $X$,从最后一层读出未来几何 slot 的预测 $\tilde{Z}^{(L_s)}_{t'+1}$ 和动作 slot 的预测 $\tilde{a}_{t'}$,并对 $\tilde{Z}^{(L_s)}_{t'+1}$ 施加 feature alignment loss 对齐冻结 GFM 给出的下一帧真实 latent $Z^{(L_s)}_{t'+1}$。(5)回灌解码:把 $\tilde{a}_{t'}$ 按视图复制 $V$ 份,与各视图的 $\tilde{Z}^{(L_s)}_{v,t'+1}$ 拼好后送入 $D_{>L_s}$,在保持因果 mask 扩展到 GFM 全局 attention 层的前提下,一次前向得到 $\tilde{Z}^{(M)}_{t'+1}$。(6)双头输出:原始 DPT depth head $h_{\mathrm{depth}}$ 输出未来深度 $\tilde{D}_{t'+1}$(用 simulator ground truth 或 GFM 自身伪深度做监督),轻量 action head $h_{\mathrm{act}}$ 聚合所有上下文时间步的 action token,回归 $\hat{a}_t\in\mathbb{R}^{C\times d_a}$,$C=8$。(7)多任务损失:$\mathcal{L}_{\mathrm{total}}=\lambda_{\mathrm{act}}\mathcal{L}_{\mathrm{act}}+\lambda_{\mathrm{feat}}\mathcal{L}_{\mathrm{feat}}+\lambda_{\mathrm{depth}}\mathcal{L}_{\mathrm{depth}}$,权重分别取 3、1、3,$\mathcal{L}_{\mathrm{act}}$ 为 $\ell_1$ 回归、$\mathcal{L}_{\mathrm{feat}}$ 为 $\tilde{Z}^{(L_s)}_{t'+1}$ 与 $Z^{(L_s)}_{t'+1}$ 的 $\ell_1$ 对齐、$\mathcal{L}_{\mathrm{depth}}$ 采用原 GFM 的 scale-invariant + gradient-matching 深度损失。(8)推理:用 KV-cache 单次前向只跑当前观测 $o_t$ 和上一动作 $a_{t-1}$,在 GH200 + CUDA Graphs 下达到 6.9 ms 每步。
技术新颖性
技术新颖性体现在三层叠加的耦合:(a)共享骨干设计——以往要么是 GFM 静态特征提取器,要么是 VLM/视频模型外挂 GFM,GAM 第一次把 GFM 自身劈成 encoder+decoder 并在中间塞入世界模型;(b)动作与几何在 latent 层共享一段自回归序列,而不是各用各的扩散/flow-matching head,由此天然解决『动作要落在正确的 3D 位置上』的对应问题;(c)极小参数预算下获得强鲁棒性——总参数 1.4B、可训练 983.2M,却比 2–7B 的 VLA/WAM 在相机扰动、真实场景 OOD 上更稳,证明几何先验比语言/视频先验对接触式操作更关键。
实验结果
(1)LIBERO 原始 4 个 suite(每 suite 10 个任务 × 50 trials,共 500 次 rollout)平均:GAM 97.6%,略低于 Cosmos-Policy 的 98.5% 但高于 OpenVLA-OFT 97.1%、$\pi_0.5$ 96.9% 和 $\pi_0$ 91.3%,原始 LIBERO 已经饱和、差距不大。(2)LIBERO-Plus 扰动评测是主战场:GAM 把整体成功率拉到 85.5%($\downarrow$12.1),是除 $\pi_0.5$(84.6%)之外仅有的两个还能保持 84+% 的方法;横向看 Cosmos-Policy 82.4%($\downarrow$16.1)、OpenVLA-OFT 69.6%($\downarrow$27.5)、NORA 39.0%、UniVLA 42.9%。(3)按扰动维度拆开看,GAM 在 Camera(83.1%,+9.7 个百分点 vs 最好的 $\pi_0.5$/Cosmos)、Background(94.3%)、Noise(95.3%)三项上同时第一;Robot 初始位姿变化 70.0%,Language 变体 84.8%,Light 97.2%,Layout 79.1%。(4)真实机器人 4 个任务(Pick & Place、Stack milk & cube、Place pot & pan、Insert cube into covered pot,每个任务 200 次左右示教、20 次测试,10 次 ID + 10 次把外置相机平移 85 cm + 旋转 45°):GAM 在 ID 和 OOD 两个设置上都显著领先 $\pi_0.5$ 与 Spatial Forcing,尤其在 OOD 设置下保持稳健(图 4),证明几何先验确实迁移到了物理硬件。(5)RoboCasa-Kitchen 24 个任务平均 69.4%,比第二名 Cosmos-Policy 67.1%、FLARE 66.4%、GROOT-N1.5 + HAMLET 66.4% 都高。(6)速度和模型大小:单卡 GH200 + CUDA Graphs 下 GAM 仅 6.9 ms(≈145 Hz),相对 Cosmos-Policy 382.4 ms 实现约 55.4× 加速,同时参数 1.4B 比 $\pi_0.5$ 3.3B、OpenVLA-OFT 7B 都小一个数量级;在统一 Torch Compile 设置下对比也仍然领先(17.5 ms vs 29.2 ms / 70.1 ms / 382.4 ms)。(7)消融:去掉 pretrain 后 LIBERO 原值变化不大但 LIBERO-Plus 从 89.7% 直接掉到 73.4%,说明 pretrain 是鲁棒性的主因;split layer $L_s=12$ 处预测性能峰值(Orig 99.6%、Plus 89.7%),过早($L_s=0$, Orig 5.4%)或过晚($L_s=27$ 为 0.0%、$L_s=33$ 为 0.0%、$L_s=39$ 为 0.0%)都会彻底崩溃,验证中间层是 frame-wise 和 cross-view attention 的『最佳缝合点』。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| LIBERO-Plus Camera Perturbation | Success Rate (%) | 83.1 | π0.5 72.0, Cosmos-Policy 73.4, OpenVLA-OFT 56.4, π0.5+Spatial Forcing 0.1, π0.5+ROCKET 30.9 | +9.7 个百分点 vs 上一档最强基线 |
| LIBERO-Plus Overall | Success Rate (%) | 85.5 (↓12.1) | π0.5 84.6 (↓12.3), Cosmos-Policy 82.4 (↓16.1), OpenVLA-OFT 69.6 (↓27.5), NORA 39.0, UniVLA 42.9 | +1 个百分点 vs 最强基线,+15.9 个百分点 vs OpenVLA-OFT |
| LIBERO Original | Success Rate (%) | 97.6 | Cosmos-Policy 98.5, OpenVLA-OFT 97.1, π0.5 96.9 | 差距 0.9–0.0 个百分点内(基本饱和) |
| RoboCasa-Kitchen (24 tasks) | Average Success Rate (%) | 69.4 | Cosmos-Policy 67.1, FLARE 66.4, GROOT-N1.5+HAMLET 66.4, π0 62.5 | +2.3 个百分点 vs 第二名 |
| Inference Latency (GH200 + CUDA Graphs) | ms / step | 6.9 (≈145 Hz) | Cosmos-Policy 382.4, OpenVLA-OFT 77.8, π0.5 29.2 | 55.4× faster vs Cosmos-Policy, 11.3× vs OpenVLA-OFT |
| Total Parameters | B | 1.4 | OpenVLA-OFT 7, π0 3.3, Cosmos-Policy 2, π0.5 3.3 | 5× smaller than OpenVLA-OFT, 1.4× smaller than Cosmos-Policy |
| Real-world Pick-and-Place (ID vs OOD) | Success Rate (%) | ≈60% (ID) / ≈40% (OOD) | π0.5 ≈40% / ≈20%, Spatial Forcing 低得多 | OOD 场景显著领先 |
局限与改进
作者在结论里明确承认了一条:GAM 的语言理解和常识推理能力受限于冻结的 T5 文本编码器,对复杂语义、长链指令、组合推理仍显吃力,未来需要把更大的 LLM 或外部推理模块塞进 predictor。我自己的观察还有几条:(1)真实机器人监督信号是 GFM 自身伪深度——这等同于让 GFM 自监督自己,并不是真值,真机几何泛化的极限有待更系统评估;(2)只展示了 4 个真机任务(200 次示教量级),距离『通用策略』还很远;(3)backbone 固化为 DA3-Giant,没有 ablation 不同 GFM 选型(VGGT vs DA3)的迁移性;(4)单臂 7-DoF 末端动作空间,没有覆盖灵巧手或双臂;(5)只在 GH200 + CUDA Graphs 上报延迟,未给出边缘部署(Jetson、Orin)实测,对『推理快』的工业价值要打折看。
独立分析的弱点
(1)语言瓶颈:$L_\ell$ 用冻结 T5-Small 提取,对指令里多对象关系、否定、复合条件的解析能力弱,比如 LIBERO-Plus Language 84.8% 已被 $\pi_0.5$ 的 86.5% 反超,可补一条 LLM-as-token 路径或换成更强开源 LLM。(2)伪深度自监督循环:真机训练时未来深度 $\tilde{D}_{t'+1}$ 的 ground truth 直接来自同一 GFM 的推理——若 GFM 在新场景下深度估计有偏,训练信号也偏,等于闭环自证,可改为引入 LiDAR、立体相机或 NeRF-derived depth 作为独立监督来源。(3)真机样本极少:每个任务 169–284 次示教(共约 800 次),远小于仿真 pretraining 的 784K,迁移到新任务需要重新收集大量数据,可补 self-collected data、PlayData 或短视频先验。(4)单模态架构:动作空间固定为末端增量位姿 + 夹爪 7 维($d_a=7$),没有覆盖关节空间控制或连续抓取力度;同理单臂操作,没有双手机器人或四足手臂平台,限制了在装配、长程、家务等场景的适用性。(5)split layer 敏感性高:$L_s$ 一旦选错就完全崩溃($L_s=27/33/39$ 直接 0%),意味着对不同 backbone 需要重新搜索,缺乏自动选择机制,是部署门槛。(6)evaluation 协议不完全统一:作者同时『用公开 checkpoint 重跑』与『直接引用文献数字』,给对比带来一定噪声。
未来方向
作者在结论明确把『集成 LLM 或外部推理模块』列为最直接的下一步;基于本文结果,我观察还可以向以下方向延伸:(i)把 split 设计做成 NAS-style 的可学习结构,让模型自己发现最优 $L_s$;(ii)把 action chunk 从 $C=8$ 扩到长程,并设计配套的分层 controller 处理长任务;(iii)把 T5 替换为冻结或微调的 7B 级 LLM,通过 cross-attention 把语言 token 注入 predictor 而不是简单拼接;(iv)在真机引入主动感知(比如额外可动相机)与深度传感器,破解伪深度自监督闭环;(v)把 GAM 扩展到 dual-arm / dexterous hand / quadruped arm 等本体,并在更难的真实任务(如整理桌面、烹饪辅助)上验证;(vi)借鉴 WAM 思路,把 diffusion 用在 action head 同时保留 GFM 几何解码,但用 predictor 的 latent 作为条件;(vii)做端到端的 sim-to-real:直接把仿真 pretrain 模型零样本搬到真机,并在真机只用少量 fine-tune。
复现评估
论文的复现性总体较好:作者来自 KAIST CVLab 和 ETH Zurich,列出项目主页 https://cvlab-kaist.github.io/Geometric-Action-Model;backbone 公开(DA3-Giant);训练数据使用 OpenX-Embodiment、MimicGen、RoboCasa365,全部开源下载;评测协议沿用 OpenVLA/OpenVLA-OFT 在 LIBERO + LIBERO-Plus 上的标准;消融做了 split layer、预测损失、context horizon、action 监督位置四项,数字稳定可重复。算力门槛偏高:pretrain 用 64 张 NVIDIA GH200、batch 1024、约 96 小时;post-train 16 张 GH200、batch 160、约 48 小时。模型相对小(1.4B),单卡 GH200 + CUDA Graphs 就能跑推理;用户需熟悉 PyTorch + AdamW + DPT head + Hugging Face Transformers,可改写为 LoRA 微调以进一步压低资源。代码仓库未在 README 明确给出,但作者大概率会开源;目前来看主要复现挑战在 64 张 GH200 的算力而非代码细节。
论文图表
横轴为 LIBERO-Plus 定义的相机扰动难度 5 个等级(越大越难),纵轴为成功率;GAM 在每个等级上都领先 Cosmos-Policy、$\pi_0.5$ 和 ROCKET,并在 L5 极限扰动下仍保持可观优势。
用连续难度等级定量刻画 GAM 的几何先验优势,比单独一个数字更有说服力,也是论文里关于『几何先验对相机鲁棒性』最直接的实证。
四组子图(a-d)分别覆盖 LIBERO-Spatial / Object / Long / Goal,每个任务给出当前 RGB、当前深度、GT 未来深度和 GAM 预测的未来深度 5 个时间步 T=0–T=10/11 的可视化;可以肉眼看到预测深度与 GT 深度在结构上吻合。
唯一直接展示 $\mathcal{L}_{\mathrm{depth}}$ 学到了什么的图,也是 $\tilde{D}_{t'+1}$ 不仅仅作为辅助损失、而是可以用作世界模型产出的关键证据。