arXiv:2606.27180cs.LGcs.AI2026-06

用视觉语言模型自动构建奖励函数,让强化学习更快更靠谱地完成任务。

Automating Potential-based Reward Shaping with Vision Language Model Guidance

论文配图:Automating Potential-based Reward Shaping with Vision Language Model Guidance
图 1 · 摘自论文原文
  • 通过视觉语言模型对比图像对,自动学习奖励函数的潜在势能。
  • 在Meta-World和Franka Kitchen中提升采样效率,且避免奖励作弊。
  • 使用轻量级模型降低计算成本,适合实际部署场景。

稀疏奖励使强化学习代理难以探索,因缺乏中间反馈,且易产生奖励劫持问题。潜在函数型奖励塑造(PBRS)可保证最优策略集不变,但需人工设计状态空间上的启发式势能函数。本文提出VLM-PBRS框架,利用轻量级视觉语言模型(VLM)对图像对进行偏好判断,基于这些偏好训练势能函数模型。由于该方法基于PBRS,保留原始最优策略,无需专家设计奖励项。尽管大型VLM反复调用成本过高,我们采用更高效的轻量VLM,虽偏好标签精度较低,但实验表明仍能显著加速学习。在Meta-World和Franka Kitchen环境中验证了方法有效性,并揭示了VLM偏好准确率与样本效率提升间的关联。贡献包括:(1)首次将VLM偏好学习用于生成PBRS的势能函数;(2)提出一种低开销、可扩展的解决方案;(3)充分实证展示了更高的样本效率与抗奖励劫持鲁棒性。

原文摘要 · Abstract (English)

Sparse rewards are inherently challenging for reinforcement learning agents as they lack intermediate feedback to guide exploration and to correctly attribute the sparse success rewards to relevant parts of the trajectory. Naive reward shaping can induce reward hacking, yielding policies that exploit auxiliary signals instead of solving the intended task. Potential-based reward shaping (PBRS) guarantees preservation of the optimal policy set, but requires the definition of a heuristic potential function over the state space. In this work, we introduce the VLM-guided PBRS framework VLM-PBRS that learns the potential function directly from vision language model (VLM) feedback. We query a lightweight VLM to obtain preferences over image pairs and train a model of the potential function using these preferences. As this approach is based on potential-based reward shaping, it preserves the original optimal policies, and removes the need for expert-designed reward shaping terms. Because large VLMs are prohibitively expensive to invoke repeatedly during policy learning, we employ smaller, more computationally efficient VLMs. Although the resulting preference labels are less accurate, empirical evidence shows that the preference labels can still be used to accelerate learning. We validate our method empirically in the Meta-World and Franka Kitchen environments and highlight the connection between VLM preference label accuracy and sample efficiency improvements. Our contributions are threefold: (1) the first application of VLM preference-based learning to synthesize a potential function for PBRS, (2) a principled, low-cost solution that leverages small VLMs, and (3) extensive empirical demonstration of improved sample efficiency and robustness to reward hacking.

强化学习奖励塑造视觉语言模型样本效率

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。