用大模型反馈自动生成密集奖励,让智能体高效学习复杂任务。
Online Intrinsic Rewards for Decision Making Agents from Large Language Model Feedback
- 通过异步大模型服务器实时为智能体经验打标签,生成内在奖励。
- 在NetHack环境中超越现有方法,在无需海量离线数据下实现最优性能。
- 适合需要持续探索与自适应奖励的强化学习应用,如游戏、机器人。
从自然语言描述中自动合成密集奖励是强化学习中的有前景范式,适用于稀疏奖励问题、开放探索和分层技能设计。近期工作利用大语言模型(LLM)先验知识取得进展,但存在重要局限:要么因需对每个观测进行LLM标注而无法扩展到需数十亿环境样本的问题,要么依赖多样化的离线数据集,而此类数据可能不存在或难以获取。本文通过算法与系统级改进解决这些挑战,提出ONI——一种分布式架构,同时学习强化学习策略与基于LLM反馈的内在奖励函数。该方法通过异步LLM服务器标注智能体收集的经验,并将其蒸馏为内在奖励模型。我们探索了多种复杂度不同的奖励建模算法,包括哈希、分类与排序模型。ONI在NetHack Learning Environment的多个挑战性任务中达到当前最优性能,且无需先前工作所需的大型离线数据集。代码已公开于https://github.com/facebookresearch/oni。
原文摘要 · Abstract (English)
Automatically synthesizing dense rewards from natural language descriptions is a promising paradigm in reinforcement learning (RL), with applications to sparse reward problems, open-ended exploration, and hierarchical skill design. Recent works have made promising steps by exploiting the prior knowledge of large language models (LLMs). However, these approaches suffer from important limitations: they are either not scalable to problems requiring billions of environment samples, due to requiring LLM annotations for each observation, or they require a diverse offline dataset, which may not exist or be impossible to collect. In this work, we address these limitations through a combination of algorithmic and systems-level contributions. We propose ONI, a distributed architecture that simultaneously learns an RL policy and an intrinsic reward function using LLM feedback. Our approach annotates the agent's collected experience via an asynchronous LLM server, which is then distilled into an intrinsic reward model. We explore a range of algorithmic choices for reward modeling with varying complexity, including hashing, classification, and ranking models. Our approach achieves state-of-the-art performance across a range of challenging tasks from the NetHack Learning Environment, while removing the need for large offline datasets required by prior work. We make our code available at https://github.com/facebookresearch/oni.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。