SegDAC: Visual Generalization in Reinforcement Learning via Dynamic Object Tokens

Alexandre Brown, Glen Berseth

Visual reinforcement learning policies trained on pixel observations often struggle to generalize when visual conditions change at test time. Object-centric representations are a p…