Domain Adaptation In Reinforcement Learning Via Latent Unified State Representation


  • Jinwei Xing University of California, Irvine
  • Takashi Nagata University of California, Irvine
  • Kexin Chen University of California, Irvine
  • Xinyun Zou University of California, Irvine
  • Emre Neftci University of California, Irvine
  • Jeffrey L. Krichmar University of California, Irvine



Reinforcement Learning


Despite the recent success of deep reinforcement learning (RL), domain adaptation remains an open problem. Although the generalization ability of RL agents is critical for the real-world applicability of Deep RL, zero-shot policy transfer is still a challenging problem since even minor visual changes could make the trained agent completely fail in the new task. To address this issue, we propose a two-stage RL agent that first learns a latent unified state representation (LUSR) which is consistent across multiple domains in the first stage, and then do RL training in one source domain based on LUSR in the second stage. The cross-domain consistency of LUSR allows the policy acquired from the source domain to generalize to other target domains without extra training. We first demonstrate our approach in variants of CarRacing games with customized manipulations, and then verify it in CARLA, an autonomous driving simulator with more complex and realistic visual observations. Our results show that this approach can achieve state-of-the-art domain adaptation performance in related RL tasks and outperforms prior approaches based on latent-representation based RL and image-to-image translation.




How to Cite

Xing, J., Nagata, T., Chen, K., Zou, X., Neftci, E., & Krichmar, J. L. (2021). Domain Adaptation In Reinforcement Learning Via Latent Unified State Representation. Proceedings of the AAAI Conference on Artificial Intelligence, 35(12), 10452-10459.



AAAI Technical Track on Machine Learning V