Scaling Dexterous Manipulation via Action Representation Pretraining
Organizations: Huazhong University of Science and Technology
Abstract
Large-scale vision, language, and video pretraining has substantially improved the perception and generalization capabilities of robotic manipulation models, but action generation for complex dexterous manipulation remains constrained by the scarcity of robot demonstrations. Limited robot demonstrations must be used to learn both action representations and task-conditioned action distributions, making it difficult to adequately learn action representations and, in turn, limiting the ability of downstream generative models to generate complex dexterous actions. To address this limitation, we propose ARPG (Action Representation Pretraining and Generation), which explicitly separates action representation learning from action generation: we first pretrain an action representation encoder on large-scale human manipulation data. We then freeze the encoder and train a compact generative backbone (130M parameters) to learn task-conditioned action distributions in the encoder's representation space. Experimental results show that ARPG achieves state-of-the-art performance on the DexJoCo benchmark and improves the average success rate on bimanual dexterous manipulation from 10.1% for to 44.7%. Scaling experiments further show that broader coverage of human manipulation behaviors effectively improves downstream performance. These results indicate that action representation pretraining provides an independent scaling dimension for improving complex dexterous manipulation capabilities.
Figures & tables
| Method | Inference cost | Success rate (%) | ||||
|---|---|---|---|---|---|---|
| Params (B) | VRAM (GB) | Latency (ms) | Single arm | Bimanual | Overall | |
| Single-task training | ||||||
| 3.35 | 9.05 | 84.9 | 69.7 | 35.3 | 54.1 | |
| GR00T N1.5 | 2.72 | 6.21 | 93.6 | 56.2 | 21.1 | 40.2 |
| ARPG | 0.46 | 2.63 | 50.8 | 75.2 | 44.5 | 61.3 |
| Multi-task training | ||||||
| (a) Action representation | (b) Aggregation depth | (c) Auxiliary objectives | |||
| Representation | Overall | Last- | Overall | Objective | Overall |
| Raw Action | 13.3 | 1 | 52.5 | w/o | 54.9 |
| Action-MAE | 36.4 | 2 | 57.8 | w/o | 56.0 |
| Action-iBOT | 30.5 | 3 | 51.6 | Full | 57.8 |
| Visual Alignment + iBOT | 45.6 | ||||
| Visual Alignment | 57.8 | ||||
Appendix figures & tables9 assets
Supplementary material from the paper’s appendix.
Appendix
| Hand | Action component | Index range | Dim. | Decoder output |
|---|---|---|---|---|
| Left | 15 local joint rotations (6D) | 90 | Yes | |
| Left | Palm rotation (6D) | 6 | Yes | |
| Left | Absolute wrist position | 3 | Yes | |
| Left | 21 wrist-relative keypoints | 63 | No | |
| Right | 15 local joint rotations (6D) | 90 | Yes | |
| Right | Palm rotation (6D) | 6 | Yes |
| Metric | 500 h | 3,000 h |
|---|---|---|
| Summed clip duration | 500 h 20 m 16 s | 3,088 h 16 m 59 s |
| Total samples | 1,801,216 | 11,117,819 |
| Number of tasks | 17,382 | 23,316 |
| Largest single-task share | 0.09% | 9.85% |
| Top-10 task share | 0.77% | 23.87% |
| Mean samples per task | 103.63 | 476.83 |
| Task | Task description | Demonstrations | Duration |
|---|---|---|---|
| Hammer Handover | Pick up, hand over, and place a hammer | 40 | 3:19:31 |
| Pouring Coffee Beans | Pick up a container and pour coffee beans | 40 | 2:15:03 |
| Cabinet Retrieval | Open a cabinet, retrieve the target, and close it | 40 | 1:47:31 |
| Total | — | 120 | 7:22:05 |
| Pretraining objective | Visual Alignment |
|---|---|
| Optimizer | AdamW |
| Batch size | 1,536 |
| Learning rate | |
| Minimum learning rate | |
| Weight decay | 0.05 |
| Warmup steps | 1,000 |
| Task | Stage 1 | Stage 2 | Stage 3 |
|---|---|---|---|
| Hammer Handover | Pick up the hammer | Complete the bimanual handover | Place the hammer at the target |
| Pouring Coffee Beans | Pick up the bean container | Successfully pour out the coffee beans | Completely pour out the coffee beans |
| Cabinet Retrieval | Open the cabinet door | Retrieve the target object | Close the cabinet door |
| Task | Single-task training | Multi-task training | |||
|---|---|---|---|---|---|
| GR00T N1.5 | ARPG | ARPG | |||
| Single-arm | |||||
| Hammer Nail | 84.7 5.0 | 67.3 4.2 | 72.0 9.2 | 74.7 2.3 | 52.7 5.0 |
| Click Mouse | 64.7 8.1 | 85.3 3.1 | 73.3 8.1 | 48.0 6.9 | 96.0 3.3 |
| Pick Bucket | 84.0 7.2 | 72.0 6.0 | 80.0 2.0 | 60.7 2.3 | 78.0 1.6 |
| Pinch Tongs | 24.0 6.9 | 12.7 2.3 | 72.7 13.0 | 44.0 15.9 | 50.0 1.6 |
Robotic Manipulation
Robotic manipulation studies how robots perceive, plan, and control physical interactions to grasp, move, assemble, and use objects.
See all 1,685 papers