用强化学习提升机器人视觉推理能力,效果超越SFT和GPT-4o。
Robot-R1: Reinforcement Learning for Enhanced Embodied Reasoning in Robotics
- 通过强化学习优化机器人视觉推理,预测任务所需关键点状态。
- 7B参数模型在低级动作控制推理上超越GPT-4o。
- 新基准评估多样化的具身推理能力,适合机器人控制研究者。
大型视觉语言模型(LVLM)在结合具身推理与机器人控制方面展现出巨大潜力。当前普遍采用监督微调(SFT)训练,但其数据集多为启发式构建,未专门优化机器人控制性能,且易导致灾难性遗忘和泛化能力下降。为此,我们提出Robot-R1框架,利用强化学习增强针对机器人控制的具身推理能力。Robot-R1基于当前场景图像和来自专家示范的环境元数据,学习预测任务完成所需的下一关键点状态。受DeepSeek-R1启发,该方法采样基于推理的响应,并强化能带来更准确预测的策略。为严格评估,我们还引入新基准,要求具备多样化的具身推理能力。实验表明,使用Robot-R1训练的模型在具身推理任务上优于SFT方法。尽管仅有7B参数,其在空间与运动推理等低级动作控制任务上仍超越GPT-4o。
原文摘要 · Abstract (English)
Large Vision-Language Models (LVLMs) have recently shown great promise in advancing robotics by combining embodied reasoning with robot control. A common approach involves training on embodied reasoning tasks related to robot control using Supervised Fine-Tuning (SFT). However, SFT datasets are often heuristically constructed and not explicitly optimized for improving robot control. Furthermore, SFT often leads to issues such as catastrophic forgetting and reduced generalization performance. To address these limitations, we introduce Robot-R1, a novel framework that leverages reinforcement learning to enhance embodied reasoning specifically for robot control. Robot-R1 learns to predict the next keypoint state required for task completion, conditioned on the current scene image and environment metadata derived from expert demonstrations. Inspired by the DeepSeek-R1 learning approach, Robot-R1 samples reasoning-based responses and reinforces those that lead to more accurate predictions. To rigorously evaluate Robot-R1, we also introduce a new benchmark that demands the diverse embodied reasoning capabilities for the task. Our experiments show that models trained with Robot-R1 outperform SFT methods on embodied reasoning tasks. Despite having only 7B parameters, Robot-R1 even surpasses GPT-4o on reasoning tasks related to low-level action control, such as spatial and movement reasoning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。