用多任务阶段感知奖励模型,让机器人自进化抓取更准更快。
SARM2: Multi-Task Stage Aware Reward Modeling for Self Improving Robotic Manipulation

- 基于动作基元的阶段估计+多门专家混合,生成细粒度每步奖励
- 在10个任务上降低80%价值估计误差,成功率达90%以上
- 适合想用低成本自演出来优化机器人策略的研究者
长程操作任务中,视觉-语言-动作(VLA)策略仍严重依赖行为克隆,需高成本高质量示范,且策略易受限于示范分布。奖励模型可减少依赖,通过重加权示范和提供密集监督,支持机器人在线强化学习,但需具备密度高、准确且泛化能力强。现有方法:任务特定的阶段感知模型准确但需逐任务标注;通用视觉-语言模型奖励粗略,难以支撑细粒度长程进展。本文提出SARM2,一种多任务阶段感知奖励模型,结合基于动作基元的阶段估计器与多门专家混合(MMoE)价值头,实现跨任务的密集每步奖励。在此基础上,提出SPIRAL(基于奖励对齐学习的自策略改进框架),利用廉价自主滚动数据迭代优化策略。在10任务基准上,SARM2相较最强基线降低80%价值估计均方误差;应用于SPIRAL时,折叠短裤任务成功率从58%提升至100%,清洁白板任务从50%升至90%,表明高质量密集奖励是构建稳定机器人数据飞轮的关键。
原文摘要 · Abstract (English)
Fine-tuning vision-language-action (VLA) policies for long-horizon manipulation still relies heavily on behavior cloning, which requires costly high-quality demonstrations and keeps policies near the demonstration distribution. Reward models can reduce this dependence by reweighting demonstrations and providing dense supervision for on-robot reinforcement learning (RL), but they must be dense, accurate, and general. Existing methods fall short: task-specific stage-aware models are accurate but require per-task annotations, while general vision-language-model (VLM) reward models are broadly applicable but too coarse for fine-grained long-horizon progress. We introduce RM, a multi-task stage-aware reward model that combines an action-primitive-based stage estimator with a multi-gate Mixture-of-Experts (MMoE) value head to produce dense per-step rewards across manipulation tasks. Building on RM, we further propose SPIRAL (Self-Policy Improvement via Reward-Aligned Learning), an on-policy reward-guided framework that improves VLA policies from cheap autonomous rollouts. On a 10-task benchmark, RM reduces value-estimation MSE by 80% over the strongest baselines; when used in SPIRAL, it improves task success from around 50% to near-perfect performance on Folding Shorts (58% to 100%) and Cleaning Whiteboard (50% to 90%), showing that high-quality dense rewards are key to a stable robot data flywheel. Project website: https://qianzhong-chen.github.io/sarm2.github.io/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。