PartiGen: Part-Aware Representations for Language-Conditioned Humanoid Motion Generation
Organizations: Fudan University · Shanghai Innovation Institute · Shanghai Jiaotong University · Zhejiang University · Tongji University · The Chinese University of Hong Kong · Humanoid Robot (Shanghai) Co., Ltd. · Beihang University
Abstract
Translating language into humanoid motion requires coordinating body parts, determining motion duration, and producing references suitable for robot execution. We introduce PartiGen, a part-aware latent diffusion framework that incorporates anatomical structure into a shared whole-body motion representation. Part-specific attention processes anatomically masked frame features, and its outputs are fused to combine local limb information with whole-body context. Conditioned on text and motion history, the model generates motion primitives autoregressively. Forward-kinematics supervision encourages kinematic consistency, while a learned end-of-sequence predictor selects a frame-level endpoint after generation. On HumanML3D retargeted to the Unitree G1, PartiGen achieves a segment-level FID of 3.165, a 16.8% reduction over TextOp, and improves R@1 by 1.4 percentage points. In the duration study with 16-frame primitives, endpoint prediction reduces mean absolute duration error from 12.471 to 3.080 frames compared with a fixed-length cutoff. We further evaluate robot execution using a pretrained, frozen GRIT tracker: 409 of 500 generated references (81.8%) pass the posture-based stability screen in MuJoCo. Following simulation screening and manual selection, 26 of 27 attempted motions are judged successful on the real G1.
Figures & tables
| Method | R@1 | R@2 | R@3 | FID | Div | mm-dist |
|---|---|---|---|---|---|---|
| Dataset | ||||||
| DART+GMR | ||||||
| MDM+GMR | ||||||
| AttT2M+GMR | ||||||
| TextOp | ||||||
| Ours |
| Method | R@1 | R@2 | R@3 | FID | Div |
|---|---|---|---|---|---|
| Compositional Dataset | |||||
| DART+GMR | |||||
| TextOp | |||||
| Ours |
| Method | Requests | Valid | Excluded | Stability passes | Stability rate ( ) | Global MPJPE (mm, ) | Root-relative MPJPE (mm, ) |
|---|---|---|---|---|---|---|---|
| PartiGen (Ours) | 500 | 496 | 4 | 409 | 81.8% | 252.402 / 108.871 | 57.995 / 39.765 |
| GT | 500 | 499 | 1 | 427 | 85.4% | 399.78 / 183.22 | 56.24 / 37.94 |
| DART+GMR | 500 | 494 | 6 | 395 | 79.0% | 288.35 / 241.03 | 58.32 / 40.07 |
| Random-mask | 500 | 494 | 6 | 401 | 80.2% | 283.90 / 157.29 | 56.38 / 38.25 |
| Variant | R@1 | R@2 | R@3 | FID | Div | PJ | Contact Acc |
|---|---|---|---|---|---|---|---|
| w/o Part Attention | |||||||
| Random-mask | |||||||
| Full |
| Variant | Duration Error | R@3 | FID | Div |
|---|---|---|---|---|
| Oracle Length | 6.900 0.045 | |||
| Fixed-Length | 12.471 3.596 | 0.584 0.005 | 4.317 0.021 | 6.760 0.038 |
| Full (16-frame primitives) | 3.080 0.385 | 0.599 0.013 | 3.207 0.033 |
Appendix figures & tables11 assets
Supplementary material from the paper’s appendix.
Appendix
| Head Index | Body Part | DOF Indices | DOF Count |
|---|---|---|---|
| Head 0 | Left Leg | [0, 1, 2, 3, 4, 5] | 6 |
| Head 1 | Right Leg | [6, 7, 8, 9, 10, 11] | 6 |
| Head 2 | Torso | [12, 13, 14] | 3 |
| Head 3 | Upper Body | [15, 16, 17, 18, 19, 20, 21, 22] | 8 |
| Partition | R@1 | R@2 | R@3 | FID | Div | PJ | Contact Acc |
|---|---|---|---|---|---|---|---|
| A | 0.279 | 0.363 | 0.588 | 3.701 | 7.726 | 0.038 | 0.870 |
| B | 0.299 | 0.370 | 0.601 | 3.287 | 7.812 | 0.021 | 0.898 |
| C | 0.265 | 0.353 | 0.569 | 3.612 | 7.776 | 0.024 | 0.871 |
| Channels | Dimension | Feature |
|---|---|---|
| 0 | 1 | |
| 1 | 1 | |
| 2 | 1 | |
| 3 | 1 | |
| 4 | 1 | : inter-frame yaw increment |
| 5–6 | 2 | Left and right foot contact indicators |
| Aggregate Term (Main Paper) | Internal Component | Description | Sub-weight ( ) |
| rec | Primary feature reconstruction | 1.0 | |
| kl | Variational bottleneck regularization | ||
| ( ) | body_trans | FK-based body position error | 0.05 |
| body_rot | FK-based body rotation error | 0.01 | |
| dof_pos | Joint angle precision | 0.03 | |
| dof_vel | Temporal joint velocity smoothness |
| Category | Parameter | Value |
| Transformer Backbone | Number of Layers | 8 |
| Hidden Dimension ( ) | 512 | |
| FFN Intermediate Dimension ( ) | 1024 | |
| Activation Function | GELU | |
| Attention Mechanism | Number of Heads ( ) | 4 |
| Dimension per Head ( ) | 128 |
| Kinematic Category | Instruction Prompt |
|---|---|
| Continuous Locomotion | A person is walking and pointing forward. |
| A person is walking sideways and crossing arms. | |
| A person is walking backward and clapping hands. | |
| A person is jogging and putting hands on hips. | |
| A person is running and waving hands. | |
| A person is running and touching the head. |
| Region | MuJoCo body names |
|---|---|
| Root | pelvis |
| Left leg | left_hip_pitch_link , left_knee_link , left_ankle_roll_link , left_toe_link |
| Right leg | right_hip_pitch_link , right_knee_link , right_ankle_roll_link , right_toe_link |
| Torso/head | torso_link , head_mimic |
| Left arm | left_shoulder_pitch_link , left_elbow_link , left_wrist_roll_link , left_hand_mimic |
| Right arm | right_shoulder_pitch_link , right_elbow_link , right_wrist_roll_link , right_hand_mimic |
| Method | Requests | Valid | Excluded | Posture passes | Pass rate ( ) | Global MPJPE (mm, ) | Root-relative MPJPE (mm, ) |
|---|---|---|---|---|---|---|---|
| Full | 500 | 496 | 4 | 409 | 81.8% | 252.402 / 108.871 | 57.995 / 39.765 |
| w/o FK | 500 | 495 | 5 | 464 | 92.8% | 275.61 / 112.37 | 60.01 / 41.82 |
| Method | Stable motions | Global MPJPE | Root-relative MPJPE |
|---|---|---|---|
| PartiGen | 409 | 198.00 / 97.08 | 42.84 / 37.98 |
| GT | 427 | 297.81 / 161.34 | 44.26 / 36.04 |
| DART+GMR | 395 | 247.33 / 145.29 | 44.37 / 36.64 |
| w/o FK | 464 | 252.47 / 140.92 | 45.20 / 37.53 |
| Category | Selected | Attempted | Successful | Failed | Unattempted |
|---|---|---|---|---|---|
| Upper-body | 10 | 10 | 10 | 0 | 0 |
| Lower-body | 10 | 7 | 7 | 0 | 3 |
| Long-sequence | 5 | 5 | 5 | 0 | 0 |
| Composite | 5 | 5 | 4 | 1 | 0 |
| Total | 30 | 27 | 26 | 1 | 3 |