Policy Transfer with Strategy Optimization
About
Computer simulation provides an automatic and safe way for training robotic control policies to achieve complex tasks such as locomotion. However, a policy trained in simulation usually does not transfer directly to the real hardware due to the differences between the two environments. Transfer learning using domain randomization is a promising approach, but it usually assumes that the target environment is close to the distribution of the training environments, thus relying heavily on accurate system identification. In this paper, we present a different approach that leverages domain randomization for transferring control policies to unknown environments. The key idea that, instead of learning a single policy in the simulation, we simultaneously learn a family of policies that exhibit different behaviors. When tested in the target environment, we directly search for the best policy in the family based on the task performance, without the need to identify the dynamic parameters. We evaluate our method on five simulated robotic control problems with different discrepancies in the training and testing environment and demonstrate that our method can overcome larger modeling errors compared to training a robust policy or an adaptive policy.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Locomotion | HalfCheetah Mass Scale 1.0x | Cumulative Reward9.60e+3 | 6 | |
| Locomotion | Ant Damping Scale 2.2x | Cumulative Reward4.96e+3 | 6 | |
| Locomotion | Ant Time-Varying Dynamics | Cumulative Reward4.79e+3 | 6 | |
| Locomotion | Ant (in-distribution) | Mean Cumulative Reward5.04e+3 | 6 | |
| Locomotion | Hopper Mass Scale 1.0x | Cumulative Reward3.26e+3 | 6 | |
| Locomotion | Hopper Structural Failure | Cumulative Reward432 | 6 | |
| Locomotion | Walker2d Damping Scale 1.0x | Cumulative Reward5.19e+3 | 6 | |
| Locomotion | Ant Damping Scale 0.3x | Cumulative Reward4.13e+3 | 6 | |
| Locomotion | Ant Damping Scale 1.0x | Cumulative Reward4.06e+3 | 6 | |
| Locomotion | Walker2d Structural Failure | Cumulative Reward504 | 6 |