APEX: Action Priors Enable Efficient Exploration for Robust Motion Tracking on Legged Robots
About
Learning natural, animal-like locomotion from demonstrations has become a core paradigm in legged robotics. While motion tracking can reproduce reference gaits, many approaches still require substantial tuning and depend on reference motion inputs at deployment, which can limit responsiveness to task objectives and reduce adaptability. We present APEX (Action Priors enable Efficient eXploration), a motion-tracking reinforcement learning (RL) framework that removes deployment-time dependence on reference motion inputs, improves sample efficiency, and reduces tuning effort. APEX integrates demonstrations into RL via decaying action priors, which guide early exploration toward demonstration-consistent actions and then fade to zero, yielding a pure RL policy at deployment. This is combined with a multi-critic framework that separates style and task + regularization learning signals. Moreover, APEX enables a single policy to learn diverse motions and transfer reference-like styles across different terrains and velocities, while remaining robust to variations in training parameters. We validate our method in simulation on both humanoid and quadruped robots, and with zero-shot deployment on a Unitree Go2 robot. Website and code: https://marmotlab.github.io/APEX/.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Humanoid motion tracking | Unitree G1 Cartwheel | Joint Position RMSE0.122 | 4 | |
| Motion Tracking | Quadruped Locomotion Canter v1 (test) | q0.077 | 4 | |
| Motion Tracking | Quadruped Locomotion Trot v1 (test) | Tracking Error (q)0.039 | 4 | |
| Motion Tracking | Quadruped Locomotion Hopturn v1 (test) | q Metric0.057 | 4 | |
| Motion Tracking | Quadruped Locomotion Pronk v1 (test) | q0.071 | 4 | |
| Motion Tracking | Quadruped Locomotion Sidestep v1 (test) | Tracking Error (q)0.041 | 4 | |
| Motion Tracking | Quadruped Locomotion Pace v1 (test) | Tracking Error (q)0.047 | 4 | |
| Humanoid motion tracking | Unitree G1 Spinkick | Joint Position RMSE (q)0.126 | 4 |