arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2610.18711cs.ROcs.AI

从少量人类视频中微调策略

Steertuning Policies from a Few Human Videos

New York University(纽约大学)

Billy Yan, Isaac Tu, Kelly Lee, Blake Chang, Kenny Ma, Kanad Patel, Dhawal Kabra, Neer Patel, Kevin Wu, Lerrel Pinto, Irmak Guzey

AI总结:

本研究提出Steertuning系统,通过预训练4亿参数扩散策略并仅用5个人类演示微调2分钟,使机器人在12个新环境的12个任务中实现71%的任务成功率,整个部署过程耗时不足5分钟。

摘要:

当前机器人学习的方案需要大量机器人演示才能获得鲁棒且自适应的策略。本研究提出Steertuning系统,使机器人能在新环境中从5个人类演示中学习新任务,整个过程耗时不到5分钟。Steertuning由两部分组成:(1)我们在包含100小时密集标注随机交互的自我中心人类数据集上预训练了一个4亿参数的扩散策略,该预训练使Steertuning对背景差异具有鲁棒性,并在部署到新任务时实现空间和对象泛化;(2)给定新任务的5个演示,我们仅微调模型2分钟(2000次训练更新),并在反向采样期间使用观测结果最接近部署环境的演示来引导策略的动作。结合精心设计的用于 embodiment 迁移的数据处理流水线,Steertuning在12个新环境的12个任务中实现了71%的成功率,环境范围从实验室到新住宅的厨房。从新任务的数据收集到机器人策略部署的整个过程耗时不到5分钟,我们的完整数据集、模型权重、代码库和机器人视频将在网站this https URL上发布。

英文摘要:

Current recipes for robot learning require massive amounts of robot demonstrations to obtain robust and adaptive policies. In this work, we introduce Steertuning, a system that enables robots to learn new tasks in new environments from 5 human demonstrations in under 5 Steertuning consists of two parts. (1) We pretrain a 400M-parameter diffusion policy on our egocentric human dataset, which consists of 100 hours of densely annotated random interactions. This pretraining makes Steertuning robust to background differences and enables spatial and object generalization when deployed on new tasks. (2) Given 5 demonstrations of a new task, we finetune our model for only 2 minutes (2000 training updates) and steer the policy's actions during reverse sampling using the demonstration whose observation is closest to that of the deployment environment. Together with a carefully designed data processing pipeline for embodiment transfer, this enables Steertuning to achieve a 71% success rate across 12 tasks in 12 new environments, ranging from lab settings to kitchens in new homes. The entire process, from data collection for a new task to robot policy deployment, takes less than 5 minutes. Our full dataset, model weights, codebase, and robot videos will be made available at our website: https://steertuning.github.io.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑