用大模型自动评估动作贡献,解决强化学习中奖励稀疏难题
Assessing the Zero-Shot Capabilities of LLMs for Action Evaluation in RL
- 用大模型分解任务为子目标,判断每步是否达成
- 零样本下在MiniHack数据集上有效提升学习效率
- 无需人工设计奖励,适合缺乏标注的复杂任务
时间信用分配是强化学习的核心挑战,需为轨迹中每个动作合理分配达成目标的影响。当反馈延迟且稀疏时,学习信号弱,动作评估困难。传统方法如奖励塑造和选项机制需大量领域知识和人工干预,难以扩展。本文提出基于大模型的信用分配框架CALM,利用大模型自动分解任务、评估状态-动作转换中的子目标达成情况。每当一个选项终止,即视为完成一个子目标,CALM提供辅助奖励信号。该信号可增强稀疏延迟奖励下的学习过程,无需人工设计奖励。我们在MiniHack的人类标注演示数据集上进行了初步评估,结果表明大模型可在零样本设置下有效进行信用分配,无需示例或微调。初步结果表明,大模型的知识是强化学习中信用分配的有前景先验,有助于将人类知识迁移到价值函数中。
原文摘要 · Abstract (English)
The temporal credit assignment problem is a central challenge in Reinforcement Learning (RL), concerned with attributing the appropriate influence to each actions in a trajectory for their ability to achieve a goal. However, when feedback is delayed and sparse, the learning signal is poor, and action evaluation becomes harder. Canonical solutions, such as reward shaping and options, require extensive domain knowledge and manual intervention, limiting their scalability and applicability. In this work, we lay the foundations for Credit Assignment with Language Models (CALM), a novel approach that leverages Large Language Models (LLMs) to automate credit assignment via reward shaping and options discovery. CALM uses LLMs to decompose a task into elementary subgoals and assess the achievement of these subgoals in state-action transitions. Every time an option terminates, a subgoal is achieved, and CALM provides an auxiliary reward. This additional reward signal can enhance the learning process when the task reward is sparse and delayed without the need for human-designed rewards. We provide a preliminary evaluation of CALM using a dataset of human-annotated demonstrations from MiniHack, suggesting that LLMs can be effective in assigning credit in zero-shot settings, without examples or LLM fine-tuning. Our preliminary results indicate that the knowledge of LLMs is a promising prior for credit assignment in RL, facilitating the transfer of human knowledge into value functions.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。