arXiv:2410.13816cs.ROcs.LG2024-10CoRL被引 121

用价值函数重排序动作,让通用机器人模型部署时更靠谱。

Steering Your Generalists: Improving Robotic Foundation Models via Value Guidance

  • 用离线强化学习训练价值函数,指导动作重排序
  • 同一价值函数提升5种不同模型在12个任务上的表现
  • 无需微调模型权重,适配多种机器人和架构

在多样化演示数据集上训练的大规模通用机器人策略,在控制多种机器人和完成多样操作任务方面表现出色。然而,这些策略所依赖的数据质量参差不齐——人类示范本就难做到完美,且数据集越大,越难筛选高质量样本。此外,某一机器人获取的最优数据对另一机器人的训练效果尚不明确。本文提出一种通用、可广泛适用的方法:通过离线强化学习训练的价值函数,在部署阶段对通用机器人策略的动作进行重排序,从而提升性能。该方法称为价值引导策略转向(V-GPS),兼容多种不同架构的通用策略,无需微调或访问策略权重。我们证明,同一价值函数能有效提升五种先进策略在多个机器人平台上的表现,覆盖总计12个任务,实现稳定性能提升。代码与视频见:https://nakamotoo.github.io/V-GPS

原文摘要 · Abstract (English)

Large, general-purpose robotic policies trained on diverse demonstration datasets have been shown to be remarkably effective both for controlling a variety of robots in a range of different scenes, and for acquiring broad repertoires of manipulation skills. However, the data that such policies are trained on is generally of mixed quality -- not only are human-collected demonstrations unlikely to perform the task perfectly, but the larger the dataset is, the harder it is to curate only the highest quality examples. It also remains unclear how optimal data from one embodiment is for training on another embodiment. In this paper, we present a general and broadly applicable approach that enhances the performance of such generalist robot policies at deployment time by re-ranking their actions according to a value function learned via offline RL. This approach, which we call Value-Guided Policy Steering (V-GPS), is compatible with a wide range of different generalist policies, without needing to fine-tune or even access the weights of the policy. We show that the same value function can improve the performance of five different state-of-the-art policies with different architectures, even though they were trained on distinct datasets, attaining consistent performance improvement on multiple robotic platforms across a total of 12 tasks. Code and videos can be found at: https://nakamotoo.github.io/V-GPS

机器人强化学习通用策略价值函数

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。