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

基于大语言模型的自动化混合奖励调度用于机器人技能学习

Automated Hybrid Reward Scheduling via Large Language Models for Robotic Skill Learning

National Engineering Laboratory for Big Data System Computing Technology(大数据系统计算技术国家工程实验室)

Changxin Huang, Junyang Liang, Yanbin Chang, Jingzhao Xu, Jianqiang Li

AI总结:

针对高自由度机器人技能学习中多奖励分量统一优化效率低的问题,提出基于大语言模型的自动化混合奖励调度框架AHRS,通过多分支价值网络动态调整各奖励分量权重,在多个任务上平均提升6.48%性能。

摘要:

让高自由度机器人学习特定技能是一项具有挑战性的任务,原因在于机器人动力学的复杂性。强化学习(RL)已成为一种有前景的解决方案;然而,解决此类问题需要设计多个奖励函数,以考虑机器人运动中的各种约束。现有方法通常不加区分地将所有奖励分量相加,以优化强化学习的价值函数和策略。我们认为,这种在策略优化中统一包含所有奖励分量的方式效率低下,并且限制了机器人的学习性能。为了解决这一问题,我们提出了一种基于大语言模型(LLMs)的自动化混合奖励调度(AHRS)框架。该范式在整个策略优化过程中动态调整每个奖励分量的学习强度,使机器人能够以渐进和结构化的方式习得技能。具体而言,我们设计了一个多分支价值网络,其中每个分支对应一个不同的奖励分量。在策略优化过程中,每个分支被赋予一个反映其重要性的权重,这些权重根据大语言模型设计的规则自动计算。大语言模型根据任务描述预先生成一个规则集,在训练过程中,它根据评估每个分支性能的语言提示从规则库中选择权重计算规则。实验结果表明,AHRS方法在多个高自由度机器人任务上平均实现了6.48%的性能提升。

英文摘要:

Enabling a high-degree-of-freedom robot to learn specific skills is a challenging task due to the complexity of robotic dynamics. Reinforcement learning (RL) has emerged as a promising solution; however, addressing such problems requires the design of multiple reward functions to account for various constraints in robotic motion. Existing approaches typically sum all reward components indiscriminately to optimize the RL value function and policy. We argue that this uniform inclusion of all reward components in policy optimization is inefficient and limits the robot's learning performance. To address this, we propose an Automated Hybrid Reward Scheduling (AHRS) framework based on Large Language Models (LLMs). This paradigm dynamically adjusts the learning intensity of each reward component throughout the policy optimization process, enabling robots to acquire skills in a gradual and structured manner. Specifically, we design a multi-branch value network, where each branch corresponds to a distinct reward component. During policy optimization, each branch is assigned a weight that reflects its importance, and these weights are automatically computed based on rules designed by LLMs. The LLM generates a rule set in advance, derived from the task description, and during training, it selects a weight calculation rule from the library based on language prompts that evaluate the performance of each branch. Experimental results demonstrate that the AHRS method achieves an average 6.48% performance improvement across multiple high-degree-of-freedom robotic tasks.

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

↑