arXiv · 2312.01915
Learning Robust Representations via Bidirectional Transition for Visual Reinforcement Learning
Abstract
Visual reinforcement learning has proven effective in solving control tasks with high-dimensional observations. However, extracting reliable and generalizable representations from vision-based observations remains a central challenge. Inspired by the human thought process, when the representation extracted from the observation can predict the future and trace history, the representation is reliable and accurate in comprehending the environment. Based on this concept, we introduce a Bidirectional Transition (BiT) model, which leverages the ability to bidirectionally predict environmental transitions both forward and backward to extract reliable representations. Our model demonstrates competitive generalization performance and sample efficiency on two settings of the DeepMind Control suite. Additionally, we utilize robotic manipulation and CARLA simulators to demonstrate the wide applicability of our method.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Xiaobo Hu, Youfang Lin, Yue Liu, Jinwen Wang, Shuo Wang, Hehe Fan, Kai Lv. 2023-12-04. Learning Robust Representations via Bidirectional Transition for Visual Reinforcement Learning. https://arxiv.org/abs/2312.01915
Cite the original work for its findings. Save a collection to share your selection of sources.