Dream to Generalize: Zero-Shot Model-Based Reinforcement Learning for Unseen Visual Distractions
To address the degradation of model-based reinforcement learning (MBRL) generalization under high-dimensional visual observations corrupted by clouds, shadows, and illumination variations, this paper proposes Dr. G—a zero-shot model-based RL framework. Our approach tackles this challenge through three key contributions: (1) a novel dual-contrastive self-supervised learning mechanism that disentangles and encodes task-relevant features from multi-view augmented data; (2) recurrent state-wise inverse dynamics modeling to enhance the world model’s temporal causal understanding; and (3) zero-shot cross-background transfer without fine-tuning. Evaluated on DeepMind Control (with complex video backgrounds) and Robosuite (with randomized environments), Dr. G achieves performance gains of 117% and 14%, respectively, over state-of-the-art methods. The implementation is publicly available.