GEM为智能体大模型提供可扩展的训练与评估环境,推动经验学习发展。
GEM: A Gym for Agentic LLMs
- 构建标准化环境接口,支持异步向量化执行和灵活封装
- 涵盖24个环境、5种训练框架,验证了REINFORCE在密集奖励下的优势
- 兼具训练与评估功能,适合研究智能体学习与强化学习算法
大语言模型的训练范式正从静态数据集转向基于经验的学习,即通过与复杂环境交互来获取技能。为促进这一转变,我们提出了GEM(General Experience Maker),一个专为大语言模型时代设计的开源环境模拟器。类似于传统强化学习中的OpenAI-Gym,GEM提供了标准化的环境-智能体接口,支持异步向量化执行以实现高吞吐量,并具备灵活的封装机制便于扩展。GEM包含多样化的环境、强大的集成工具,以及单文件示例脚本,展示如何与五种主流强化学习训练框架结合使用。同时,我们还提供了基于回溯批量归一化(ReBN)的REINFORCE算法在24个环境上的基准结果,该方法相比GRPO更兼容密集每回合奖励设置,且具有更好的信用分配能力。进一步地,我们利用GEM对PPO、GRPO和REINFORCE在单轮与多轮场景下进行了公平对比,揭示算法设计差异的影响。此外,GEM还可作为便捷的评估工具使用。我们希望该框架能加速未来智能体大模型的研究。
原文摘要 · Abstract (English)
The training paradigm for large language models (LLMs) is moving from static datasets to experience-based learning, where agents acquire skills via interacting with complex environments. To facilitate this transition we introduce GEM (General Experience Maker), an open-source environment simulator designed for the age of LLMs. Analogous to OpenAI-Gym for traditional reinforcement learning (RL), GEM provides a standardized framework for the environment-agent interface, including asynchronous vectorized execution for high throughput, and flexible wrappers for easy extensibility. GEM also features a diverse suite of environments, robust integrated tools, and single-file example scripts demonstrating using GEM with five popular RL training frameworks. Along with this, we also provide a set of baselines across 24 environments using REINFORCE with Return Batch Normalization (ReBN), which -- unlike GRPO -- is compatible with the full RL setting of dense per-turn rewards and offers better credit assignment. We further conduct apple-to-apple benchmarking of PPO, GRPO and REINFORCE in both single- and multi-turn settings using GEM to shed light on the algorithmic designs. Lastly, GEM also functions as a convenient evaluation toolkit besides a training environment. We hope this framework can help accelerate future agentic LLM research.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。