用大模型生成结构化目标,让强化学习更高效探索。
Goal-Guided Efficient Exploration via Large Language Model in Reinforcement Learning
- 用大模型生成可复用的结构化目标,优先级动态调整。
- 在Crafter和Craftax-Classic上超越现有最佳方法。
- 适合需要长时规划与高效探索的任务场景。
现实世界决策任务通常发生在复杂开放环境中,对强化学习(RL)智能体的探索效率和长时序规划能力提出巨大挑战。一种有前景的方法是利用大语言模型(LLM)增强的RL,借助LLM丰富的先验知识和强大规划能力,引导RL智能体高效探索。然而,现有方法大多依赖频繁且昂贵的LLM调用,且受限于语义不匹配导致性能不佳。本文提出结构化目标引导强化学习(SGRL),通过结构化目标规划器和目标条件动作剪枝器,引导智能体高效探索。其中,结构化目标规划器利用LLM生成可复用的结构化目标函数,动态优先排序目标,并生成前瞻性的目标以引导策略走向更有希望的决策路径。目标条件动作剪枝器采用动作掩码机制,过滤与当前目标不符的动作,约束智能体选择与目标一致的策略。在Crafter和Craftax-Classic上的实验表明,SGRL优于现有最先进方法。
原文摘要 · Abstract (English)
Real-world decision-making tasks typically occur in complex and open environments, posing significant challenges to reinforcement learning (RL) agents' exploration efficiency and long-horizon planning capabilities. A promising approach is LLM-enhanced RL, which leverages the rich prior knowledge and strong planning capabilities of LLMs to guide RL agents in efficient exploration. However, existing methods mostly rely on frequent and costly LLM invocations and suffer from limited performance due to the semantic mismatch. In this paper, we introduce a Structured Goal-guided Reinforcement Learning (SGRL) method that integrates a structured goal planner and a goal-conditioned action pruner to guide RL agents toward efficient exploration. Specifically, the structured goal planner utilizes LLMs to generate a reusable, structured function for goal generation, in which goals are prioritized. Furthermore, by utilizing LLMs to determine goals' priority weights, it dynamically generates forward-looking goals to guide the agent's policy toward more promising decision-making trajectories. The goal-conditioned action pruner employs an action masking mechanism that filters out actions misaligned with the current goal, thereby constraining the RL agent to select goal-consistent policies. We evaluate the proposed method on Crafter and Craftax-Classic, and experimental results demonstrate that SGRL achieves superior performance compared to existing state-of-the-art methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。