Synergistic Action and Style Priors for Learning Legged Locomotion
Authors: Oliver Hausdörfer, Alexander von Rohr, Éric Lefort, Angela Schoellig
Organizations: Technical University of Munich, Germany; TUM School of Computation, Information and Technology, Department of Computer Engineering, Learning Systems and Robotics Lab; Munich Institute of Robotics and Machine Intelligence.
Deep Reinforcement Learning (DRL) enables robots to learn complex behaviors through interaction with the environment. However, due to the unrestricted nature of the learning algorithms, the resulting solutions are often brittle and appear unnatural. This is especially true for learning direct joint-level torque control, as inductive biases are difficult to integrate into the learning process. We propose an inductive bias for learning locomotion that is especially useful for torque control: latent actions learned from a small dataset of expert demonstrations. This prior allows the policy to directly leverage knowledge contained in the expert's actions and facilitates more efficient exploration. We observe that the agent is not restricted to the reward levels of the demonstration, and performance in transfer tasks is improved significantly. Latent action priors combined with style rewards for imitation lead to a closer replication of the expert's behavior. Videos are available at https://sites.google.com/view/latent-action-priors.
Figures & tables
Fig. 2: We evaluate the synergies of combining the simple data-driven action prior with a style reward term on (a) - (c) three Gymnasium Mujoco [ 25 ] environments of varying difficulty, and (d) a simulated Unitree A1 quadruped with position and torque control from the loco-mujoco benchmark [ 26 ] . (e) - (f) We then test our approach on a Unitree Go2 for two gait types and terrains using LLM-generated demonstrations, and deploy the policy on hardware for both position and direct 500Hz torque control.
Fig. 3: Prompt template we use to generate the gait controllers.
Fig. 4: All motions from the demonstrations can be expressed with only a few Principal components (PCs), motivating the use of a compressed action space that exploits such correlations. PCs left of the vertical red line explain >95% of the variance in the actions and the black lines show the cumulative explained variance. Results from data in Tab. II .
Fig. 5: Results on benchmark environments. A data-driven action prior significantly outperforms using unbiased PPO and a simple reference position in the action space regarding both sample efficiency during DRL and maximum reward for all environments. This strongly motivates the use of demonstrations for policy learning rather than static joint position references. Adding a single style reward term (+style) creates synergistic effects except for Humanoid, where the demonstration is brittle. We run all experiments on five seeds and report the means and standard deviations on task rewards. BC is omitted as it did not improve upon PPO.
Fig. 6: Policy learning for different hyperparameters introduced by the latent action prior: action space weighting wa and the dimension of the latent action space dim( al ) for the Unitree A1 . Shown are the mean and standard deviation on 5 seeds.
Fig. 7: The combination of a latent action prior with a single style reward term is significantly less sensitive towards reward tuning than baselines with only DeepMimic-style style rewards and reference joint positions. Shown is the ablation on the style reward weights wstyle for Unitree A1 ( wtask=1−wstyle ). Shown are the mean and standard deviation on 5 seeds.
Fig. 8: Joint positions for the left hind leg of the Unitree A1 for torque control in simulation. The synergy of both priors leads to joint positions close to the expert. The BC policy tracks the expert demonstration for the first frames but diverges quickly. RMSE [rad]: PPO+latent+style: 0.05, PPO+latent: 0.17, PPO+reference+style: 0.31, PPO+style: 0.52.
Fig. 9: LLMs can effectively design open-loop gait controllers for various locomotion scenarios leveraging established methods [ 30 , 31 ] : (a) trot, (b) walk, and (c) stair climbing. Fig. (d) Shows the produced feet contact timing for the trotting and walking gaits when executing the open-loop controllers in simulation and the resulting policies. The feet contact patterns translate to the trained policy indicating imitation.
Fig. 10: While position control can directly be deployed to hardware, torque control on the robot leads to high-frequency vibrations when no penalty for the action rate is added. Shown are the commanded torques sent to the robot’s rear left calf joint.
Reinforcement learning (RL) for legged robots is advancing locomotion, demonstrating its ability to adapt to new and challenging terrain. Traditionally, these RL locomotion frameworks are position-based, making the policy less adaptable to terrain types and requiring state estimation techniques in the observation space, i.e., linear velocity. Moreover, these RL frameworks often use small, lightweight quadrupeds that are limited in their viability for high-complexity tasks due to hardware constraints. This work explores an RL torque control framework for heavyweight high-torque quadrupeds. The RL framework in this paper can traverse rough terrain and effectively track a desired linear velocity without requiring knowledge of the agent's current velocity. Using Nvidia's Isaac Sim and Isaac Lab, simulation results of the RL torque control policy are shown on the Unitree B1 quadruped, achieving speeds of 3.5 m/s and 1.5 rad/s. In addition, the quadruped can walk up and down stairs without the aid of an exteroceptive sensor.
Jordan Dowdy, Jean Chagas Vaz
Department of Electrical and Computer Engineering at the University of Louisville. · Faculty of Electrical and Computer Engineering, University of Louisville, Louisville, KY, 40208, USA.
Despite recent advances in humanoid locomotion, controllers optimized for command tracking and robustness tend to produce mechanical gaits, whereas controllers tied to human motion data often fail to generalize to commands outside the data distribution. This work introduces a learning framework that balances these competing objectives to synthesize real-time steerable, robust, and biomimetic locomotion policies from human data. Using an in-house curated locomotion dataset covering diverse speeds and directions, we first learn a natural locomotion prior policy through a teacher-student distillation process. Specifically, we train a full-body reference-conditioned policy with Reinforcement Learning (RL), then distill it into a lightweight prior policy conditioned solely on proprioception and a planar torso-velocity steering command. Next, we fine-tune the prior policy with multi-task RL to expand command coverage and robustness beyond the data distribution, pairing a goal-conditioned task that tracks arbitrary commands with a reference-guided task that tracks the human data as an explicit style regularizer. We validate our framework on three humanoid robots: the Boston Dynamics Atlas R1, Atlas D1, and Unitree G1. Experimental results demonstrate robust performance across real-world scenarios, including direct user-controlled locomotion in indoor and outdoor environments, and integration as the locomotion layer within hierarchical control stacks. Benchmarks against Tabula Rasa RL policies trained without human data and ablation studies confirm that our framework yields a lightweight, deployable policy that reconstructs coordinated whole-body behavior from a steering command, retaining the human gait characteristics while remaining robust and fully steerable.
Mike Zhang, Dongho Kang, Kevin Bergamin +19
RAI Institute, USA · Boston Dynamics, USA · Carnegie Mellon University, USA
Real-world learning for dexterous hands remains brittle because high-dimensional hand actions amplify imitation errors and make reinforcement-learning exploration prone to contact-breaking motion. While combining imitation learning (IL) with online reinforcement learning (RL) can reduce manual supervision, unconstrained exploration in raw hand-action spaces is sample-inefficient and risky for physical hardware. We introduce a latent motion prior module (\prior{}) that maps recent hand-action histories to a compact, history-conditioned latent prior and decodes continuous latent commands into executable high-dimensional hand targets. Built on this prior, \method{} is a three-stage real-world dexterous learning framework: it pretrains \prior{} from demonstrations, trains a visuomotor policy that predicts native arm commands and latent hand-action offsets, and improves the policy with online residual RL in the same latent hand-action space. This shared, decodable interface lets residual exploration make local corrections near demonstrated, contact-consistent hand motions rather than perturbing every finger joint independently. We evaluate \method{} on four real-robot dexterous manipulation tasks against raw, linear, and discrete hand-action interfaces. Starting from small task-specific demonstration sets, \method{} achieves a 56.25% average IL success rate and raises it to 98.75% after online RL, reaching 100% final success on three tasks and 95% on the remaining task.
Xinye Yang, Zhiyuan Ma, Hongze Yu +5
Fudan University · PsiBot · Zhongguancun Academy +2