用视觉特征生成细粒度奖励,提升机器人长程操作的强化学习效果
Feat2Go: Visual Feature-Grounded Value Estimation for Embodied Reinforcement Learning

- 基于视觉世界模型计算局部相似性,自动划分任务阶段并生成连续进展目标
- 在ManiSkill3上将离域成功率从17.5%提升至82.9%,保持96.9%的域内性能
- 无需人工设计奖励,适配PPO/GRPO,适用于单臂与双臂操作场景
强化学习是提升视觉-语言-动作(VLA)模型能力的有前途方法,可避免模仿学习带来的高数据需求。然而,其在长时程操作任务中常受限于稀疏监督和难设计的奖励信号。本文提出Feat2Go,一种面向具身强化学习的细粒度价值估计框架。该框架首先通过预训练视觉世界模型,测量图像块级相似性以提取连续进展目标,并利用趋势聚类将任务过程划分为语义阶段。随后训练一个具身价值模型,从当前观测与任务指令中预测该结构化进展,并将其用于重塑策略优化中的终端奖励。该方法兼容现有VLA策略强化学习流程(如PPO与GRPO),无需人工奖励工程。在ManiSkill3与RoboTwin 2.0上的大量实验表明,Feat2Go显著提升现有VLA模型性能:在ManiSkill3上,OpenVLAOFT的离域平均成功率达82.9%(原为17.5%),域内性能保持96.9%;在RoboTwin 2.0的领域随机任务设置中,平均成功率达到88.8%,优于先前强化学习方法。
原文摘要 · Abstract (English)
Reinforcement learning is a promising approach for improving the capabilities of vision-language-action (VLA) models while avoiding the heavy data requirements of imitation learning. However, its effectiveness for VLA models is often constrained by sparse supervision and the difficulty of designing informative reward signals for long-horizon manipulation. In this work, we present Feat2Go, a fine-grained value estimation framework for embodied reinforcement learning. Specifically, Feat2Go first derives a continuous progress target from a pretrained visual world model by measuring patch-level similarity to subgoal states and partitioning episodes into semantic stages with trend-based clustering. We then train an embodied value model to predict this structural progress from the current observation and task instruction, and use the predicted value to reshape terminal rewards during policy optimization. The proposed framework is compatible with existing VLA policy reinforcement learning pipelines, including PPO and GRPO, and does not rely on manual reward engineering. Extensive experiments on ManiSkill3 and RoboTwin 2.0 demonstrate that Feat2Go consistently improves the performance of existing VLA models under both single-arm and bimanual manipulation settings. More specifically, on ManiSkill3, Feat2Go improves OpenVLAOFT from 17.5% to 82.9% average out-of-distribution success while retaining 96.9% in-distribution performance. On RoboTwin 2.0, Feat2Go achieves an average success rate of 88.8% in domain-randomized task settings, outperforming prior reinforcement learning methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。