arXiv:2601.00675cs.RO2026-01被引 49

用视觉语言模型自动构建机器人任务奖励,提升强化学习效率。

RoboReward: General-Purpose Vision-Language Reward Models for Robotics

  • 基于真实机器人数据构建奖励数据集,引入反事实重标注生成失败样例。
  • 训练出4B/8B参数的通用奖励模型,性能优于更大模型。
  • 在真实机器人上部署,接近人工奖励效果,适合机器人强化学习研究者。

有效的奖励设计对基于强化学习的策略优化至关重要。在现实机器人任务中,获取此类奖励通常依赖耗时的人工标注或脆弱的手工设计目标。视觉语言模型(VLMs)展现出作为自动奖励模型的潜力,但其在真实机器人任务中的有效性尚不明确。本文提出:(1) RoboReward,一个基于Open X-Embodiment (OXE) 和 RoboArena 大规模真实机器人语料库构建的机器人奖励数据集与基准;(2) 在该数据集上训练的视觉语言奖励模型(RoboReward 4B/8B)。由于OXE以成功案例为主,缺乏失败样本,我们提出一种负样本数据增强流程,通过反事实重标注成功片段并结合时间截断,生成校准后的负样本和近似失败样本。利用此框架,构建涵盖多种任务与机器人的大规模训练与评估数据集,检验先进VLM在机器人学习中的奖励分配能力。评估发现,无模型在所有任务上均表现不佳,表明仍有巨大提升空间。随后训练的通用4B与8B参数模型,在短时程机器人任务中表现优于更大规模的VLM。最后,将8B模型部署于真实机器人强化学习,其性能超越Gemini Robotics-ER 1.5,并缩小了与人工奖励训练的差距。项目代码、数据集与评测工具已开源,以推动机器人通用奖励模型发展。

原文摘要 · Abstract (English)

A well-designed reward is critical for effective reinforcement learning-based policy improvement. In real-world robotics, obtaining such rewards typically requires either labor-intensive human labeling or brittle, handcrafted objectives. Vision-language models (VLMs) have shown promise as automatic reward models, yet their effectiveness on real robot tasks is poorly understood. In this work, we aim to close this gap by introducing (1) RoboReward, a robotics reward dataset and benchmark built on large-scale real-robot corpora from Open X-Embodiment (OXE) and RoboArena, and (2) vision-language reward models trained on this dataset (RoboReward 4B/8B). Because OXE is success-heavy and lacks failure examples, we propose a negative examples data augmentation pipeline that generates calibrated negative and near-misses via counterfactual relabeling of successful episodes and temporal clipping to create partial-progress outcomes from the same videos. Using this framework, we build a large training and evaluation dataset spanning diverse tasks and embodiments to test whether state-of-the-art VLMs can reliably provide rewards for robot learning. Our evaluation of open and proprietary VLMs finds that no model excels across tasks, highlighting substantial room for improvement. We then train general-purpose 4B- and 8B-parameter models that outperform much larger VLMs in assigning rewards for short-horizon robotic tasks. Finally, we deploy the 8B model in real-robot reinforcement learning and find that it improves policy learning over Gemini Robotics-ER 1.5 while narrowing the gap to RL training with human-provided rewards. We release the full dataset, trained reward models, and evaluation suite on our website to advance the development of general-purpose reward models in robotics: https://crfm.stanford.edu/helm/robo-reward-bench (project website).

机器人视觉语言强化学习奖励模型

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