arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2405.03341cs.LGcs.AI
收藏

利用大语言模型启发式增强Q学习

Enhancing Q-Learning with Large Language Model Heuristics

School of Computer Science(计算机学院);Wuhan University(武汉大学)

Xiefeng Wu

AI总结:

提出LLM引导的Q学习框架,利用大语言模型作为启发式辅助Q函数学习,提升样本效率并避免性能偏差,实验验证其通用性与鲁棒性。

摘要:

Q学习在序贯决策任务中擅长从反馈中学习,但通常需要大量采样才能实现显著改进。虽然奖励塑形可以提升学习效率,但非基于势的方法会引入影响性能的偏差,而基于势的奖励塑形虽无偏,却缺乏为状态-动作对提供启发式的能力,限制了其在复杂环境中的有效性。大语言模型(LLMs)能够对较简单任务进行零样本学习,但存在推理速度低和偶发幻觉的问题。为应对这些挑战,我们提出了\textbf{LLM引导的Q学习}(LLM-guided Q-learning),这是一个利用LLMs作为启发式来辅助强化学习Q函数学习的框架。我们的理论分析表明,该方法能适应幻觉、提升样本效率,并避免使最终性能产生偏差。实验结果表明,我们的算法具有通用性、鲁棒性,并能防止无效探索。

英文摘要:

Q-learning excels in learning from feedback within sequential decision-making tasks but often requires extensive sampling to achieve significant improvements. While reward shaping can enhance learning efficiency, non-potential-based methods introduce biases that affect performance, and potential-based reward shaping, though unbiased, lacks the ability to provide heuristics for state-action pairs, limiting its effectiveness in complex environments. Large language models (LLMs) can achieve zero-shot learning for simpler tasks, but they suffer from low inference speeds and occasional hallucinations. To address these challenges, we propose \textbf{LLM-guided Q-learning}, a framework that leverages LLMs as heuristics to aid in learning the Q-function for reinforcement learning. Our theoretical analysis demonstrates that this approach adapts to hallucinations, improves sample efficiency, and avoids biasing final performance. Experimental results show that our algorithm is general, robust, and capable of preventing ineffective exploration.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑