Self-Improving Embodied Foundation Models

Embodied AI and Robotics Imitation Learning World Models
2025年09月18日
在基于网络规模数据训练的基础模型已彻底改变了机器人技术的背景下,这些模型在底层控制中的应用仍主要局限于行为克隆。受强化学习阶段在微调大语言模型中取得成功的启发,我们提出了一种面向机器人的两阶段后训练方法。第一阶段为监督微调(SFT),通过两种方式对预训练的基础模型进行微调:a)行为克隆,以及 b)步数预测目标。第二阶段为自我提升(Self-Improvement),利用步数预测能力提取出具有良好形态的奖励函数和鲁棒的成功检测器,从而使一组机器人能够在极少人工监督的情况下自主练习下游任务。通过对真实世界和模拟机器人实体的大量实验,我们这一新颖的后训练方法在具身基础模型上取得了显著成果。首先,我们证明了监督微调与自我提升相结合的方法,在样本效率上显著优于单纯扩大监督学习中的模仿数据采集;并且由此得到的策略具有明显更高的成功率。进一步的消融实验表明,网络规模的预训练与自我提升的结合,是实现这种高样本效率的关键。其次,我们展示了所提出方法的独特优势——这是当前其他方法无法实现的能力:机器人能够自主练习并掌握全新的技能,且这些技能的泛化能力远远超出了训练期间所用模仿学习数据集中观察到的行为范围。这些发现凸显了将预训练基础模型与在线自我提升相结合,在推动机器人实现自主技能获取方面的变革性潜力。我们的项目网站位于 https://self-improving-efms.github.io 。
Foundation models trained on web-scale data have revolutionized robotics, but their application to low-level control remains largely limited to behavioral cloning. Drawing inspiration from the success of the reinforcement learning stage in fine-tuning large language models, we propose a two-stage post-training approach for robotics. The first stage, Supervised Fine-Tuning (SFT), fine-tunes pretrained foundation models using both: a) behavioral cloning, and b) steps-to-go prediction objectives. In the second stage, Self-Improvement, steps-to-go prediction enables the extraction of a well-shaped reward function and a robust success detector, enabling a fleet of robots to autonomously practice downstream tasks with minimal human supervision. Through extensive experiments on real-world and simulated robot embodiments, our novel post-training recipe unveils significant results on Embodied Foundation Models. First, we demonstrate that the combination of SFT and Self-Improvement is significantly more sample-efficient than scaling imitation data collection for supervised learning, and that it leads to policies with significantly higher success rates. Further ablations highlight that the combination of web-scale pretraining and Self-Improvement is the key to this sample-efficiency. Next, we demonstrate that our proposed combination uniquely unlocks a capability that current methods cannot achieve: autonomously practicing and acquiring novel skills that generalize far beyond the behaviors observed in the imitation learning datasets used during training. These findings highlight the transformative potential of combining pretrained foundation models with online Self-Improvement to enable autonomous skill acquisition in robotics. Our project website can be found at https://self-improving-efms.github.io .
许愿