让机器人通过自我改进持续学习新技能,减少对人工数据的依赖。
Self-Improving Embodied Foundation Models
- 先用行为克隆和未来步骤预测微调模型,再通过自我改进自主练习。
- 相比大量收集模仿数据,样本效率提升显著,成功率更高。
- 可自主习得泛化能力极强的新技能,现有方法无法实现。
基于网络规模数据训练的基座模型已革新机器人技术,但其在底层控制中的应用仍主要限于行为克隆。受大语言模型强化学习微调成功的启发,我们提出一种两阶段后训练方法。第一阶段为监督微调(SFT),利用行为克隆与“剩余步骤预测”目标对预训练基座模型进行微调。第二阶段为自改进,通过剩余步骤预测提取优质奖励函数与鲁棒成功检测器,使一组机器人可在极少人工干预下自主练习下游任务。在真实世界与仿真机器人上的广泛实验表明,SFT与自改进结合显著提升了样本效率,并带来更高的任务成功率。消融实验证明,网络规模预训练与自改进的结合是高效性的关键。进一步发现,该方法首次实现了当前技术无法达成的能力:自主练习并习得远超模仿数据集范围的泛化新技能。这揭示了将预训练基座模型与在线自改进结合,在机器人中实现自主技能获取的巨大潜力。
原文摘要 · Abstract (English)
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 .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。