用视觉语言模型让智能体通过文字指令完成任务,零样本泛化更强。
From Goal-Conditioned to Language-Conditioned Agents via Vision-Language Models
- 先找高评分环境配置,再用预训练目标导向策略到达该配置。
- 零样本泛化性能超越多任务强化学习基线,无需任务描述训练。
- 通过多视角评估和模型蒸馏提升效率与准确性,适合快速部署。
视觉语言模型(VLM)在语言具身方面潜力巨大,可使语言条件智能体(LCA)执行由文本指定的多样化任务。这促使研究基于强化学习(RL)的LCA,利用渲染环境图像并由VLM评估来获得奖励。若采用单任务RL,为每个新任务训练策略成本高昂且耗时;多任务RL虽为自然替代方案,但需精心设计任务语料库,且泛化能力不稳定。为此,本文提出构建LCA的新分解方法:首先寻找对任务文本具有高VLM得分的环境配置;然后使用预训练的目标条件策略到达该配置。此外,还探索了多项改进措施,包括使用蒸馏模型、从多个视角评估配置以解决单一二维视图带来的歧义问题。在Humanoid环境中验证了该方法,结果表明其在零样本泛化上优于多任务强化学习基线,且训练期间无需任何文本任务描述或环境特异性标注。视频与交互演示见 https://europe.naverlabs.com/text2control。
原文摘要 · Abstract (English)
Vision-language models (VLMs) have tremendous potential for grounding language, and thus enabling language-conditioned agents (LCAs) to perform diverse tasks specified with text. This has motivated the study of LCAs based on reinforcement learning (RL) with rewards given by rendering images of an environment and evaluating those images with VLMs. If single-task RL is employed, such approaches are limited by the cost and time required to train a policy for each new task. Multi-task RL (MTRL) is a natural alternative, but requires a carefully designed corpus of training tasks and does not always generalize reliably to new tasks. Therefore, this paper introduces a novel decomposition of the problem of building an LCA: first find an environment configuration that has a high VLM score for text describing a task; then use a (pretrained) goal-conditioned policy to reach that configuration. We also explore several enhancements to the speed and quality of VLM-based LCAs, notably, the use of distilled models, and the evaluation of configurations from multiple viewpoints to resolve the ambiguities inherent in a single 2D view. We demonstrate our approach on the Humanoid environment, showing that it results in LCAs that outperform MTRL baselines in zero-shot generalization, without requiring any textual task descriptions or other forms of environment-specific annotation during training. Videos and an interactive demo can be found at https://europe.naverlabs.com/text2control
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。