Abstract:Reinforcement learning (RL) trains agents to accomplish complex tasks through environmental interaction data, but its capacity is also limited by the scope of the available data. To obtain a knowledgeable agent, a promising approach is to leverage the knowledge from large language models (LLMs). Despite previous studies combining LLMs with RL, seamless integration of the two components remains challenging due to their semantic gap. This paper introduces a novel method, Knowledgeable Agents from Language Model Rollouts (KALM), which extracts knowledge from LLMs in the form of imaginary rollouts that can be easily learned by the agent through offline reinforcement learning methods. The primary challenge of KALM lies in LLM grounding, as LLMs are inherently limited to textual data, whereas environmental data often comprise numerical vectors unseen to LLMs. To address this, KALM fine-tunes the LLM to perform various tasks based on environmental data, including bidirectional translation between natural language descriptions of skills and their corresponding rollout data. This grounding process enhances the LLM's comprehension of environmental dynamics, enabling it to generate diverse and meaningful imaginary rollouts that reflect novel skills. Initial empirical evaluations on the CLEVR-Robot environment demonstrate that KALM enables agents to complete complex rephrasings of task goals and extend their capabilities to novel tasks requiring unprecedented optimal behaviors. KALM achieves a success rate of 46% in executing tasks with unseen goals, substantially surpassing the 26% success rate achieved by baseline methods. Furthermore, KALM effectively enables the LLM to comprehend environmental dynamics, resulting in the generation of meaningful imaginary rollouts that reflect novel skills and demonstrate the seamless integration of large language models and reinforcement learning.

LLMs for Generalizable Language-Conditioned Policy Learning under Minimal Data Requirements

Large Language Model as a Policy Teacher for Training Reinforcement Learning Agents

Large Language Models as Generalizable Policies for Embodied Tasks

ExpeL: LLM Agents Are Experiential Learners

Learning with Language-Guided State Abstractions

Language-Conditioned Offline RL for Multi-Robot Navigation

From Words to Actions: Unveiling the Theoretical Underpinnings of LLM-Driven Autonomous Systems

Zero-shot Model-based Reinforcement Learning using Large Language Models

Synthesizing Programmatic Reinforcement Learning Policies with Large Language Model Guided Search

LgTS: Dynamic Task Sampling using LLM-generated sub-goals for Reinforcement Learning Agents

Policy Learning with a Language Bottleneck

Language-Driven Policy Distillation for Cooperative Driving in Multi-Agent Reinforcement Learning

Learning to Program with Natural Language

From Novice to Expert: LLM Agent Policy Optimization via Step-wise Reinforcement Learning

Generalizable Task Representation Learning for Offline Meta-Reinforcement Learning with Data Limitations

Grounding Language with Visual Affordances over Unstructured Data

LLM4RL: Enhancing Reinforcement Learning with Large Language Models

Enabling Intelligent Interactions between an Agent and an LLM: A Reinforcement Learning Approach

Knowledgeable Agents by Offline Reinforcement Learning from Large Language Model Rollouts

Agent Q: Advanced Reasoning and Learning for Autonomous AI Agents