← 最新论文
💻 computer science

WAM4D: Fast 4D World Action Model via Spatial Register Tokens

WAM4D 是一种快速的 4D 世界动作模型,它利用轻量级空间寄存器标记将预训练的几何先验转移到因果视频动作 Transformer 中,从而在避免密集几何解码计算成本的同时,实现精确的 3D 操作预测与高效推理。

原作者: Ying Li, Xiaobao Wei, Jiajun Cao, Hao Wang, Xiaowei Chi, Chengyu Bai, Qianpu Sun, Jiajun Li, Xiaojie Zhang, Jian Tang, Sirui Han, Shanghang Zhang

发布于 2026-06-15
📖 1 分钟阅读☕ 轻松阅读

原作者: Ying Li, Xiaobao Wei, Jiajun Cao, Hao Wang, Xiaowei Chi, Chengyu Bai, Qianpu Sun, Jiajun Li, Xiaojie Zhang, Jian Tang, Sirui Han, Shanghang Zhang

原始论文采用 CC BY 4.0 许可(http://creativecommons.org/licenses/by/4.0/)。 ✨ 这是对下方论文的AI生成解释。它不是由作者撰写或认可的。如需技术准确性,请参阅原始论文。 阅读完整免责声明

想象一下,你正在教一个机器人如何拿起一个精致的杯子并将水倒入玻璃杯中。为了做好这件事,机器人不仅需要理解杯子“看起来”是什么样的,还需要理解它在3D空间中的位置、它可能有多少重量,以及如果它撞到桌子会发生什么。

长期以来,机器人的大脑(AI模型)就像是能够画出一幅美丽的未来景象,却无法准确告诉你杯子离你有多远的艺术家。它们可以预测“杯子会在这里”的2D视频,但往往会忽略隐藏的部分或进行抓取物体而不掉落所需的精确距离。

WAM4D 是一个旨在解决这一问题的全新“机器人大脑”。以下是它的工作原理,我们使用简单的类比来解释:

1. 问题所在:“扁平”的未来

目前的多数机器人模型就像是在看一部平面的电视电影。它们可以预测视频的下一帧看起来是什么样的,但并不真正理解场景的“深度”。如果机器人试图抓取一个被其他物体部分遮挡的物体,一个“扁平”的模型可能会猜错位置,因为它无法“看到”3D形状。

2. 解决方案:“幽灵建筑师”(空间寄存器标记/Spatial Register Tokens)

作者创造了一个聪明的技巧,称为空间寄存器标记(Spatial Register Tokens)。你可以把它们想象成“幽灵建筑师”或“隐形的便利贴”,机器人仅在“学习”期间使用它们。

  • 在训练期间: 想象机器人正在学习玩电子游戏。当它练习时,这些“幽灵建筑师”就会出现。它们观察机器人目前所看到的所有历史信息,并询问一个超智能的、预训练好的3D专家(几何基础模型):“基于我们目前所见到的情况,未来的深度图看起来是什么样的?”
  • 教学过程: 机器人尝试预测未来的视频。这些“幽灵建筑师”会检查机器人的预测在3D空间中是否合理。如果机器人预测杯子悬浮在半空中,而实际上它应该在桌子上,幽灵建筑师就会说:“不对,这在3D空间里是错的,”于是机器人就从这个错误中学习。
  • 神奇之处: 这教会了机器人理解3D几何结构,而无需在每次移动时都实际输出一个复杂的3D地图。

3. 结果:“轻量级飞行员”

最棒的部分在于:一旦机器人完成学习,幽灵建筑师就会消失。

  • 在现实世界中: 当机器人实际执行任务(比如拿起杯子)时,它不需要计算复杂的3D地图,也不需要运行沉重的3D解码。它只需使用从“幽灵建筑师”那里学到的“肌肉记忆”。
  • 为什么这很重要: 这就像一名学生在学习过程中通过导师(幽灵建筑师)花费大量时间钻研深奥的数学,但在考试当天,他们只需快速回答问题,而不需要导师在场。这使得机器人既快速又高效,同时依然足够聪明,能够处理3D空间。

4. “交通警察”(因果混合注意力/Causal Mixture Attention)

为了确保机器人不会“作弊”,作者加入了一个“交通警察”系统。在AI领域,“作弊”意味着通过观察未来来预测现在。交通警察确保机器人只能根据已经发生过的事情(过去的视频和动作)来决定下一步该做什么。它严格禁止机器人在做决策时“偷看”未来的深度或未来的视频,从而确保机器人的行为是实时的,就像人类一样。

他们证明了什么?

团队在双臂机器人(RoboTwin)和真实世界机器人(AstriBot)上测试了这个全新的机器人大脑。

  • 更高的准确度: 由于更好地理解了3D形状,机器人在需要精确接触的任务(如堆叠积木、取下笔盖或分类物品)中表现得更加出色。
  • 速度: 尽管它是通过3D几何结构进行学习的,但由于在实际工作中移除了沉重的3D数学计算,它的运行速度与其他快速机器人一样快。
  • 现实世界的成功: 它成功完成了复杂的现实任务,如搬运盘子和分类乐高积木,其表现优于以往未使用这种3D“幽灵建筑师”技巧的模型。

总结

WAM4D 是一个通过在训练期间使用临时的、隐形的3D导师,来学习理解3D世界的机器人大脑。一旦训练完成,它就会忘掉沉重的数学计算,变成一个快速、高效的飞行员,能够精准地在复杂的3D环境中导航。它弥合了“观看视频”与“理解物理世界”之间的鸿沟。

您所在领域的论文太多了?

获取与您研究关键词匹配的最新论文每日摘要——附技术摘要,使用您的语言。

试用 Digest →