UAI Conference 2024 Conference Paper
SMuCo: Reinforcement Learning for Visual Control via Sequential Multi-view Total Correlation
- Tong Cheng
- Hang Dong 0004
- Lu Wang 0029
- Bo Qiao 0001
- Qingwei Lin
- Saravan Rajmohan
- Thomas Moscibroda
The advent of abundant image data has catalyzed the advancement of visual control in reinforcement learning (RL) systems, leveraging multiple view- points to capture the same physical states, which could enhance control performance theoretically. However, integrating multi-view data into representation learning remains challenging. In this paper, we introduce SMuCo, an innovative multi-view reinforcement learning algorithm that constructs robust latent representations by optimizing multi- view sequential total correlation. This technique effectively captures task-relevant information and temporal dynamics while filtering out irrelevant data. Our method supports an unlimited number of views and demonstrates superior performance over leading model-free and model-based RL algorithms. Empirical results from the DeepMind Control Suite and the Sapien Basic Manipulation Task confirm SMuCo’s enhanced efficacy, significantly improving task performance across diverse scenarios and views.