CoRL

Entity Abstraction in Visual Model-Based Reinforcement Learning

The Object-centric perception, prediction, and planning (OP3) framework demonstrates strong generalization to novel configurations in block stacking tasks by symmetrically processing entity representations extracted from raw visual observations.