让大模型不仅能说话,还能直接操作外部环境,提升复杂任务推理能力。
Expanding the Action Space of LLMs to Reason Beyond Language
- 将模型动作扩展到语言外的专用环境,实现自由切换与交互
- 在排序任务中达99.8%准确率,自发现高效算法
- 适合需要多步规划与外部系统协同的任务研究者
大语言模型虽在自然语言推理上强大,但其行动通常局限于输出词汇符号。与外部环境(如符号运算符或模拟器)交互时,需通过预定义格式的文本表达、解析并路由至接口,使语言负担过重,并依赖人工编写的解析器。为此,本文提出将环境交互内化为超越词汇的扩展动作空间(ExpA),模型可在默认语言环境推理,随时触发路由动作切换至外部环境,仅能调用环境特定操作,接收反馈后可返回语言环境。为促进对新动作空间和环境的有效探索,引入基于反事实策略优化的强化学习方法(EARL)。在需多轮交互与条件规划的任务中,EARL显著优于仅限词汇动作的基线模型,在基于计算器的多任务学习中表现稳健;在部分可观测排序问题中,实现100%的Sort-4准确率,并自主发现性能媲美经典设计的高效算法。
原文摘要 · Abstract (English)
Large Language Models (LLMs) are powerful reasoners in natural language, but their actions are typically confined to outputting vocabulary tokens. As a result, interactions with external environments -- such as symbolic operators or simulators -- must be expressed through text in predefined formats, parsed, and routed to external interfaces. This overloads the model's language with both reasoning and control duties, and requires a hand-crafted parser, external to the LLM. To address this, we decouple environment interactions from language by internalizing them in an Expanded Action space (ExpA), beyond the vocabulary. The model starts reasoning in the default language environment, but may trigger routing actions and switch to an external environment at any time. From there, the model can only invoke environment-specific actions, receive feedback from the environment, and potentially route back to language as a result. To promote effective exploration of the expanded action space and new environments, we introduce ExpA Reinforcement Learning (EARL) with counterfactual policy optimization. On tasks requiring multi-turn interactions and contingent planning, EARL outperforms strong baselines with vocabulary-constrained actions. It performs robustly across calculator-based multi-task learning and, in the partially observed sorting problem, achieves perfect Sort-4 accuracy while self-discovering an efficient algorithm competitive with classical designs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。