【免费下载链接】Robo-Dopamine
[CVPR 2026] Official implementation of "Robo-Dopamine: General Process Reward Modeling for High-Precision Robotic Manipulation"
Robo-Dopamine 是一个已被 CVPR 2026 接收的机器人过程奖励建模(Process Reward Modeling)开源项目,它通过通用奖励模型 GRM(General Reward Model)为机器人操作任务提供密集、稳定的过程奖励信号,从根本上缓解强化学习中长期存在的稀疏奖励难题。对于想用多模态大模型给机器人"打分"的工程师和研究员来说,这份代码库是少见的可直接上手的高质量实现。
🧠 先搞懂:什么是机器人的稀疏奖励难题?
传统强化学习中,机器人往往只有在任务完成的那一刻才收到一个奖励(成功 +1,失败 0),中间成百上千步操作全都没有反馈。这就像学生写一篇论文,老师只在最后给一个分数——过程全对也可能被误判,过程全错也可能侥幸过关。
后果有两个:
- 训练极慢:奖励信号太少,策略难以探索;
- 无法定位错误:机器人不知道自己哪一步走偏了,甚至可能在"走神"期间拿到正反馈。
Robo-Dopamine 的思路很直白:给机器人装一个"多巴胺系统"——每一步都让它尝到甜头或苦头。项目名中的 Dopamine(多巴胺)正是这个比喻:Joy is dopamine's handiwork,快乐是多巴胺的手艺活,无论对人还是对机器人。
整个方案由两大组件构成,官方称之为Dopamine-Reward与Dopamine-RL,下方会逐一拆解。
⚡ 核心一:GRM 通用奖励模型如何把稀疏奖励变密集
GRM 本质上是一个视觉语言模型(VLM),它的输入很特别:
- 一段任务描述(如 "organize the table");
- 多视角图片:初始状态、目标状态、以及待评估的 "BEFORE" 与 "AFTER" 三视图(主视角 + 左右腕部相机);
- 输出:AFTER 相对 BEFORE 的进度变化(Progress Hop 或 Regress Hop,即前进或倒退的百分比)。
三种预测视角,一次融合
项目设计了三种推理模式来估计进度,避免单一视角带来的偏差:
| 模式 | 做法 | 回答的问题 |
|---|---|---|
| Incremental(增量式) | 拿"上一状态"作为 BEFORE | 比上一步好了多少? |
| Forward-Anchored(前向锚定) | 拿"初始状态"作为 BEFORE | 比一开始好了多少? |
| Backward-Anchored(后向锚定) | 拿"目标状态"作为 BEFORE | 离终点更近了多少? |
随后通过Multi-Perspective Progress Fusion(多视角进度融合)把三路预测融合为最终奖励。官方强烈建议将三种模式的输出取平均作为强化学习使用的最终奖励,以获得更稳的信号。推理端的完整提示词(System Prompt)可直接在 examples/inference.py 中阅读,其中对"任务对齐、完整性、视角一致性、忽略光照干扰"等判分准则写得非常细致,本身就是一份很好的奖励建模 checklist。
One-Shot 适配 + 策略不变奖励塑形
拿到通用 GRM 后,还有两把"瑞士军刀":
- One-Shot GRM Adaptation(单样本适配):只给 GRM一条演示轨迹,就能让它适配一个全新任务,大幅降低新任务冷启动成本;
- Policy-Invariant Reward Shaping(策略不变奖励塑形):把 GRM 的密集输出转换成加速学习的奖励信号,且理论上不改变最优策略——这正是过程奖励建模中最容易踩的坑(奖励塑形导致策略跑偏)。
这一设计让整套方案对 RL 算法高度"通吃":PPO、ReinFlow、Cal-QL 等在线/离线/在线到在线算法均可兼容。
📈 效果如何:进度估计曲线直观可见
下面这张图展示了 GRM Monitor 对 6 个不同任务(开抽屉、叠方块、擦桌子、叠裤子等)的逐帧进度估计。绿点代表模型判断为"前进",红点代表"倒退",曲线与真实视觉对齐的进度高度吻合——这正是稀疏奖励给不了的能力:每一步都有可解释的信号。
在项目自带评测基准 Robo-Dopamine-Bench 上,GRM-8B 在 AGIBOT、DROID、Galaxea、EgoDex、LIBERO、RoboCasa 六个数据集上的平均进度识别准确率均超过 96%,明显领先于 GPT-5.2 等通用大模型(评测脚本见 eval/evaluation_grm.py)。
🚀 快速上手:三步在本地跑通推理
环境要求 CUDA ≥ 12.8,Python 3.10:
git clone https://gitcode.com/gh_mirrors/ro/Robo-Dopamine cd Robo-Dopamine conda create -n robo-dopamine python=3.10 conda activate robo-dopamine pip install -r requirements.txt项目内置了一个"整理桌面"的多视角演示(俯视 + 左右腕相机三个视频),对应的目标图长这样:
推理只需几行 Python,核心是调用 examples/inference.py 中的GRMInference.run_pipeline:传入三个视角的视频路径、任务描述、目标图,指定eval_mode为forward/incremental/backward之一即可。若任务没有参考目标图,用空图 examples/blank_goal.png 占位,GRM-2.0 系列同样支持单视角、无目标图的场景。完整的调用示例可直接参考 README.md 的 "Simple Usage" 一节。
🛠️ 进阶:用自己的数据微调一个专属奖励模型
如果希望 GRM 适配自己的机器人平台,仓库提供了完整的数据管线(示例原始数据在 dataset/example_raw_data/):
- 预处理:dataset/utils/0_preprocess_data.py 按采样因子切分视频;
- 生成训练样本:dataset/utils/1_generate_data.py 用分箱采样策略构造 BEFORE/AFTER 对比样本;
- 后处理:dataset/utils/2_posprocess_data.py 做替换增强,产出最终训练集;
- 微调:修改数据注册与路径后,运行 train/scripts/finetune_grm.sh(基于 Qwen-VL + DeepSpeed,配置见 train/scripts/zero2.json)。
一键执行前两步可参考 dataset/process.sh。
✨ 小结:谁应该关注这个项目?
- 做机器人强化学习、被稀疏奖励折磨过的算法工程师——直接抄它的"GRM 密集奖励 + 策略不变塑形"组合拳;
- 做具身智能/VLA 评测的团队——GRM 已被用作 VLA 模型的"裁判"(PRM-as-a-Judge 方向);
- 想研究过程奖励建模本身的新手——从 examples/inference.py 的提示词到 dataset/utils/ 的数据构造,每一步都可读可跑。
一句话总结:Robo-Dopamine 证明了"用多模态大模型给机器人每一步打分"不仅能做,而且能做出比通用大模型更准的密集奖励,这或许就是下一代机器人策略学习的"多巴胺引擎"。
【免费下载链接】Robo-Dopamine
[CVPR 2026] Official implementation of "Robo-Dopamine: General Process Reward Modeling for High-Precision Robotic Manipulation"
相关推荐
破解机器人学习奖励稀疏难题:SARM多步策略聚合技术全解析
破解机器人学习奖励稀疏难题:SARM多步策略聚合技术全解析 在机器人学习领域, 奖励稀疏 是阻碍算法性能提升的关键瓶颈。当机器人需要在复杂环境中完成多步骤任务时
人工智能具身智能机器人深度学习强化学习数据集计算机视觉告别训练波动与稀疏奖励:Easy RL核心问题解决方案
告别训练波动与稀疏奖励:Easy RL核心问题解决方案 你是否在训练强化学习模型时遇到过这些问题:奖励忽高忽低如同过山车?智能体在复杂环境中始终原地打转?本文基
强化学习人工智能教程机器学习深度学习Easy-RL 稀疏奖励专题:设计奖励、好奇心驱动与课程学习的强化学习解决方案
Easy RL 稀疏奖励专题:设计奖励、好奇心驱动与课程学习的强化学习解决方案 导读 本文围绕《Easy RL(蘑菇书)》第十章"稀疏奖励"展开:当环境中绝大多
教程机器学习深度学习
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考