PFM-HR: Pose Flow Matching for Humanoid Robots
Abstract
Motion priors improve reinforcement learning for physics-based humanoid tracking, but temporal priors require ordered motion clips, while pose priors provide limited guidance for policy-induced pose transitions. We present Pose Flow Matching for Humanoid Robots (PFM-HR), a reusable flow matching prior trained directly on large-scale unordered pose data. PFM-HR introduces the Pose Geometry Score (PGS), which quantifies how joint coordinate changes during rollouts align with the local geometry of pose variation captured by the prior. Using PGS to modulate the tracking reward guides policy exploration toward structured pose changes while keeping the prior frozen across tracking tasks. Experiments demonstrate that PFM-HR improves both single-motion and general motion tracking, especially for highly dynamic motions.
I Introduction
Humanoid robots hold considerable promise for operating in human-centered environments, where their humanlike morphology allows them to use the same spaces, tools, and interaction patterns as people [1, 35, 39, 7]. Recent advances in reinforcement learning (RL) have enabled increasingly capable humanoid locomotion and control. Yet dynamic feasibility alone does not specify how the joints should vary together to produce coordinated humanlike motion. Even with task and tracking rewards, the policy receives limited guidance on such joint dependencies and must discover them through exploration in a high-dimensional control space. This burden becomes particularly severe for dynamic motions that require precise whole-body coordination.
Data-driven motion priors address this problem by guiding policy optimization toward behaviors supported by demonstrations. DeepMimic tracks reference trajectories to reproduce complex skills [23], while adversarial methods such as AMP and ADD learn discriminator-based rewards that encourage realistic motion and reduce manual reward engineering [25, 4, 38]. Although effective, these discriminator-based priors are commonly trained together with the control policy, which limits their reuse across tasks. Recent frozen priors improve modularity but introduce a different trade-off. PDF-HR learns a reusable pose-distance field from retargeted humanoid data and improves downstream tracking [5], but it requires precomputed pose–distance supervision and scores each pose without considering the relative joint changes that produced it. SMP instead studies reusable score-matching motion priors in a different control formulation, repurposing a pretrained motion diffusion model as a reward through score distillation [20]. This captures temporal motion structure, but requires ordered motion clips and incurs repeated score-model evaluations during RL. This leaves a gap between efficient pose-wise priors, which do not evaluate how joints change together, and temporal priors, which require ordered data and heavier online inference.
We address this gap with Pose Flow Matching for Humanoid Robots (PFM-HR), a reusable Flow Matching prior trained directly on individual humanoid poses, with the training corpus treated as an unordered collection [11]. PFM-HR requires neither pose–distance supervision nor temporally segmented motion clips. It is pretrained independently of the control policy, remains frozen during downstream RL, and can be extended through continued training as additional pose data become available.
Our key insight is that a denoiser trained only on marginal pose samples still captures local pose geometry. Under the Gaussian Flow Matching path, the Jacobian of the population-optimal denoising map is proportional to the conditional covariance of clean poses [19, 21]. Its directional response therefore reflects conditional modes of joint co-variation.
PFM-HR applies this Jacobian to the normalized joint-coordinate change between consecutive rollout poses and defines the squared Jacobian–vector response as the Pose Geometry Score (PGS). A high PGS indicates that the relative joint changes align with co-variation modes strongly represented by the pose prior. After reference-specific calibration, PGS modulates the tracking reward to reduce learning signals from weakly supported change patterns.
Our contributions are threefold:
-
•
We introduce a scalable, reusable Flow Matching pose prior trained on unordered humanoid poses. We demonstrate pretraining on 60 million poses and efficient expansion through continued training.
-
•
We establish a conditional covariance interpretation of the denoiser’s response and formulate the Pose Geometry Score for measuring response along the joint coordinate change patterns of policy generated pose updates.
-
•
We integrate PGS into an efficient, reference calibrated reward modulation mechanism for humanoid tracking. Experiments on single-trajectory tracking and general motion tracking benchmarks demonstrate improved sample efficiency and tracking accuracy over baselines.
II Related Work
Physics-based Humanoid Motion Tracking. Deep reinforcement learning has become a central paradigm for physics-based humanoid motion tracking. DeepMimic [23] demonstrated that physically simulated characters can reproduce highly dynamic skills by tracking motion-capture references. Subsequent work learns reusable motion representations to improve control and downstream task learning. These include VAE-based controllers [33, 31, 28], discrete representations [34, 41], and reusable skill spaces such as ASE, CALM, and PULSE [24, 29, 16]. Adversarial methods instead derive motion-style rewards directly from demonstrations. AMP [25, 4] replaces hand-designed style objectives with a discriminator, while ADD [38] extends this formulation by balancing multiple tracking objectives.
Large-scale trackers further improve motion coverage and robustness. PHC [15] scales physics-based imitation to thousands of motion clips and supports recovery from large deviations. More recent systems, including GMT [3], BeyondMimic [10], and SONIC [17], target diverse and previously unseen motions. Heracles [27] further connects generative motion synthesis with physics-based tracking through a state-conditioned middleware that adapts high-level references under large state deviations. These methods substantially broaden the capabilities of humanoid controllers, but their motion knowledge is typically embedded in a pretrained policy or learned jointly with the controller. Scaling them therefore requires large motion datasets [18, 6, 2] and extensive policy training. PFM-HR instead studies a frozen, controller-independent prior that can be attached to task-specific tracking policies as a dense reward signal.
Human and Humanoid Motion Priors. Human pose and motion priors are widely used to regularize motion recovery and synthesis [14, 13, 37]. PoseNDF [30] models pose plausibility with an implicit neural distance field. NRDF [8] improves its geometric consistency, while NRMF [36] extends distance-based priors to continuous motion. PDF-HR [5] adapts neural pose distance fields to humanoid control and demonstrates that a frozen pose prior can improve downstream tracking. These methods provide reusable state-level regularization, but constructing distance supervision requires an additional sampling stage.
Generative models provide another route to reusable priors. SMP [20] converts a pretrained motion diffusion model into an RL reward through Score Distillation Sampling, separating prior learning from policy optimization while retaining temporal motion information. This motion-level representation captures dependencies across consecutive states, but requires ordered clips. In contrast, priors trained on individual poses can use unordered pose collections and avoid modeling fixed temporal windows. However, they typically evaluate each pose independently and do not explicitly characterize how joint coordinates vary together around the current pose.
This gap motivates a reusable prior that can be trained on unordered pose collections while still revealing how joint coordinates vary together. Such a prior would retain the efficiency of pose based modeling while providing richer guidance than an independent pose plausibility score.
III Methodology
III-A Problem Formulation
Robot pose representation. We represent a humanoid pose by its actuated joint configuration. Let the robot have actuated degrees of freedom. At control step , the pose is written as [5]
| (1) |
where denotes the -th actuated joint coordinate and is bounded by the robot joint limits. We use normalized joint coordinates as the input representation for the pose prior. For brevity, denotes the normalized pose hereafter.
During a rollout, the policy generates a pose sequence . We characterize the rollout update ending at step by the finite-difference joint-coordinate velocity
| (2) |
where is the control interval. While identifies the location of the rollout in pose space, records the joint-coordinate changes induced by the policy over one control interval. This pattern describes which joint coordinates vary together, including their relative magnitudes and signs.
RL-based motion tracking. We formulate humanoid motion tracking as a finite-horizon discrete-time Markov decision process . At each control step , the policy observes and samples an action . The simulator then evolves according to . The policy is trained to maximize
| (3) |
In humanoid motion tracking, the reward commonly combines task and tracking terms,
| (4) |
where measures task progress, measures agreement with the reference motion, and and are scalar weights. Optimizing these task-specific objectives alone provides no explicit guidance from the statistical dependencies among joint coordinates in human pose data.
PFM-HR augments the tracking reward using a frozen Flow Matching pose prior trained on unordered poses. At each rollout step, PGS applies the denoiser Jacobian at to the normalized joint-coordinate change, measuring its alignment with the joint co-variation structure encoded by the pose distribution.
III-B Pose Prior Learning with Flow Matching
We pretrain an unconditional Flow Matching model on individual poses sampled from a large-scale motion corpus. The training data are treated as an unordered pose collection, from which the model learns the marginal pose distribution . Let be a clean pose and an independent Gaussian sample. Both training poses and rollout poses use the normalized representation defined in Section III-A. We sample from a logit-normal distribution and construct the linear Flow Matching path [11, 12]
| (5) |
This path interpolates between the Gaussian source distribution at and the pose distribution at . Its conditional velocity is
| (6) |
Following JiT [9], we parameterize the velocity field through a clean-pose predictor . The predicted velocity is
| (7) |
where stabilizes the denominator near . We optimize the Flow Matching objective
| (8) |
For , the velocity residual satisfies . Therefore, at any fixed timestep in this regime, minimizing Eq. (8) is equivalent to clean pose regression, whose population optimal predictor is the conditional mean
| (9) |
We denote the learned approximation to this conditional mean by
| (10) |
During policy learning, remains frozen and is queried at , where .
III-C Pose Geometry Score
The Pose Geometry Score measures whether the relative joint changes produced by the policy agree with the conditional co-variation structure encoded by the pose prior.
Denoiser-induced pose geometry. For fixed , the population-optimal denoiser in Eq. (9) satisfies [19]
| (11) |
where . In particular, its off-diagonal entries describe conditional co-variation between joint coordinates. For the learned denoiser, we define and
| (12) |
Pose Geometry Score. At rollout step , we construct a query using the same corruption parameterization as in pretraining:
| (13) |
where is fixed during policy learning. We then set and .
The joint-coordinate change pattern is represented by
| (14) |
which retains the relative changes across joints while reducing sensitivity to their overall magnitude. We define the Pose Geometry Score (PGS) as
| (15) |
The score is computed by a Jacobian–vector product without explicitly constructing the Jacobian or geometric tensor.
At the population optimum, with ,
| (16) |
This expression weights the projection of onto each conditional co-variation mode by its squared variance. A high PGS therefore indicates a joint-change pattern strongly supported by the pose prior. After calibration, PGS increases the reward contribution of such coordinated updates and reduces that of weakly supported patterns, biasing policy optimization toward coordinated humanoid poses. The detailed covariance derivation is provided in the supplementary material.
III-D Score Calibration and Reward Modulation
Evaluation-Timestep Selection. The evaluation timestep determines the noise scale at which the prior is queried and therefore affects the sensitivity of pose geometry score to pose-change patterns. We examine this sensitivity under the general motion tracking setting. For each , we train a single policy on the same 34-sequence subset of LaFAN1 [6], excluding the six FallAndGetUp sequences. Each policy is trained for 12 billion samples and evaluated with tracking horizons of 10, 20, and 30 seconds, while all other settings are held fixed. Figure 3 shows that intermediate timesteps provide faster convergence and lower position errors. Strong corruption at small reduces pose specificity, whereas near the denoising map approaches the identity and becomes less discriminative. We therefore use throughout the remaining experiments. Additional timestep selection details are included in the supplementary material.
Reference-calibrated reward modulation. For each transition in a reference motion , we define and obtain using Eq. (14). Before policy training, we compute PGS for all reference transitions and construct the empirical CDF . During rollout, the score is mapped to its reference percentile and converted into
| (17) |
where , , and . The final reward is
| (18) |
The tracking reward is unchanged above and maximally attenuated below .
IV Experiments
IV-A Experimental Setup
Baselines and prior training. Across both tracking settings, we compare vanilla ADD [38], ADD w/ PDF-HR [5], and ADD w/ PFM-HR under the same ADD-based motion tracking backbone; score-matching prior SMP [20] uses a different control formulation and is not directly compatible with this plug-in comparison. For a controlled comparison, PDF-HR and PFM-HR are pretrained on the same unmirrored BONES-SEED [2] pose corpus. PDF-HR constructs pose–distance pairs using its official implementation.
Evaluation metrics. We evaluate sample efficiency and tracking accuracy. Policies are evaluated every 100 training iterations on 4096 test episodes. Success rate is defined as the mean fraction of reference trajectory tracked continuously. Sample efficiency is measured by the number of training samples required to first reach an success rate. We additionally measure position and rotation errors:
| (19) | ||||
| (20) |
where is the number of non-root joints. Joint errors are measured in the robot-local frame, while root errors are measured in the global frame. Lower values indicate better tracking accuracy. Due to space constraints, additional metrics and rotation errors are reported in the supplementary material.
Implementation details. All policies are trained on 8 NVIDIA RTX 4090 GPUs using 4096 parallel environments and the default MimicKit configuration. PFM-HR uses a residual MLP with 10 blocks and a hidden dimension of 1024. The Flow Matching timestep is injected through adaLN-Zero conditioning [22], and the pretrained model remains frozen during policy training. Further implementation details for pretraining and policy learning are provided in the supplementary material.
IV-B Single-trajectory Motion Tracking
Experiment settings. For single-trajectory tracking, we compare vanilla ADD [38] with two prior-augmented variants, ADD w/ PDF-HR [5] and ADD w/ PFM-HR, on nine MimicKit tasks [26]: Walk, Run, Jump, Spinkick, Cartwheel, Backflip, Sideflip, Double Kong, and Speed Vault.
Results. Table II reports sample efficiency and positional tracking error. ADD w/ PDF-HR is more sample-efficient on Walk, Run, and Spinkick, while ADD w/ PFM-HR performs best on the other six tasks. Both prior-based methods solve Backflip and Double Kong, where vanilla ADD fails, with PFM-HR converging faster on both. ADD w/ PFM-HR also achieves the lowest or tied-lowest positional error on eight tasks, although the margins are generally small. Overall, PFM-HR provides the best balance between convergence and tracking accuracy, particularly on dynamic motions. By evaluating the prior response along rollout pose changes, PFM-HR guides learning toward plausible joint-coordinate transitions. Qualitative comparisons are shown in Fig. 5.
Real-world deployment. We use the BeyondMimic training and deployment pipeline [10], adding PFM-HR solely as a frozen prior during simulation training while leaving the deployment pipeline unchanged.
| Skill | BM | BM w/ PDF-HR | BM w/ PFM-HR |
|---|---|---|---|
| Spinkick | |||
| Kick combo | |||
| Cartwheel | |||
| Backflip |
We evaluate four dynamic skills: spinkick, kick combo, cartwheel, and backflip. In Table I, BM denotes the original BeyondMimic method without an auxiliary pose prior, while BM w/ PDF-HR and BM w/ PFM-HR augment BM with PDF-HR and our PFM-HR, respectively. BM w/ PFM-HR requires the fewest simulation samples to reach across all skills, reducing the requirements by – relative to BM. The successful executions in Fig. 4 demonstrate PFM-HR’s compatibility with real-world deployment without additional deployment-time computation.
IV-C General Motion Tracking
Experiment settings. For general motion tracking, we train a single policy on a subset of LaFAN1 [6]. The subset contains 34 sequences, excluding six FallAndGetUp sequences. Each policy is trained for 12 billion samples and evaluated with tracking horizons of 10, 20, and 30 seconds.
Results. Table II reports general motion tracking errors across different episode lengths. Under the same ADD backbone and training budget, PFM-HR achieves the lowest position and rotation errors at all three horizons. Averaged across horizons, it reduces position error by over ADD and over PDF-HR, while reducing rotation error by and , respectively. In contrast, PDF-HR yields errors comparable to or higher than vanilla ADD.
IV-D Ablation Study
Impact of Prior-Data Scale and Coverage. We train pose priors on three nested BONES-SEED subsets [2], . The M subset is balanced across Locomotion, Dances, Gaming, Sport, and Other; the M subset adds stratified samples from the same categories; and the M subset contains the full corpus. All model and policy settings are fixed, and the reference-specific PGS thresholds are recomputed for each prior. Figure 2 shows that larger pretraining sets generally improve both locomotion and acrobatic tracking, with the M prior achieving the strongest overall performance. Because scale and coverage increase together, the results reflect their combined effect.
Extending the M prior to M requires GPU-hours, compared with GPU-hours for training from scratch. At the same scale, generating the pose-distance supervision for PDF-HR [5] requires over GPU-hours in our implementation, excluding network training.
| Single-Trajectory Tracking | ||||||
|---|---|---|---|---|---|---|
| Skill | Samples () [M] () | Position Error [m] () | ||||
| ADD | ADD w/ PDF-HR | ADD w/ PFM-HR | ADD | ADD w/ PDF-HR | ADD w/ PFM-HR | |
| Walk | ||||||
| Run | ||||||
| Jump | ||||||
| Spinkick | ||||||
| Cartwheel | ||||||
| Backflip | Failed | Failed | ||||
| Sideflip | ||||||
| Speed Vault | ||||||
| Double Kong | Failed | Failed | ||||
| General Motion Tracking Across Episode Lengths | ||||||
| Episode | Position Error [m] () | Rotation Error [rad] () | ||||
| Length [s] | ADD | ADD w/ PDF-HR | ADD w/ PFM-HR | ADD | ADD w/ PDF-HR | ADD w/ PFM-HR |
| 10 | ||||||
| 20 | ||||||
| 30 | ||||||
| Skill | Samples () [M] () | Position Error [m] () | ||||
|---|---|---|---|---|---|---|
| -Pred | FM-Recon | Ours | -Pred | FM-Recon | Ours | |
| Walk | ||||||
| Run | ||||||
| Backflip | ||||||
| Sideflip | ||||||
Reward Formulation Comparison. We compare ADD, FM-Recon, and PFM-HR to separate the contribution of the learned prior from its reward formulation. ADD provides a prior-free baseline. FM-Recon uses the same frozen denoiser to score individual poses through reconstruction [32, 40], whereas PFM-HR evaluates Jacobian response along rollout pose changes. For rollout pose , FM-Recon evaluates three noise levels , whose signal-to-noise ratios follow the noise-level ensemble used by SMP. At each , it samples and computes
| (21) |
To account for scale differences across noise levels, each error is converted to its reverse percentile under the corresponding reference distribution. The resulting percentiles are averaged and mapped to the same reward range as PGS, with all other settings unchanged.
Comparing ADD with FM-Recon evaluates the contribution of the frozen prior, while comparing FM-Recon with PFM-HR assesses multi-level pose reconstruction against directional Jacobian scoring. Table III shows that PFM-HR provides larger improvements on dynamic motions. With a batch size of 4096 on an NVIDIA GeForce RTX 4090 GPU, three denoiser evaluations require ms, whereas one Jacobian–vector product requires ms.
Prediction Parameterization Comparison. We compare -prediction and -prediction while keeping the training data, architecture, Flow Matching path, and policy-training configuration fixed. For -prediction, the clean-pose map used by PGS is recovered as
| (22) |
We then construct PGS from its input Jacobian using the same evaluation timestep, calibration, and reward modulation, thereby isolating the effect of the prediction target. As shown in Table III, both parameterizations perform similarly on basic locomotion, whereas -prediction performs better on dynamic motions, particularly Backflip. Because the two variants share the same architecture and prior-query cost, the results suggest that direct clean-pose prediction provides a more effective Jacobian response for a finite-capacity model.
V Conclusion
We presented PFM-HR, a reusable Flow Matching pose prior for physics-based humanoid motion tracking that can be trained directly on large-scale unordered pose data, without pose-distance supervision or temporally segmented motion clips. Its Pose Geometry Score uses the denoiser Jacobian to evaluate whether policy-induced pose changes align with the joint co-variation patterns captured by the prior. This score can be computed efficiently through a Jacobian–vector product and used to modulate tracking rewards while keeping the prior frozen. By separating prior learning from policy optimization, PFM-HR allows a single pretrained model to provide geometry-aware guidance across different tracking tasks. Experiments demonstrate improved sample efficiency and tracking accuracy, particularly for dynamic and acrobatic motions, as well as compatibility with real-world humanoid deployment. These results show that the local geometry learned from unordered poses can provide useful control guidance even without explicit temporal supervision.
VI Limitations
PFM-HR captures local co-variation in the marginal pose distribution rather than temporal dynamics. Because the training poses are unordered and PGS is invariant to direction reversal at a fixed query pose, it cannot assess the likelihood, direction, or ordering of temporal transitions. Consequently, opposite transitions may receive similar geometric evaluations even when only one is consistent with the intended motion. The reliability of PGS also depends on the coverage of the pose dataset, and its guidance may be less informative for poses far from the training distribution. Future work will explore sign-sensitive scores and temporally conditioned priors while preserving the efficiency and reusability of PFM-HR.
References
- [1] (2025) Visual imitation enables contextual humanoid control. In Proceedings of the Conference on Robot Learning (CoRL), Cited by: §I.
- [2] (2026) Ai datasets for machine learning and motion capture. Note: https://bones.studio/ai-datasets/Accessed: 2026-05-06 Cited by: §II, §IV-A, §IV-D.
- [3] (2025) GMT: general motion tracking for humanoid whole-body control. arXiv:2506.14770. Cited by: §II.
- [4] (2022) Adversarial motion priors make good substitutes for complex reward functions. In 2022 IEEE/RSJ International Conference on Intelligent Robots and Systems (IROS), pp. 25–32. Cited by: §I, §II.
- [5] (2026) PDF-hr: pose distance fields for humanoid robots. arXiv preprint arXiv:2602.04851. Cited by: §I, §II, §III-A, §IV-A, §IV-B, §IV-D, TABLE II.
- [6] (2020) Robust motion in-betweening. ACM Transactions on Graphics (TOG) 39 (4), pp. 60–1. Cited by: §II, §III-D, §IV-C.
- [7] (2026) ULTRA: unified multimodal control for autonomous humanoid whole-body loco-manipulation. arXiv preprint arXiv:2603.03279. Cited by: §I.
- [8] (2024) NRDF: neural riemannian distance fields for learning articulated pose priors. In Conference on Computer Vision and Pattern Recognition (CVPR), Cited by: §II.
- [9] (2025) Back to basics: let denoising generative models denoise. arXiv preprint arXiv:2511.13720. Cited by: §III-B.
- [10] (2025) BeyondMimic: from motion tracking to versatile humanoid control via guided diffusion. External Links: 2508.08241, Link Cited by: §II, §IV-B.
- [11] (2023) Flow matching for generative modeling. In The Eleventh International Conference on Learning Representations, External Links: Link Cited by: §I, §III-B.
- [12] (2022) Flow straight and fast: learning to generate and transfer data with rectified flow. arXiv preprint arXiv:2209.03003. Cited by: §III-B.
- [13] (2025) DPoser-x: diffusion model as robust 3d whole-body human pose prior. arXiv preprint arXiv:2508.00599. Cited by: §II.
- [14] (2024) DPoser: diffusion model as robust 3d human pose prior. External Links: 2312.05541, Link Cited by: §II.
- [15] (2023) Perpetual humanoid control for real-time simulated avatars. In Proceedings of the IEEE/CVF International Conference on Computer Vision, pp. 10895–10904. Cited by: §II.
- [16] (2024) Universal humanoid motion representations for physics-based control. In International Conference on Learning Representations, Vol. 2024, pp. 56766–56782. Cited by: §II.
- [17] (2025) SONIC: supersizing motion tracking for natural humanoid whole-body control. arXiv preprint arXiv:2511.07820. Cited by: §II.
- [18] (2019) AMASS: archive of motion capture as surface shapes. In Proceedings of the IEEE/CVF international conference on computer vision, pp. 5442–5451. Cited by: §II.
- [19] (2023) On the posterior distribution in denoising: application to uncertainty quantification. arXiv preprint arXiv:2309.13598. Cited by: §I, §III-C.
- [20] (2026) SMP: reusable score-matching motion priors for physics-based character control. ACM Transactions on Graphics (Proceedings of SIGGRAPH 2026). Cited by: §I, §II, §IV-A.
- [21] (2026) PoseD-flow: versatile and guided flow matching model of human pose. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 21165–21175. Cited by: §I.
- [22] (2023) Scalable diffusion models with transformers. In Proceedings of the IEEE/CVF international conference on computer vision, pp. 4195–4205. Cited by: §IV-A.
- [23] (2018) Deepmimic: example-guided deep reinforcement learning of physics-based character skills. ACM Transactions On Graphics (TOG) 37 (4), pp. 1–14. Cited by: §I, §II.
- [24] (2022) Ase: large-scale reusable adversarial skill embeddings for physically simulated characters. ACM Transactions On Graphics (TOG) 41 (4), pp. 1–17. Cited by: §II.
- [25] (2021) Amp: adversarial motion priors for stylized physics-based character control. ACM Transactions on Graphics (ToG) 40 (4), pp. 1–20. Cited by: §I, §II.
- [26] (2025) MimicKit: a reinforcement learning framework for motion imitation and control. arXiv preprint arXiv:2510.13794. Cited by: §IV-B.
- [27] (2026) Heracles: bridging precise tracking and generative synthesis for general humanoid control. arXiv preprint arXiv:2603.27756. Cited by: §II.
- [28] (2024) MaskedMimic: unified physics-based character control through masked motion inpainting. ACM Transactions on Graphics (TOG). Cited by: §II.
- [29] (2023) Calm: conditional adversarial latent models for directable virtual characters. In ACM SIGGRAPH 2023 conference proceedings, pp. 1–9. Cited by: §II.
- [30] (2022) Pose-ndf: modeling human pose manifolds with neural distance fields. In European Conference on Computer Vision, pp. 572–589. Cited by: §II.
- [31] (2022) Physics-based character controllers using conditional vaes. ACM Trans. Graph. 41 (4). External Links: Link Cited by: §II.
- [32] (2025) Text-to-image rectified flow as plug-and-play priors. In International Conference on Learning Representations, Vol. 2025, pp. 13896–13920. Cited by: §IV-D.
- [33] (2022) ControlVAE: Model-Based Learning of Generative Controllers for Physics-Based Characters. ACM Transactions on Graphics 41 (6), pp. 183:1–183:16. External Links: ISSN 0730-0301 Cited by: §II.
- [34] (2024) Moconvq: unified physics-based motion control via scalable discrete representations. ACM Transactions on Graphics (TOG) 43 (4), pp. 1–21. Cited by: §II.
- [35] (2025) VisualMimic: visual humanoid loco-manipulation via motion tracking and generation. arXiv preprint arXiv:2509.20322. Cited by: §I.
- [36] (2025) Geometric neural distance fields for learning human motion priors. arXiv preprint arXiv:2509.09667. Cited by: §II.
- [37] (2024) RoHM: robust human motion reconstruction via diffusion. In CVPR, Cited by: §II.
- [38] (2025) Physics-based motion imitation with adversarial differential discriminators. In Proceedings of the SIGGRAPH Asia 2025 Conference Papers, pp. 1–12. Cited by: §I, §II, §IV-A, §IV-B, TABLE II.
- [39] (2025) ResMimic: from general motion tracking to humanoid whole-body loco-manipulation via residual learning. External Links: 2510.05070, Link Cited by: §I.
- [40] (2025) Score distillation of flow matching models. arXiv preprint arXiv:2509.25127. Cited by: §IV-D.
- [41] (2023) Neural categorical priors for physics-based character control. ACM Transactions on Graphics (TOG) 42 (6), pp. 1–16. Cited by: §II.
Supplementary Material
Appendix A Denoiser-Induced Pose Geometry
This section establishes the connection between the population-optimal clean-pose predictor and the conditional covariance of clean poses, and then derives the Pose Geometry Score (PGS). Fix and let , where is a small constant (we use ) that clamps the velocity denominator near . We assume that the clean-pose distribution has a finite second moment and satisfies the regularity conditions required to exchange differentiation and integration below. Let denote the dimension of the pose representation. In this work, we use for the Unitree G1 humanoid robot, matching the normalized pose representation of the main paper.
A-A Population-Optimal Clean-Pose Predictor
Let and be independent, where denotes the clean-pose distribution (a probability measure on ). Under the linear Flow Matching path, with and held fixed along the path,
| (23) |
A clean-pose predictor induces the velocity field
| (24) |
Therefore, at a fixed timestep , the Flow Matching objective can be written as
| (25) |
The derivation below considers the regime , which is also the regime used for PGS evaluation. In this regime, the denominator clamp in the implemented velocity parameterization is inactive and the velocity residual is exactly proportional to the clean-pose regression residual. Because depends only on , it does not alter the pointwise population minimizer at that timestep. Thus, over the unrestricted class of square-integrable predictors, a population-optimal clean-pose predictor is
| (26) |
defined -almost everywhere, where is the marginal density of .
A-B Jacobian–Conditional Covariance Identity
The Gaussian likelihood induced by the path in Eq. (23), together with its score, is
| (27) | ||||
| (28) |
Let denote the clean-pose probability measure, i.e., the measure underlying introduced above. The conditional mean can be expressed as
| (29) |
Differentiating this ratio gives, for coordinates ,
| (30) |
Consequently,
| (31) |
where . This is the Jacobian–conditional covariance identity used in Eq. (11) of the main paper. In particular, the off-diagonal entries of describe posterior co-variation between pose coordinates at the queried noisy pose.
A-C Induced Positive-Semidefinite Geometry
For a differentiable learned predictor , define
| (32) | ||||
| (33) |
For any ,
| (34) |
Thus, defines a positive-semidefinite pullback quadratic form, possibly degenerate, which we refer to as the denoiser-induced local pose geometry. This establishes Eq. (12) of the main paper. At the population optimum, Eq. (31) and the symmetry of the conditional covariance give
| (35) |
where denotes the matrix square .
A-D Pose Geometry Score
At rollout step , the finite-difference joint-coordinate velocity and its stabilized direction are
| (36) | ||||
| (37) |
Here, denotes a small numerical stabilization constant (we use ; this is the of the main paper) that prevents division by near-zero values, and denotes the normalized joint-coordinate pose representation used in the main paper. For a corruption draw , let
| (38) | ||||
| (39) |
where . The PGS is then
| (40) |
This recovers Eq. (15) of the main paper. The product can be evaluated directly with one Jacobian–vector product, without explicitly constructing either or .
To obtain the population interpretation, suppose that the learned Jacobian agrees with the population-optimal Jacobian at the queried point, and define
| (41) |
Equation (31) then gives
| (42) |
which is Eq. (16) of the main paper.
Because is symmetric and positive semidefinite, it admits the eigendecomposition
| (43) |
Writing the stabilized rollout direction in this eigenbasis as
| (44) |
we obtain
| (45) |
Thus, at a fixed corruption timestep, the population-optimal PGS is large when the stabilized rollout direction has substantial components along posterior pose-variation modes with large conditional covariance eigenvalues. This interpretation concerns the local co-variation structure of the marginal pose distribution and does not require temporally ordered prior-training data. For a learned predictor, the interpretation is approximate to the extent that its input Jacobian agrees with the population-optimal Jacobian at the queried point.
Appendix B Robustness of Online PGS Evaluation
PGS is evaluated online using a stochastically corrupted pose query. Given a transition , we corrupt the pose query by interpolating toward Gaussian noise at the evaluation timestep :
| (46) |
and obtain the score
| (47) |
We use the same frozen pose prior as in the main experiments and evaluate on transitions sampled from 100 motions randomly selected from the BONES-SEED dataset; all reported metrics are averaged over these transitions. We note that these transitions are drawn from the pose prior’s training corpus; the study measures the internal consistency of the learned PGS under pose-change perturbations rather than its generalization to held-out poses. The remainder of this section is organized as follows. We first select the evaluation timestep by measuring how well PGS discriminates coordinated joint changes from structurally perturbed ones across timesteps. We then verify that, at the selected timestep, the single-sample estimate remains consistent under stochastic corruption. Finally, we show that PGS is insensitive to the overall magnitude of a pose change while remaining sensitive to the inter-joint coordination pattern.
B-A Effect of the Evaluation Timestep
The evaluation timestep determines the corruption scale at which the frozen pose prior is queried, and it may consequently affect the ability of PGS to distinguish coordinated joint changes from structurally perturbed ones. To study this, we measure the structural discriminability of PGS across
| (48) |
For every transition, we retain the original pose and the joint-change direction , and construct perturbed change directions that preserve the pose query. To measure how the structural discrimination varies with the amount of disruption, we sweep the severity of each perturbation:
| Rank corr. | Percentile MAE | Region agree. | Reward MAE | Latency (ms) | |
|---|---|---|---|---|---|
| 1 | 0.976 | 0.013 | 97.9% | 0.010 | 0.750 |
| 2 | 0.981 | 0.011 | 98.4% | 0.007 | 1.308 |
| 4 | 0.990 | 0.009 | 98.9% | 0.004 | 2.847 |
| 8 | 0.996 | 0.005 | 99.3% | 0.002 | 5.991 |
| Ref. () | 1.000 | 0.000 | 100.0% | 0.000 | 97.112 |
Angular perturbation. The change direction is rotated by within the plane spanned by and a unit vector sampled uniformly from the orthogonal complement of :
| (49) |
Which rotates the change direction while approximately preserving its norm (the stabilized direction deviates from unit norm only by the small stabilization term of Eq. (37)).
Joint permutation. A random subset comprising a fraction of the joint-coordinate blocks is selected and permuted, disrupting the relative change pattern while preserving the set of local changes.
Independent sign flip. A random subset comprising a fraction of the joint-coordinate blocks is selected, and the change directions of the selected blocks are reversed. For , this disrupts the relative change pattern; at it is a global sign flip, which leaves the quadratic PGS invariant at a fixed query pose.
For each transition, the original and perturbed directions are evaluated using the corrupted pose query of Eq. (46) and the Gaussian noise sample. The corrupted query and the Gaussian noise sample are held fixed across the original and perturbed evaluations, so their score difference is caused only by the change in the inter-joint structure.
Timestep-specific calibration. Raw PGS values cannot be compared directly across evaluation timesteps because the scale of the denoiser Jacobian depends on . We therefore independently recompute the reference PGS distribution and empirical CDF for every timestep, evaluating each reference PGS with a single corruption sample from Eq. (46), consistent with the online estimator.
For the original transition, we compute
| (50) |
and for a perturbed direction ,
| (51) |
We define the structural discrimination score for perturbation at severity level as the average decrease in the calibrated percentile:
| (52) |
Because each perturbation is swept over a set of severity levels (the rotation angle for the angular perturbation and the joint fraction for joint permutation and sign flip), we report the severity-averaged score
| (53) |
A larger positive value indicates that PGS assigns substantially lower reference-relative scores after the joint-change structure is disrupted and therefore provides stronger structural discrimination. Values close to zero indicate that PGS responds similarly to the original and perturbed change patterns.
| Angular | Permutation | Sign flip | Average | |
|---|---|---|---|---|
| 0.40 | 0.061 | 0.050 | 0.019 | 0.043 |
| 0.50 | 0.089 | 0.076 | 0.028 | 0.064 |
| 0.60 | 0.121 | 0.105 | 0.040 | 0.089 |
| 0.70 | 0.159 | 0.135 | 0.053 | 0.116 |
| 0.75 | 0.169 | 0.146 | 0.057 | 0.124 |
| 0.80 | 0.164 | 0.138 | 0.054 | 0.119 |
| 0.90 | 0.106 | 0.086 | 0.033 | 0.075 |
| Perturbation | Severity | Normalized PGS | Percentile | Region agreement | |
| None | – | 1.000 | 0.000 | 0.000 | 100.0% |
| Magnitude scaling | 0.984 | 99.1% | |||
| 0.995 | 99.7% | ||||
| 1.001 | 99.8% | ||||
| 1.002 | 99.6% | ||||
| Angular perturbation | 0.928 | 96.1% | |||
| 0.794 | 89.4% | ||||
| 0.623 | 78.6% | ||||
| 0.441 | 65.2% | ||||
| 0.350 | 57.5% | ||||
| 0.268 | 50.7% | ||||
| Joint permutation | 25% joints | 0.842 | 91.8% | ||
| 50% joints | 0.654 | 81.0% | |||
| 75% joints | 0.510 | 70.3% | |||
| 100% joints | 0.372 | 59.5% | |||
| Independent sign flip | 25% joints | 0.878 | 93.4% | ||
| 50% joints | 0.712 | 84.8% | |||
| 75% joints | 0.660 | 83.0% | |||
| 100% joints | 1.000 | 100.0% |
| Component | Configuration |
|---|---|
| Input/output dimension | =29 |
| Backbone | Residual MLP |
| Number of residual blocks | 10 |
| Hidden dimension | 1024 |
| Timestep embedding | Sinusoidal embedding + MLP |
| Timestep conditioning | adaLN-Zero |
| Normalization | RMSNorm |
| Activation | SiLU |
| Prediction target | Clean pose |
| Training loss | Flow Matching velocity loss |
| Hyperparameter | Value | Hyperparameter | Value |
| Parallel environments | 4,096 | Rollout length | 32 |
| Samples per iteration | 131,072 | Discount factor | 0.99 |
| TD() / GAE parameter | 0.95 | PPO clipping ratio | 0.2 |
| Normalized advantage clipping | 4.0 | Actor optimization epochs | 5 |
| Critic optimization epochs | 2 | Actor minibatch size | 16,384 |
| Critic minibatch size | 8,192 | Actor learning rate | |
| Critic learning rate | Optimizer | SGD | |
| Actor network | Critic network | ||
| Action distribution | Fixed Gaussian | Action standard deviation | 0.05 |
| Actor output init. scale | 0.01 | Action-bound coefficient | 10.0 |
| Entropy coefficient | 0.0 | Action regularization | 0.0 |
| Mixed-precision training | Disabled | Normalizer samples |
Table V shows an inverted-U-shaped relationship between the evaluation timestep and the structural discriminability of PGS. The average discrimination score increases from at to at , and subsequently decreases to at . A similar trend is observed consistently across angular perturbation, joint permutation, and independent sign flip.
The neighboring timesteps and also obtain relatively strong average scores of and , respectively, indicating that PGS remains effective within an intermediate timestep range rather than depending on a narrowly tuned value. Among the evaluated settings, provides the strongest average response to disrupted joint-change structures, which is consistent with the timestep selected by the validation sweep reported in the main paper.
B-B Stability under Stochastic Corruption
At the selected evaluation timestep , we first verify that the online estimate remains consistent under stochastic corruption. For a fixed transition , we draw independent corrupted queries from Eq. (46) and obtain score samples , , one per corrupted query. The estimate obtained using independent corruption samples is
| (54) |
We use an independently sampled estimate as the high-accuracy reference and compare it with . Importantly, the proposed method uses during policy training, drawing a fresh corruption sample at every rollout step, and therefore requires only one JVP per rollout step. We report four metrics. The Spearman rank correlation measures how faithfully the ordering of the -sample estimates matches the reference ordering across transitions. The percentile MAE is the mean absolute error between the reference-relative percentiles obtained from the -sample and reference estimates, and the reward MAE is the corresponding mean absolute error of the resulting reward multiplier. Finally, we measure the calibration-region agreement: each reference-relative percentile is mapped to one of three reward regions by
| (55) |
and the agreement is the fraction of transitions for which the -sample and reference estimates fall into the same region. The same binning is reused for the perturbation study below. Latency is measured for one online PGS evaluation step using a batch size of 4096 on an NVIDIA GeForce RTX 4090 GPU.
Results. As shown in Table IV, the single-sample estimator achieves a Spearman rank correlation of and a calibration-region agreement of with the 128-sample reference, while adding only ms per evaluation step, over two orders of magnitude cheaper than the ms reference estimate. Increasing the number of corruption samples provides only marginal improvements, supporting the use of a single JVP as a stable and efficient online PGS estimator during policy training.
B-C Robustness to Pose-Change Perturbations
At the selected evaluation timestep , we examine whether PGS distinguishes the overall magnitude of a joint-coordinate change from its inter-joint coordination pattern.
For magnitude scaling, denoting by the finite-difference velocity of transition , we construct
| (56) |
and obtain the corresponding normalized direction as
| (57) |
Since PGS depends primarily on the normalized joint-change direction, its response should remain approximately invariant to , except when the change magnitude becomes comparable to the numerical stabilization term.
To assess the sensitivity to the inter-joint coordination pattern, we evaluate the perturbation family of Section B-A across severities: the angular rotation of Eq. (49) at , and joint permutation and independent sign flips applied to of the joint-coordinate blocks. Table VI reports all of these severities. The corrupted pose query (Eq. (46)) and sampled noise are held fixed before and after each perturbation, ensuring that the observed difference is caused only by the joint-change direction.
Let , , and denote the PGS, reference-relative percentile, and reward multiplier of the original transition, respectively. Their perturbed counterparts are denoted by , , and . We evaluate each perturbation using four metrics, averaged over transitions.
First, the normalized PGS measures the fraction of the original score retained after perturbation (with the stabilization constant added to both scores to keep the ratio stable near zero):
| (58) |
A value close to one indicates that the PGS response is preserved, whereas a value below one indicates a reduced alignment with the geometry encoded by the pose prior.
Second, the percentile change measures the signed displacement in the reference-calibrated PGS distribution:
| (59) |
Negative values indicate that the perturbed joint-change patterns are assigned lower reference-relative percentiles.
Third, the reward change measures the corresponding effect on the PGS reward multiplier:
| (60) |
A negative value indicates that the perturbation leads to stronger downweighting of the tracking reward.
Fourth, the region agreement measures whether a transition remains in the same reward-calibration region after perturbation, using the binning of Eq. (55):
| (61) |
Magnitude scaling is expected to produce , small percentile and reward changes, and high region agreement. In contrast, perturbations that progressively disrupt the inter-joint coordination pattern are expected to reduce , decrease the calibrated percentile and reward multiplier, and lower the region agreement.
| Skill | ADD | ADD w/ PDF-HR | ADD w/ PFM-HR |
|---|---|---|---|
| Walk | |||
| Run | |||
| Jump | |||
| Spinkick | |||
| Cartwheel | |||
| Backflip | Failed | ||
| Sideflip | |||
| SpeedVault | |||
| DoubleKong | Failed |
Results. Table VI shows that PGS is nearly invariant to the overall magnitude of a pose change. Across scaling factors from to , the normalized score remains within of the unperturbed value, and the calibration-region agreement remains above . In contrast, PGS decreases consistently as the joint-change pattern is perturbed. At an angular deviation of , only of the original score is retained, and fully permuting the joint changes reduces the normalized score to . Independently flipping the directions of a random subset of joints also lowers the score, to and at and of the joints, respectively, whereas a global sign flip of all joint directions leaves the score unchanged (), consistent with the sign-invariance of the quadratic PGS at a fixed query pose. These results suggest that PGS is robust to nuisance variations while remaining sensitive to the inter-joint coordination patterns.
Appendix C Implementation Details
C-A Pose Flow Matching Network Architecture
The pose prior is pretrained on the 60M-pose BONES-SEED corpus using the same normalized dimensional representation as in the main paper. We use the linear Flow Matching path and sample from the logit-normal distribution used in our implementation. The network is trained with the clean-pose parameterization and the induced velocity-space loss. The prior is pretrained independently of all downstream policies and is frozen before reinforcement learning.
PFM-HR follows the clean-data prediction formulation of JiT, adapted from image inputs to vector-valued humanoid poses. Since each pose is represented by an -dimensional joint-coordinate vector, we use a residual MLP instead of a patch-based Transformer. The network consists of an input projection, 10 residual MLP blocks with a hidden dimension of 1024, and a linear output projection. The Flow Matching timestep is encoded using sinusoidal features followed by a two-layer MLP and is injected into every residual block through adaLN-Zero conditioning. The output dimension is , allowing the network to directly predict the clean normalized pose. Following JiT, we use -prediction with a velocity-space loss. Given a clean pose , Gaussian noise , and timestep , the corrupted pose is
| (62) |
The network directly predicts the clean pose,
| (63) |
and the corresponding velocity prediction is
| (64) |
Using the target velocity , we optimize
| (65) |
Thus, the model output remains a clean-pose prediction, while the training error is evaluated in the induced Flow Matching velocity space. The resulting clean-pose map is subsequently frozen and used to compute the Jacobian–vector products required by PGS.
C-B RL Training Details
We implement all policies based on the ADD framework in MimicKit and use the Isaac Gym backend with 4,096 parallel environments. Unless otherwise specified, the PPO hyperparameters and actor–critic architecture follow the official ADD configuration as shown in Table VIII. The pretrained prior in PFM-HR is kept frozen throughout policy learning. All discriminator-related hyperparameters, including the discriminator architecture, learning rate, replay-buffer size, replay-sampling count, gradient penalty, and discriminator reward scale, follow the official MimicKit ADD configuration without modification.
The actor and critic are two-layer fully connected networks with 1,024 hidden units per layer. MimicKit specifies the actor and critic minibatch sizes as multiples of the number of parallel environments. Therefore, the corresponding minibatch sizes are and , respectively. All compared methods use the same training budget and environment configuration. We evaluate each policy every 100 training iterations using 4,096 episodes and report the mean and standard deviation over three seeds.
Appendix D Additional Experimental Results
D-A Rotation Error on Single-Trajectory Tasks
Table IX reports the rotational tracking errors for all nine single-trajectory tracking tasks. These results complement the positional tracking errors reported in Table 2 of the main paper. We use the same evaluation protocol and report the mean and one standard deviation across three independent random seeds. PFM-HR obtains the lowest mean rotational error on six of the nine tasks and ties for the lowest error on SpeedVault. The vanilla ADD baseline performs best on the two basic locomotion tasks, Walk and Run. In contrast, PFM-HR performs best on the more dynamic tasks, including Jump, Spinkick, Cartwheel, Backflip, Sideflip, and Double Kong. In particular, PFM-HR maintains low rotational errors on Backflip and Double Kong, for which vanilla ADD fails.
D-B Training Curves on Single-motion Tracking
Figure 7 presents the complete training curves for the nine single-motion tracking tasks. Both pose-prior methods generally improve sample efficiency over vanilla ADD, yet the magnitude of the benefit depends strongly on the complexity of the reference motion. On dynamic and acrobatic skills such as Jump, Cartwheel, Backflip, Sideflip, SpeedVault, and Double Kong, PFM-HR converges particularly rapidly, reaching the full reference length well before vanilla ADD. Its advantage is most pronounced on Backflip and Double Kong, where vanilla ADD fails to reach the full reference length within the training budget, whereas PFM-HR consistently solves the motions across all three seeds. On the comparatively simpler Walk, Run, and Spinkick tasks, where joint trajectories are more predictable and rely less on tightly coordinated multi-joint timing, PDF-HR converges slightly faster than or comparably to PFM-HR, although both methods remain substantially more sample-efficient than ADD.
D-C Training Curves on General Motion Tracking
Figure 6 shows the success-rate curves for general-motion tracking under three episode horizons (10, 20, and 30 seconds). Longer horizons require the policy to track the reference over many more timesteps and to recover from error accumulation, making them a substantially harder test of reward quality. PFM-HR consistently shifts the learning curves to the left, reaching high success rates with fewer training samples than both vanilla ADD and ADD with PDF-HR. This advantage is observed across all three horizons and becomes more evident as the episode horizon increases, suggesting that the benefit of geometry-aware reward modulation grows with the amount of accumulated drift the policy must correct. PDF-HR also improves sample efficiency over vanilla ADD, but its gains are smaller than those of PFM-HR, indicating that the richer inter-joint geometry information used by PFM-HR translates into a clearer learning signal. In addition to faster convergence, PFM-HR achieves the highest final success rate across all three horizons, showing that PGS-based reward modulation improves both learning speed and long-horizon tracking reliability rather than trading one off against the other.