用大模型当导师指导强化学习,加速收敛并复用建议。
Accelerating Reinforcement Learning Algorithms Convergence using Pre-trained Large Language Models as Tutors With Advice Reusing
- 让预训练大模型作为导师,生成指导建议辅助强化学习。
- 实验显示大模型指导使收敛速度提升,54组配置中效果显著。
- 建议复用能进一步提速,但可能降低训练稳定性,适合调参者参考。
强化学习算法在复杂稀疏奖励环境中往往需要长时间训练才能见效。尽管已有奖励塑形和课程学习等加速方法,但这些通常高度依赖领域专家经验。本研究探索将预训练大语言模型(LLM)作为导师,在学生-教师架构中指导强化学习算法,假设其生成的指导可加速收敛。特别关注大模型建议的复用机制对训练动态的影响。通过涵盖54种配置的广泛实证研究,包括不同强化学习算法(DQN、PPO、A2C)、大模型导师(Llama、Vicuna、DeepSeek)及环境(Blackjack、Snake、Connect Four),结果表明:大模型指导显著加速收敛,且最终性能与基线相当;建议复用机制进一步缩短训练时间,但导致收敛更不稳定。研究发现,大模型指导整体有效,但效果受任务、算法与模型组合影响显著。
原文摘要 · Abstract (English)
Reinforcement Learning (RL) algorithms often require long training to become useful, especially in complex environments with sparse rewards. While techniques like reward shaping and curriculum learning exist to accelerate training, these are often extremely specific and require the developer's professionalism and dedicated expertise in the problem's domain. Tackling this challenge, in this study, we explore the effectiveness of pre-trained Large Language Models (LLMs) as tutors in a student-teacher architecture with RL algorithms, hypothesizing that LLM-generated guidance allows for faster convergence. In particular, we explore the effectiveness of reusing the LLM's advice on the RL's convergence dynamics. Through an extensive empirical examination, which included 54 configurations, varying the RL algorithm (DQN, PPO, A2C), LLM tutor (Llama, Vicuna, DeepSeek), and environment (Blackjack, Snake, Connect Four), our results demonstrate that LLM tutoring significantly accelerates RL convergence while maintaining comparable optimal performance. Furthermore, the advice reuse mechanism shows a further improvement in training duration but also results in less stable convergence dynamics. Our findings suggest that LLM tutoring generally improves convergence, and its effectiveness is sensitive to the specific task, RL algorithm, and LLM model combination.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。