用轨迹草图和策略感知奖励,提升视觉语言模型反馈的机器人强化学习效果。
VARP: Reinforcement Learning from Vision-Language Model Feedback with Agent Regularized Preferences
- 在终态图像上叠加轨迹草图,让VLM更准确判断动作优劣
- 结合当前策略生成的数据优化奖励模型,使奖励学习更贴合实际表现
- 在复杂任务中成功率超70%,显著优于传统方法
为连续控制机器人设计奖励函数常导致细微偏差或奖励黑客问题,尤其在复杂任务中。偏好型强化学习通过比较反馈学习奖励,避免人工标注,但大规模人类标注仍难实现。现有工作利用视觉语言模型(VLM)自动标注偏好,但仅依赖终态图像难以捕捉完整运动过程。本文提出两阶段解决方案:首先,在终态观察上叠加轨迹草图,帮助VLM更准确判断动作路径,使偏好标注准确率在MetaWorld任务中提升约15-20%;其次,通过引入代理性能对奖励学习进行正则化,确保奖励模型基于当前策略生成的数据优化,使行走类任务的回合回报提升20-30%。在MetaWorld上的实验证明,本方法所有任务的成功率可达70%-80%,远超标准方法的50%以下。结果表明,结合更丰富的视觉表示与代理感知奖励正则化可显著提升学习效果。
原文摘要 · Abstract (English)
Designing reward functions for continuous-control robotics often leads to subtle misalignments or reward hacking, especially in complex tasks. Preference-based RL mitigates some of these pitfalls by learning rewards from comparative feedback rather than hand-crafted signals, yet scaling human annotations remains challenging. Recent work uses Vision-Language Models (VLMs) to automate preference labeling, but a single final-state image generally fails to capture the agent's full motion. In this paper, we present a two-part solution that both improves feedback accuracy and better aligns reward learning with the agent's policy. First, we overlay trajectory sketches on final observations to reveal the path taken, allowing VLMs to provide more reliable preferences-improving preference accuracy by approximately 15-20% in metaworld tasks. Second, we regularize reward learning by incorporating the agent's performance, ensuring that the reward model is optimized based on data generated by the current policy; this addition boosts episode returns by 20-30% in locomotion tasks. Empirical studies on metaworld demonstrate that our method achieves, for instance, around 70-80% success rate in all tasks, compared to below 50% for standard approaches. These results underscore the efficacy of combining richer visual representations with agent-aware reward regularization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。