arXiv:2505.15293cs.LGcs.AI2025-05NeurIPS被引 6

用大模型动态生成探索策略,让强化学习更聪明地试错。

LLM-Explorer: A Plug-in Reinforcement Learning Policy Exploration Enhancement Driven by Large Language Models

  • 用大模型分析学习状态,生成任务专属探索概率分布
  • 在Atari和MuJoCo上平均提升37.27%性能
  • 可插件式接入DQN、DDPG等主流算法

策略探索在强化学习中至关重要。现有方法如贪婪策略、高斯过程等依赖预设随机过程,不考虑任务特性,且训练过程中调整僵硬,仅降低方差。受大语言模型分析推理能力启发,我们提出LLM-Explorer,通过采样智能体学习轨迹并提示大模型分析当前策略学习状态,生成未来探索的概率分布。定期更新该分布,构建适配具体任务且动态响应学习状态的随机过程。该设计为可插件模块,兼容DQN系列、DDPG、TD3及其变体。在Atari和MuJoCo基准上的大量实验表明,其显著提升了强化学习的策略探索能力,平均性能提升达37.27%。代码已开源:https://github.com/tsinghua-fib-lab/LLM-Explorer。

原文摘要 · Abstract (English)

Policy exploration is critical in reinforcement learning (RL), where existing approaches include greedy, Gaussian process, etc. However, these approaches utilize preset stochastic processes and are indiscriminately applied in all kinds of RL tasks without considering task-specific features that influence policy exploration. Moreover, during RL training, the evolution of such stochastic processes is rigid, which typically only incorporates a decay in the variance, failing to adjust flexibly according to the agent's real-time learning status. Inspired by the analyzing and reasoning capability of large language models (LLMs), we design LLM-Explorer to adaptively generate task-specific exploration strategies with LLMs, enhancing the policy exploration in RL. In our design, we sample the learning trajectory of the agent during the RL training in a given task and prompt the LLM to analyze the agent's current policy learning status and then generate a probability distribution for future policy exploration. Updating the probability distribution periodically, we derive a stochastic process specialized for the particular task and dynamically adjusted to adapt to the learning process. Our design is a plug-in module compatible with various widely applied RL algorithms, including the DQN series, DDPG, TD3, and any possible variants developed based on them. Through extensive experiments on the Atari and MuJoCo benchmarks, we demonstrate LLM-Explorer's capability to enhance RL policy exploration, achieving an average performance improvement up to 37.27%. Our code is open-source at https://github.com/tsinghua-fib-lab/LLM-Explorer for reproducibility.

强化学习大模型应用探索策略可插件

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。