Learning to Move Before Learning to Do: Task-Agnostic pretraining for VLAs

AK's Picks Embodied AI and Robotics Imitation Learning World Models
视觉—语言—动作(VLA)模型的根本瓶颈在于专家示范数据的严重匮乏——这类数据由观测状态、指令与对应动作三元组构成,而大规模采集此类高质量数据成本高昂。我们认为,这一瓶颈源于将两个截然不同的学习目标混为一谈:一是习得物理操作能力(即“如何运动”),二是实现语义对齐能力(即“该做什么”)。关键在于,仅后者需要语言监督信号。基于这一“解耦假设”(Decomposition Hypothesis),我们提出了“任务无关预训练”(Task-Agnostic Pretraining, TAP)方法,这是一种两阶段框架:第一阶段利用廉价且无需标注的交互数据(包括被弃用的非任务相关轨迹以及机器人自主探索产生的“自玩”数据),通过自监督的逆动力学(Inverse Dynamics)目标,学习可迁移的运动先验知识;第二阶段则仅需极少量专家数据,即可轻量级地将这些运动先验与自然语言指令进行对齐。在SIMPLER基准测试中,TAP仅使用远少于百万级标注轨迹的数据量,即达到了与依赖超百万条专家轨迹训练的模型相当的性能,相较标准行为克隆(behavior cloning)方法实现了10个百分点的绝对性能提升。在真实世界的WidowX机械臂平台上,当摄像头输入遭受扰动时,TAP仍能保持25%的成功率,而依赖互联网规模数据训练的基线模型则完全失效(成功率跌至0%)。这表明,任务无关预训练能够生成鲁棒且可迁移的物理表征,为具身智能(Embodied AI)提供了一条切实可行、可规模化扩展的发展路径。
Vision-Language-Action (VLA) models are fundamentally bottlenecked by the scarcity of expert demonstrations -- triplets of observations, instructions, and actions that are costly to collect at scale. We argue that this bottleneck stems from conflating two distinct learning objectives: acquiring physical competence (how to move) and acquiring semantic alignment (what to do). Crucially, only the latter requires language supervision. Building on this Decomposition Hypothesis, we propose Task-Agnostic Pretraining (TAP), a two-stage framework that first learns transferable motor priors from cheap, unlabeled interaction data -- including discarded off-task trajectories and autonomous robot play -- via a self-supervised Inverse Dynamics objective. A lightweight second stage then grounds these priors in language using minimal expert data. On the SIMPLER benchmark, TAP matches models trained on over 1M expert trajectories while using orders of magnitude less labeled data, yielding a 10% absolute gain over standard behavior cloning. On a real-world WidowX platform, TAP retains 25% success under camera perturbations where internet-scale baselines collapse to 0%, demonstrating that task-agnostic pretraining produces robust, transferable physical representations and offers a scalable path forward for Embodied AI.
许愿