面向机器人强化学习的通用过程奖励建模
General Process Reward Modeling for Robotic Reinforcement Learning
多视角奖励融合与逐步奖励离散化为视觉研究者提供了从多视角观测中提取细粒度操作进度信号的通用范式,可直接复用于机器人操作策略的稠密奖励建模与单样本任务适配。
原文依据 leverages Step-wise Reward Discretization for structural understanding and Multi-Perspective Reward Fusion to overcome perceptual limitations
摘要译文
由模型翻译 · 术语统一 · 请对照原文核对关键结论将强化学习应用于真实世界机器人的首要障碍在于有效奖励函数的设计。尽管近年来基于学习的过程奖励模型(PRM)是一个有前景的方向,但它们往往受制于两个根本性局限:其奖励模型缺乏步骤感知理解且依赖单视角感知,导致对细粒度操作进度的评估不可靠;其奖励塑形流程在理论上不健全,常引发语义陷阱从而误导策略优化。为解决这些问题,我们提出 Dopamine-Reward,一种新颖的奖励建模方法,用于从多视角输入中学习通用、具备步骤感知的过程奖励模型。其核心是通用奖励模型(GRM),在超过 3400 小时的大规模数据集上训练,利用逐步奖励离散化实现结构性理解,并借助多视角奖励融合克服感知局限。在 Dopamine-Reward 基础上,我们提出 Dopamine-RL,一个稳健的策略学习框架,采用理论上健全的策略不变奖励塑形方法,使智能体能够利用稠密奖励进行高效自我改进而不改变最优策略,从而从根本上避免语义陷阱。在多种仿真与真实世界任务上的大量实验验证了我们的方法。GRM 在奖励评估上达到最优准确率,基于 GRM 的 Dopamine-RL 显著提升策略学习效率。例如,在 GRM 以单样本方式从单条专家轨迹适配到新任务后,所得奖励模型使 Dopamine-RL 仅用 150 次在线 rollout(约 1 小时真实机器人交互)便将策略从接近零提升至 95% 成功率,同时保持跨任务的强泛化能力。
英文原文摘要 1861 字符
The primary obstacle for applying reinforcement learning (RL) to real-world robotics is the design of effective reward functions. While recently learning-based Process Reward Models (PRMs) are a promising direction, they are often hindered by two fundamental limitations: their reward models lack step-aware understanding and rely on single-view perception, leading to unreliable assessments of fine-grained manipulation progress; and their reward shaping procedures are theoretically unsound, often inducing a semantic trap that misguides policy optimization. To address these, we introduce Dopamine-Reward, a novel reward modeling method for learning a general-purpose, step-aware process reward model from multi-view inputs. At its core is our General Reward Model (GRM), trained on a vast 3,400+ hour dataset, which leverages Step-wise Reward Discretization for structural understanding and Multi-Perspective Reward Fusion to overcome perceptual limitations. Building upon Dopamine-Reward, we propose Dopamine-RL, a robust policy learning framework that employs a theoretically-sound Policy-Invariant Reward Shaping method, which enables the agent to leverage dense rewards for efficient self-improvement without altering the optimal policy, thereby fundamentally avoiding the semantic trap. Extensive experiments across diverse simulated and real-world tasks validate our approach. GRM achieves state-of-the-art accuracy in reward assessment, and Dopamine-RL built on GRM significantly improves policy learning efficiency. For instance, after GRM is adapted to a new task in a one-shot manner from a single expert trajectory, the resulting reward model enables Dopamine-RL to improve the policy from near-zero to 95% success with only 150 online rollouts (approximately 1 hour of real robot interaction), while retaining strong generalization across tasks.