通过合成失败轨迹,让机器人获得细粒度的实时奖励反馈。
DenseReward: Dense Reward Learning via Failure Synthesis for Robotic Manipulation

- 自动在仿真中生成多样物理失败数据,无需人工标注。
- 能从视觉和语言中预测每帧的奖励值,精度高于现有模型。
- 适合需要精细反馈的机器人操控强化学习任务。
强化学习有望突破模仿学习的局限,提升机器人策略性能,但其实际应用受限于缺乏可靠的视觉-语言奖励模型,难以提供密集且有信息量的反馈。核心挑战在于:如何大规模获取多样化失败数据,以及如何获得超越稀疏轨迹级成功标签的细粒度奖励信号。传统收集失败轨迹需大量人工参与,而通过重标记成功示范生成的伪失败无法覆盖真实执行中的多种物理失效模式。现有奖励模型多输出稀疏的二值或轨迹级奖励,难以有效指导策略优化。本文提出DenseReward,一种解决上述双重挑战的密集型机器人奖励模型。我们构建了自动化失败数据生成流水线,在仿真中无需人工标注即可合成具有物理真实性的失败轨迹,涵盖碰撞、抓取失败、物体掉落及恢复行为等多样失效模式。DenseReward基于视觉观测与语言指令,预测帧级别的密集奖励分数,实现对任务进展的细粒度评估。实验表明,DenseReward在模拟与真实世界操控任务中,均优于通用视觉语言模型(VLM)和现有机器人奖励模型。进一步验证显示,该模型可有效指导下游模型预测控制与强化学习。我们开源了数据集、训练好的奖励模型及评估套件,以推动故障感知密集奖励建模的发展。
原文摘要 · Abstract (English)
Reinforcement learning holds great promise for improving robot policies beyond the limits of imitation learning. However, its practical adoption remains bottlenecked by the lack of reliable vision-language reward models that provide dense and informative feedback. Two key challenges remain: acquiring diverse failure data at scale and obtaining fine-grained reward signals beyond sparse trajectory-level success labels. Collecting failure trajectories typically requires laborious human effort, while pseudo-failures constructed by relabeling successful demonstrations fail to capture the diverse physical failure modes that arise during robot execution. Meanwhile, existing reward models often predict sparse binary or trajectory-level rewards, which provide limited guidance for efficient policy optimization. We introduce DenseReward, a dense robotic reward model that addresses both challenges. To train DenseReward, we develop an automated failure data generation pipeline that synthesizes physically realistic failure trajectories in simulation without human labeling, covering diverse failure modes such as collisions, missed grasps, object drops, and recovery behaviors. DenseReward predicts dense frame-level reward scores from visual observations and language instructions, enabling fine-grained estimation of task progress throughout an episode. Experiments show that DenseReward outperforms general-purpose VLMs and existing robotic reward models in dense reward prediction across both simulated and real-world manipulation. We further demonstrate that DenseReward provides effective reward guidance for downstream model predictive control and reinforcement learning. We release the dataset, trained reward models, and evaluation suite to support the development of failure-aware dense reward modeling for robot learning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。