通过集中训练与分解评价,实现机器人抓取的高效在线强化学习。
Efficient Real-World Online Reinforcement Learning for Robot Manipulation via Centralized Training and Critic Decomposition

- 采用集中训练+分布式执行框架,共享多头评价网络。
- 任务与抓取奖励分离设计,成功率提升至95%以上。
- 适合需要持续优化的真实机器人抓取任务,尤其擅长复杂场景。
现实世界中的在线强化学习为直接在物理环境中训练机器人操作策略提供了可行路径,避免了仿真到现实的差距,并可通过人机交互实现策略的持续优化。现有方法虽在人类干预下实现了样本高效学习,但仅限于小范围随机化,且面临多智能体同时训练带来的非平稳性挑战。为此,本文提出统一框架,结合集中训练与分散执行(CTDE)及混合奖励架构(HRA),使多个执行者共享一个中心化的多头评价器。该评价器被分解为任务头和抓取头,分别对应稀疏的任务奖励与基于势能的抓取奖励。相应地,重构评价器与执行器目标函数,利用分解后的Q值并显式处理离散夹爪策略的类别动作分布。实验表明,所提框架显著提升了样本效率与策略性能。我们在两台机械臂和一个模拟人形机器人上验证了该方法,在网球球和香蕉拾取放置、锅具复位、模拟积木重定位任务中应用维度级领域随机化,随机化范围约为之前工作的5-25倍。相比最先进基线,成功率从60%提升至80%(网球球拾取),60%提升至90%(香蕉拾取),25%提升至95%(模拟积木重定位),并在基线持续失败的任务中取得成功。视频与更多细节见项目网站:https://hil-harc.github.io/。
原文摘要 · Abstract (English)
Real-world online reinforcement learning (RL) provides a promising approach for training robotic manipulation policies directly in the physical world, avoiding the sim-to-real gap and enabling continuous policy refinement through human-in-the-loop interaction. Recent methods have demonstrated sample-efficient learning through human intervention but remain limited to small randomization ranges and encounter challenges with the non-stationarity induced by concurrently training multiple agents. To address these limitations, we introduce a unified framework that combines centralized training with decentralized execution (CTDE) and a Hybrid Reward Architecture (HRA). This enables multiple actors to share a centralized multi-head critic. The critic is decomposed into task and grasp heads, corresponding to the sparse task reward and a potential-based grasping reward, respectively. We accordingly reformulate the critic and actor objectives to exploit the decomposed Q-values while explicitly accounting for the categorical action distribution of the discrete gripper policy. Experimental results demonstrate that the proposed framework substantially improves both sample efficiency and policy performance. We validate our approach on two robotic arms and a simulated humanoid robot across tennis ball and banana pick-and-place, pot reset, and simulated block relocation tasks under dimension-wise domain randomization, approximately 5-25x larger than those considered in prior work. Compared with a state-of-the-art baseline, our method improves the success rate from 60% to 80% on tennis ball pick-and-place, from 60% to 90% on banana pick-and-place, and from 25% to 95% on simulated block relocation, while also successfully accomplishing a task where the baseline consistently fails. Videos and more details are available at our project website: https://hil-harc.github.io/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。