UniGraspTransformer: Simplified Policy Distillation for Scalable Dexterous Robotic Grasping

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wang, Wenbo, Wei, Fangyun, Zhou, Lei, Chen, Xi, Luo, Lin, Yi, Xiaohan, Zhang, Yizhong, Liang, Yaobo, Xu, Chang, Lu, Yan, Yang, Jiaolong, Guo, Baining
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913718155083776
author Wang, Wenbo
Wei, Fangyun
Zhou, Lei
Chen, Xi
Luo, Lin
Yi, Xiaohan
Zhang, Yizhong
Liang, Yaobo
Xu, Chang
Lu, Yan
Yang, Jiaolong
Guo, Baining
author_facet Wang, Wenbo
Wei, Fangyun
Zhou, Lei
Chen, Xi
Luo, Lin
Yi, Xiaohan
Zhang, Yizhong
Liang, Yaobo
Xu, Chang
Lu, Yan
Yang, Jiaolong
Guo, Baining
contents We introduce UniGraspTransformer, a universal Transformer-based network for dexterous robotic grasping that simplifies training while enhancing scalability and performance. Unlike prior methods such as UniDexGrasp++, which require complex, multi-step training pipelines, UniGraspTransformer follows a streamlined process: first, dedicated policy networks are trained for individual objects using reinforcement learning to generate successful grasp trajectories; then, these trajectories are distilled into a single, universal network. Our approach enables UniGraspTransformer to scale effectively, incorporating up to 12 self-attention blocks for handling thousands of objects with diverse poses. Additionally, it generalizes well to both idealized and real-world inputs, evaluated in state-based and vision-based settings. Notably, UniGraspTransformer generates a broader range of grasping poses for objects in various shapes and orientations, resulting in more diverse grasp strategies. Experimental results demonstrate significant improvements over state-of-the-art, UniDexGrasp++, across various object categories, achieving success rate gains of 3.5%, 7.7%, and 10.1% on seen objects, unseen objects within seen categories, and completely unseen objects, respectively, in the vision-based setting. Project page: https://dexhand.github.io/UniGraspTransformer.
format Preprint
id arxiv_https___arxiv_org_abs_2412_02699
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle UniGraspTransformer: Simplified Policy Distillation for Scalable Dexterous Robotic Grasping
Wang, Wenbo
Wei, Fangyun
Zhou, Lei
Chen, Xi
Luo, Lin
Yi, Xiaohan
Zhang, Yizhong
Liang, Yaobo
Xu, Chang
Lu, Yan
Yang, Jiaolong
Guo, Baining
Robotics
We introduce UniGraspTransformer, a universal Transformer-based network for dexterous robotic grasping that simplifies training while enhancing scalability and performance. Unlike prior methods such as UniDexGrasp++, which require complex, multi-step training pipelines, UniGraspTransformer follows a streamlined process: first, dedicated policy networks are trained for individual objects using reinforcement learning to generate successful grasp trajectories; then, these trajectories are distilled into a single, universal network. Our approach enables UniGraspTransformer to scale effectively, incorporating up to 12 self-attention blocks for handling thousands of objects with diverse poses. Additionally, it generalizes well to both idealized and real-world inputs, evaluated in state-based and vision-based settings. Notably, UniGraspTransformer generates a broader range of grasping poses for objects in various shapes and orientations, resulting in more diverse grasp strategies. Experimental results demonstrate significant improvements over state-of-the-art, UniDexGrasp++, across various object categories, achieving success rate gains of 3.5%, 7.7%, and 10.1% on seen objects, unseen objects within seen categories, and completely unseen objects, respectively, in the vision-based setting. Project page: https://dexhand.github.io/UniGraspTransformer.
title UniGraspTransformer: Simplified Policy Distillation for Scalable Dexterous Robotic Grasping
topic Robotics
url https://arxiv.org/abs/2412.02699