用大模型动态优化奖励函数,让机器人技能学习更高效。
Efficient Language-instructed Skill Acquisition via Reward-Policy Co-Evolution
- 奖励与策略协同进化,边学边调奖励函数。
- 仅用少量数据就实现95.3%的平均性能提升。
- 适合高维机器人控制任务,降低对人工设计的依赖。
自主探索与解决任务的能力对具身智能的自我发展至关重要。尽管强化学习可减少人力投入,但复杂高维机器人控制任务中设计奖励函数仍具挑战性。近期大语言模型(LLMs)使自动设计奖励函数成为可能,但现有方法需从头重新训练策略,对奖励函数要求过高。本文提出一种奖励-策略协同进化框架:奖励函数与策略在每轮迭代中动态优化、相互促进。奖励演化将机器人历史最优奖励函数、任务与环境描述转化为文本输入,通过LLM生成多样化的奖励候选。策略演化则融合历史最优与随机策略生成新种群。结合改进的贝叶斯优化,高效识别最强大且具可塑性的奖励-策略组合,进入下一轮协同进化。尽管数据使用量少,该方法在多种高维机器人技能任务中实现了平均95.3%的归一化性能提升。
原文摘要 · Abstract (English)
The ability to autonomously explore and resolve tasks with minimal human guidance is crucial for the self-development of embodied intelligence. Although reinforcement learning methods can largely ease human effort, it's challenging to design reward functions for real-world tasks, especially for high-dimensional robotic control, due to complex relationships among joints and tasks. Recent advancements large language models (LLMs) enable automatic reward function design. However, approaches evaluate reward functions by re-training policies from scratch placing an undue burden on the reward function, expecting it to be effective throughout the whole policy improvement process. We argue for a more practical strategy in robotic autonomy, focusing on refining existing policies with policy-dependent reward functions rather than a universal one. To this end, we propose a novel reward-policy co-evolution framework where the reward function and the learned policy benefit from each other's progressive on-the-fly improvements, resulting in more efficient and higher-performing skill acquisition. Specifically, the reward evolution process translates the robot's previous best reward function, descriptions of tasks and environment into text inputs. These inputs are used to query LLMs to generate a dynamic amount of reward function candidates, ensuring continuous improvement at each round of evolution. For policy evolution, our method generates new policy populations by hybridizing historically optimal and random policies. Through an improved Bayesian optimization, our approach efficiently and robustly identifies the most capable and plastic reward-policy combination, which then proceeds to the next round of co-evolution. Despite using less data, our approach demonstrates an average normalized improvement of 95.3% across various high-dimensional robotic skill learning tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。