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

基于大语言模型的强化学习任务奖励引导:LMGT框架

Reward Guidance for Reinforcement Learning Tasks Based on Large Language Models: The LMGT Framework

Shanghai University of Engineering Science(上海工程技术大学);INFLY TECH (Shanghai) Co., Ltd.(上海英飞科技有限公司)

Yongxin Deng, Xihe Qiu, Jue Chen, Xiaoyu Tan

AI总结:

针对强化学习中稀疏奖励场景下探索与利用平衡难、样本效率低的问题,提出利用大语言模型先验知识的LMGT奖励调优框架,经多任务及Housekeep环境验证,性能优于基线且显著降低训练算力需求。

摘要:

强化学习(RL)的环境转移模型存在固有不确定性,因此需要在探索与利用之间实现精妙平衡。这种平衡对于优化计算资源、准确估计智能体的期望奖励至关重要。在奖励稀疏的场景(例如机器人控制系统)中,实现这种平衡尤其具有挑战性。然而,鉴于许多环境拥有丰富的先验知识,在这类场景中从零开始学习可能是冗余的。为解决这一问题,我们提出了语言模型引导奖励调优(Language Model Guided reward Tuning,LMGT)——一种全新的样本高效框架。LMGT利用大语言模型(LLMs)中蕴含的丰富先验知识,以及它们处理非标准数据形式(如维基教程)的能力。通过采用LLM引导的奖励偏移策略,LMGT能够巧妙平衡探索与利用,从而引导智能体的探索行为并提升样本效率。我们已在多种强化学习任务中对LMGT进行了严格评估,并在具身机器人环境Housekeep中开展了测试。结果表明,LMGT的性能始终优于基线方法。此外,研究结果显示我们的框架能够大幅降低强化学习训练阶段所需的计算资源。

英文摘要:

The inherent uncertainty in the environmental transition model of Reinforcement Learning (RL) necessitates a delicate balance between exploration and exploitation. This balance is crucial for optimizing computational resources to accurately estimate expected rewards for the agent. In scenarios with sparse rewards, such as robotic control systems, achieving this balance is particularly challenging. However, given that many environments possess extensive prior knowledge, learning from the ground up in such contexts may be redundant. To address this issue, we propose Language Model Guided reward Tuning (LMGT), a novel, sample-efficient framework. LMGT leverages the comprehensive prior knowledge embedded in Large Language Models (LLMs) and their proficiency in processing non-standard data forms, such as wiki tutorials. By utilizing LLM-guided reward shifts, LMGT adeptly balances exploration and exploitation, thereby guiding the agent's exploratory behavior and enhancing sample efficiency. We have rigorously evaluated LMGT across various RL tasks and evaluated it in the embodied robotic environment Housekeep. Our results demonstrate that LMGT consistently outperforms baseline methods. Furthermore, the findings suggest that our framework can substantially reduce the computational resources required during the RL training phase.

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

↑