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

基于专家混合的高效残差学习用于通用灵巧抓取

Efficient Residual Learning with Mixture-of-Experts for Universal Dexterous Grasping

Peking University(北京大学);Beijing Academy of Artificial Intelligence(北京人工智能研究院)

Ziye Huang, Haoqi Yuan, Yuhui Fu, Zongqing Lu

AI总结:

本文提出 ResDex,通过将残差策略学习与专家混合框架结合,利用几何无关基础策略实现跨物体的通用灵巧抓取,在 DexGraspNet 上以 88.8% 成功率达到最优性能,且单 GPU 训练仅需 12 小时。

摘要:

跨多样化物体的通用灵巧抓取是机器人学习中一个根本性但极具挑战性的难题。现有使用强化学习(RL)在大型物体数据集上开发策略的方法面临关键局限,包括多任务学习中复杂的课程设计以及对未见物体的泛化能力有限。为克服这些挑战,我们提出 ResDex,一种将残差策略学习与专家混合(MoE)框架相结合的新方法。ResDex 的显著特点在于使用几何无关的基础策略,这些策略能够在单个物体上高效习得,并具备跨广泛未见物体泛化的能力。我们的 MoE 框架整合了多个基础策略,以促进适用于不同物体的多样化抓取方式。通过学习残差动作以及组合这些基础策略的权重,ResDex 实现了用于通用灵巧抓取的高效多任务强化学习。ResDex 在包含 3,200 个物体的 DexGraspNet 数据集上取得了最先进的性能,成功率达到 88.8%。它对未见物体没有表现出泛化差距,并展示了卓越的训练效率,在单个 GPU 上仅用 12 小时即可掌握所有任务。

英文摘要:

Universal dexterous grasping across diverse objects presents a fundamental yet formidable challenge in robot learning. Existing approaches using reinforcement learning (RL) to develop policies on extensive object datasets face critical limitations, including complex curriculum design for multi-task learning and limited generalization to unseen objects. To overcome these challenges, we introduce ResDex, a novel approach that integrates residual policy learning with a mixture-of-experts (MoE) framework. ResDex is distinguished by its use of geometry-unaware base policies that are efficiently acquired on individual objects and capable of generalizing across a wide range of unseen objects. Our MoE framework incorporates several base policies to facilitate diverse grasping styles suitable for various objects. By learning residual actions alongside weights that combine these base policies, ResDex enables efficient multi-task RL for universal dexterous grasping. ResDex achieves state-of-the-art performance on the DexGraspNet dataset comprising 3,200 objects with an 88.8% success rate. It exhibits no generalization gap with unseen objects and demonstrates superior training efficiency, mastering all tasks within only 12 hours on a single GPU.

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

↑