arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.04764cs.LG
收藏

代码即奖励:利用视觉语言模型增强强化学习

Code as Reward: Empowering Reinforcement Learning with VLMs

Mila(米拉计算研究所);McGill University(麦吉尔大学);Google DeepMind(谷歌DeepMind);University of California, Berkeley(加利福尼亚大学伯克利分校)

David Venuto, Sami Nur Islam, Martin Klissarov, Doina Precup, Sherry Yang, Ankit Anand

AI总结:

针对VLMs推理昂贵的问题,提出VLM-CaR框架,通过代码生成稠密奖励函数,在多种环境中准确且高效地训练强化学习策略。

摘要:

预训练的视觉语言模型(VLMs)能够理解视觉概念,描述复杂任务并将其分解为子任务,还能对任务完成情况提供反馈。本文旨在利用这些能力来支持强化学习(RL)智能体的训练。原则上,VLMs非常适合这一目的,因为它们能够自然地分析基于图像的观测,并对学习进展提供反馈(奖励)。然而,VLMs的推理计算成本高昂,因此频繁查询它们来计算奖励会显著减慢RL智能体的训练速度。为解决这一挑战,我们提出了一种名为“代码即奖励”(VLM-CaR)的框架。VLM-CaR通过代码生成从VLMs中产生稠密奖励函数,从而显著减轻了直接查询VLM的计算负担。我们证明,通过我们的方法生成的稠密奖励在多种离散和连续环境中非常准确,并且在训练RL策略时比原始的稀疏环境奖励更为有效。

英文摘要:

Pre-trained Vision-Language Models (VLMs) are able to understand visual concepts, describe and decompose complex tasks into sub-tasks, and provide feedback on task completion. In this paper, we aim to leverage these capabilities to support the training of reinforcement learning (RL) agents. In principle, VLMs are well suited for this purpose, as they can naturally analyze image-based observations and provide feedback (reward) on learning progress. However, inference in VLMs is computationally expensive, so querying them frequently to compute rewards would significantly slowdown the training of an RL agent. To address this challenge, we propose a framework named Code as Reward (VLM-CaR). VLM-CaR produces dense reward functions from VLMs through code generation, thereby significantly reducing the computational burden of querying the VLM directly. We show that the dense rewards generated through our approach are very accurate across a diverse set of discrete and continuous environments, and can be more effective in training RL policies than the original sparse environment rewards.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑