Focus On What Matters: Separated Models For Visual-Based RL Generalization

no code implementations • 29 Sep 2024 • Di Zhang, Bowen Lv, Hai Zhang, Feifan Yang, Junqiao Zhao, Hang Yu, Chang Huang, Hongtu Zhou, Chen Ye, Changjun Jiang

Perceiving the pre-eminence of image reconstruction in representation learning, we propose SMG (Separated Models for Generalization), a novel approach that exploits image reconstruction for generalization.

Image Reconstruction Reinforcement Learning (RL) +1