Numerous locomotion controllers have been designed based on Reinforcement Learning (RL) to facilitate blind quadrupedal locomotion traversing challenging terrains. Nevertheless, locomotion control is still a challenging task for quadruped robots traversing diverse terrains amidst unforeseen disturbances. Recently, privileged learning has been employed to learn reliable and robust quadrupedal locomotion over various terrains based on a teacher-student architecture. However, its one-encoder structure is not adequate in addressing external force perturbations. The student policy would experience inevitable performance degradation due to the feature embedding discrepancy between the feature encoder of the teacher policy and the one of the student policy. Hence, this paper presents a privileged learning framework with multiple feature encoders and a residual policy network for robust and reliable quadruped locomotion subject to various external perturbations. The multi-encoder structure can decouple latent features from different privileged information, ultimately leading to enhanced performance of the learned policy in terms of robustness, stability, and reliability. The efficiency of the proposed feature encoding module is analyzed in depth using extensive simulation data. The introduction of the residual policy network helps mitigate the performance degradation experienced by the student policy that attempts to clone the behaviors of a teacher policy. The proposed framework is evaluated on a Unitree GO1 robot, showcasing its performance enhancement over the state-of-the-art privileged learning algorithm through extensive experiments conducted on diverse terrains. Ablation studies are conducted to illustrate the efficiency of the residual policy network.
Figures & tables
Fig. 1: A Unitree Go1 quadruped robot is subject to a kick, while standing on a grass field.
Fig. 2: The proposed PA-LOCO integrates a teacher-student framework with a residual network and multiple feature encoders. The training process involves three phases. In the first phase, the teacher policy is trained with proprioceptive observations ot and privileged information Ft,Tt,St that is unknown for deployment. In the second phase, the student policy is trained using observations from the proprioceptive sensors. The student policy is learned to clone the teacher’s actions and latent features by supervised learning. In the third phase, the residual policy network is trained to further enhance the performance of the student policy against perturbations.
Name
Expression
Weight
Linear velocity tracking
exp(−0.25∣∣vb,xy∗−vb,xy∣∣2)
1dt
Angular velocity tracking
exp(−0.25∣∣ωb,z∗−ωb,z∣∣2)
0.5dt
Linear velocity penalty
vb,z2
-2dt
Angular velocity penalty
∣∣ωb,xy∣∣2
-0.05dt
Trunk orientation
∣∣g^x∣∣2+∣∣g^y∣∣2
-1dt
Trunk height
∣∣hb−hb∗∣∣2
-5dt
TABLE I: Reward functions and weights
Parameter
Value
Learning rate
Adaptive
Batch size
98304 (4096 × 24)
Mini-batch size
24576 (4096 × 6)
Discount factor
0.99
GAE lambda
0.95
Desired KL-divergence
0.01
TABLE II: PPO hyper-parameters
Parameter
Range
Randomized dynamics
Trunk mass
[-1,1] kg
COM displacement
[-0.03, 0.03] m
Ground friction coefficient
[0.25, 1.5]
Trunk impulse
External force magnitude Fx,Fy
[-60, 60] N
External force magnitude Fz
[-10, 10] N
External force noise
[-2,2] N
TABLE III: Parameter setting for training
Module (MLP)
Input
Hidden Layers
Output
πT
ot,l^t
[512, 256, 128]
a^t
V
ot,Ft,St,Tt
[512, 256, 128]
Vt
EF
Ft
[64, 32]
l^tF
ET
Tt
[256, 128]
l^tT
ES
St
[64, 32]
l^tS
πS
ot,lt
[512, 256, 128]
at
TABLE IV: Network configurations
Fig. 3: The locomotion behavior when subjected to external force impulses from the front. The trunk velocity and height responses are provided in the second image from the top. The feet’s contact patterns with the ground (F/R denotes Front/Rear and R/L denotes Right/Left) are given in the third image from the top. The image at the bottom shows the plots of the force latent variables given by the force encoder.
Fig. 4: Indoor experiment setup.
Weights
Metrics
Algorithms
Robust
SEFA
MEFA
PA-LOCO
2.2 kg
SR (%) LO (m) RT (s) HO (m)
90 (9/10)
0
100 (10/10)
100 (10/10)
0.33
-
0.17
0.16
1.06
-
0.50
0.45
0.00
-
-0.01
−0.02
4.6 kg
SR (%) LO (m) RT (s) HO (m)
43 (3/7)
0
80 (8/10)
90 (9/10)
TABLE V: Metric results for different algorithms
Fig. 5: The t-SNE visualization of learned latent representation. In different trials, the robot is subjected to a constant backward force of different magnitudes when it moves forward. It indicates that the student policy is perturbation-aware due to the force encoder.
Fig. 6: The correlation heatmap between latent features and different force directions. lT , lS , lF denote the latent features of the terrain profile Tt , robot state St , and external forces Ft respectively. In different trials, the robot experiences a constant external force of 20 N in different directions.