arXiv:2410.09187cs.LGcs.AI2024-10被引 6

用大模型生成任务进展函数,大幅减少奖励设计试错次数。

Automated Rewards via LLM-Generated Progress Functions

  • 让大模型基于任务知识生成进展估计函数,指导奖励设计。
  • 在Bi-DexHands上仅用1/20的样本量达到顶尖性能。
  • 结合进展函数与计数奖励,适合自动化强化学习研发者。

大语言模型具备利用跨任务领域知识自动进行奖励工程的潜力,但通常需要大量试错才能生成有效奖励函数,而每次评估都需完成完整的策略优化过程,成本高昂。本文提出一种由大模型驱动的奖励生成框架,在挑战性任务Bi-DexHands上仅需前序最优方法1/20的奖励函数采样量,即可获得最先进策略。核心思路是将特定任务奖励生成问题转化为粗略估计任务进展的问题。我们的两步法利用大模型的任务领域知识和代码生成能力,构建从当前状态估算任务进展的进度函数;随后基于该进展对状态进行离散化,并在低维状态空间中生成基于计数的内在奖励。实验表明,大模型生成的进展函数与计数型内在奖励的结合是性能提升的关键,而通用哈希计数或直接以进展作为奖励则效果较差。

原文摘要 · Abstract (English)

Large Language Models (LLMs) have the potential to automate reward engineering by leveraging their broad domain knowledge across various tasks. However, they often need many iterations of trial-and-error to generate effective reward functions. This process is costly because evaluating every sampled reward function requires completing the full policy optimization process for each function. In this paper, we introduce an LLM-driven reward generation framework that is able to produce state-of-the-art policies on the challenging Bi-DexHands benchmark with 20x fewer reward function samples than the prior state-of-the-art work. Our key insight is that we reduce the problem of generating task-specific rewards to the problem of coarsely estimating task progress. Our two-step solution leverages the task domain knowledge and the code synthesis abilities of LLMs to author progress functions that estimate task progress from a given state. Then, we use this notion of progress to discretize states, and generate count-based intrinsic rewards using the low-dimensional state space. We show that the combination of LLM-generated progress functions and count-based intrinsic rewards is essential for our performance gains, while alternatives such as generic hash-based counts or using progress directly as a reward function fall short.

强化学习大模型奖励设计自动化

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