Self-supervised Visual Reinforcement Learning with Object-centric Representations
Not on the map. Why?
Abstract (arXiv)
Autonomous agents need large repertoires of skills to act reasonably on new tasks that they have not seen before. However, acquiring these skills using only a stream of high-dimensional, unstructured, and unlabeled observations is a tricky challenge for any autonomous agent. Previous methods have used variational autoencoders to encode a scene into a low-dimensional vector that can be used as a goal for an agent to discover new skills. Nevertheless, in compositional/multi-object environments it is difficult to disentangle all the factors of variation into such a fixed-length representation of the whole scene. We propose to use object-centric representations as a modular and structured observation space, which is learned with a compositional generative world model. We show that the structure in the representations in combination with goal-conditioned attention policies helps the autonomous agent to discover and learn useful skills. These skills can be further combined to address compositional tasks like the manipulation of several different objects.
Key facts. Cited by 43 papers in the map (data through 7 Oct 2026). First posted on arXiv 29 Nov 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 3; 2022: 6; 2023: 8; 2024: 14; 2025: 6; 2026: 6.
43 papers.
Sort:
By area:
- 2022Bridging the Gap to Real-World Object-Centric Learningcited by 219
- 2023SlotDiffusion: Object-Centric Generative Modeling with Diffusion Modelscited by 85
- 2023Object-Centric Learning for Real-World Videos by Predicting Temporal Feature Similaritiescited by 78
- 2023An Investigation into Pre-Training Object-Centric Representations for Reinforcement Learningcited by 49
- 2024Dream to Manipulate: Compositional World Models Empowering Robot Imitation Learning with Imaginationcited by 45
- 2024Temporally Consistent Object-Centric Learning by Contrasting Slotscited by 30
- 2024Entity-Centric Reinforcement Learning for Object Manipulation from Pixelscited by 27
- 2022Policy Architectures for Compositional Generalization in Controlcited by 23
- 2021Towards Deeper Deep Reinforcement Learning with Spectral Normalizationcited by 19
- 2022Compositional Multi-Object Reinforcement Learning with Linear Relation Networkscited by 19
- 2024Zero-Shot Object-Centric Representation Learningcited by 19
- 2021Self-supervised Reinforcement Learning with Independently Controllable Subgoalscited by 16
31 more
- 2025Object-Centric World Model for Language-Guided Manipulationcited by 16
- 2023Learning Dynamic Attribute-factored World Models for Efficient Multi-object Reinforcement Learningcited by 15
- 2024BlendRL: A Framework for Merging Symbolic and Neural Policy Learningcited by 14
- 2022Learning to Generalize with Object-centric Agents in the Open World Survival Game Craftercited by 13
- 2025CTRL-O: Language-Controllable Object-Centric Visual Representation Learningcited by 12
- 2022Unsupervised Multi-object Segmentation Using Attention and Soft-argmaxcited by 11
- 2025Learning Interactive World Model for Object-Centric Reinforcement Learningcited by 11
- 2021Object-Centric Representation Learning with Generative Spatial-Temporal Factorizationcited by 10
- 2024Fine-Grained Causal Dynamics Learning with Quantization for Improving Robustness in Reinforcement Learningcited by 10
- 2024SkiLD: Unsupervised Skill Discovery Guided by Factor Interactionscited by 9
- 2025SENSEI: Semantic Exploration Guided by Foundation Models to Learn Versatile World Modelscited by 9
- 2024Recurrent Complex-Weighted Autoencoders for Unsupervised Object Discoverycited by 8
- 2024EC-Diffuser: Multi-Object Manipulation via Entity-Centric Behavior Generationcited by 8
- 2022Object-Category Aware Reinforcement Learningcited by 6
- 2025Object-Centric Pretraining via Target Encoder Bootstrappingcited by 6
- 2026Hierarchical Entity-centric Reinforcement Learning with Factored Subgoal Diffusioncited by 6
- 2024Object-Centric Temporal Consistency via Conditional Autoregressive Inductive Biasescited by 5
- 2026OCRA: Object-Centric Learning with 3D and Tactile Priors for Human-to-Robot Action Transfercited by 5
- 2023Consciousness-Inspired Spatio-Temporal Abstractions for Better Generalization in Reinforcement Learningcited by 4
- 2023Relational Object-Centric Actor-Criticcited by 4
- 2023Intrinsic Motivation in Model-based Reinforcement Learning: A Brief Reviewcited by 3
- 2024OCK: Unsupervised Dynamic Video Prediction with Object-Centric Kinematicscited by 3
- 2024Learning Causal Dynamics Models in Object-Oriented Environmentscited by 3
- 2025Object-Centric World Models for Causality-Aware Reinforcement Learningcited by 3
- 2024Object-centric proto-symbolic behavioural reasoning from pixelscited by 2
- 2023Discovering Object-Centric Generalized Value Functions From Pixelscited by 1
- 2024Unsupervised Salient Patch Selection for Data-Efficient Reinforcement Learningcited by 0
- 2026QASA: Quality-Guided K-Adaptive Slot Attention for Unsupervised Object-Centric Learningcited by 0
- 2026GRAIL: Autonomous Concept Grounding for Neuro-Symbolic Reinforcement Learningcited by 0
- 2026Learning Bilevel Policies over Symbolic World Models for Long-Horizon Planningcited by 0
- 2026Semantic Slots for Video Object-Centric Learningcited by 0
- 2026Semantic Slots for Video Object-Centric Learning21 Aug 2026
- 2026Learning Bilevel Policies over Symbolic World Models for Long-Horizon Planning15 May 2026
- 2026GRAIL: Autonomous Concept Grounding for Neuro-Symbolic Reinforcement Learning18 Apr 2026
- 2026OCRA: Object-Centric Learning with 3D and Tactile Priors for Human-to-Robot Action Transfer15 Mar 2026
- 2026Hierarchical Entity-centric Reinforcement Learning with Factored Subgoal Diffusion2 Feb 2026
- 2026QASA: Quality-Guided K-Adaptive Slot Attention for Unsupervised Object-Centric Learning19 Jan 2026
- 2025Object-Centric World Models for Causality-Aware Reinforcement Learning18 Nov 2025
- 2025Learning Interactive World Model for Object-Centric Reinforcement Learning4 Nov 2025
- 2025CTRL-O: Language-Controllable Object-Centric Visual Representation Learning27 Mar 2025
- 2025Object-Centric Pretraining via Target Encoder Bootstrapping19 Mar 2025
- 2025Object-Centric World Model for Language-Guided Manipulation8 Mar 2025
- 2025SENSEI: Semantic Exploration Guided by Foundation Models to Learn Versatile World Models3 Mar 2025
31 more
- 2024EC-Diffuser: Multi-Object Manipulation via Entity-Centric Behavior Generation25 Dec 2024
- 2024Dream to Manipulate: Compositional World Models Empowering Robot Imitation Learning with Imagination19 Dec 2024
- 2024Temporally Consistent Object-Centric Learning by Contrasting Slots18 Dec 2024
- 2024Object-centric proto-symbolic behavioural reasoning from pixels26 Nov 2024
- 2024SkiLD: Unsupervised Skill Discovery Guided by Factor Interactions24 Oct 2024
- 2024Object-Centric Temporal Consistency via Conditional Autoregressive Inductive Biases21 Oct 2024
- 2024BlendRL: A Framework for Merging Symbolic and Neural Policy Learning15 Oct 2024
- 2024Zero-Shot Object-Centric Representation Learning17 Aug 2024
- 2024Fine-Grained Causal Dynamics Learning with Quantization for Improving Robustness in Reinforcement Learning5 Jun 2024
- 2024Recurrent Complex-Weighted Autoencoders for Unsupervised Object Discovery27 May 2024
- 2024Learning Causal Dynamics Models in Object-Oriented Environments21 May 2024
- 2024OCK: Unsupervised Dynamic Video Prediction with Object-Centric Kinematics29 Apr 2024
- 2024Entity-Centric Reinforcement Learning for Object Manipulation from Pixels1 Apr 2024
- 2024Unsupervised Salient Patch Selection for Data-Efficient Reinforcement Learning10 Jan 2024
- 2023Relational Object-Centric Actor-Critic26 Oct 2023
- 2023Consciousness-Inspired Spatio-Temporal Abstractions for Better Generalization in Reinforcement Learning30 Sep 2023
- 2023Learning Dynamic Attribute-factored World Models for Efficient Multi-object Reinforcement Learning18 Jul 2023
- 2023Object-Centric Learning for Real-World Videos by Predicting Temporal Feature Similarities7 Jun 2023
- 2023SlotDiffusion: Object-Centric Generative Modeling with Diffusion Models18 May 2023
- 2023Discovering Object-Centric Generalized Value Functions From Pixels27 Apr 2023
- 2023An Investigation into Pre-Training Object-Centric Representations for Reinforcement Learning9 Feb 2023
- 2023Intrinsic Motivation in Model-based Reinforcement Learning: A Brief Review24 Jan 2023
- 2022Object-Category Aware Reinforcement Learning13 Oct 2022
- 2022Bridging the Gap to Real-World Object-Centric Learning29 Sep 2022
- 2022Learning to Generalize with Object-centric Agents in the Open World Survival Game Crafter5 Aug 2022
- 2022Unsupervised Multi-object Segmentation Using Attention and Soft-argmax26 May 2022
- 2022Policy Architectures for Compositional Generalization in Control10 Mar 2022
- 2022Compositional Multi-Object Reinforcement Learning with Linear Relation Networks31 Jan 2022
- 2021Object-Centric Representation Learning with Generative Spatial-Temporal Factorization9 Nov 2021
- 2021Self-supervised Reinforcement Learning with Independently Controllable Subgoals9 Sep 2021
- 2021Towards Deeper Deep Reinforcement Learning with Spectral Normalization2 Jun 2021
- 2024Dream to Manipulate: Compositional World Models Empowering Robot Imitation Learning with Imaginationcited by 45
- 2025Object-Centric World Model for Language-Guided Manipulationcited by 16
- 2023Learning Dynamic Attribute-factored World Models for Efficient Multi-object Reinforcement Learningcited by 15
- 2025Learning Interactive World Model for Object-Centric Reinforcement Learningcited by 11
- 2025SENSEI: Semantic Exploration Guided by Foundation Models to Learn Versatile World Modelscited by 9
- 2023Relational Object-Centric Actor-Criticcited by 4
- 2023Intrinsic Motivation in Model-based Reinforcement Learning: A Brief Reviewcited by 3
- 2025Object-Centric World Models for Causality-Aware Reinforcement Learningcited by 3
- 2026Learning Bilevel Policies over Symbolic World Models for Long-Horizon Planningcited by 0
- 2026Learning Bilevel Policies over Symbolic World Models for Long-Horizon Planning15 May 2026
- 2025Object-Centric World Models for Causality-Aware Reinforcement Learning18 Nov 2025
- 2025Learning Interactive World Model for Object-Centric Reinforcement Learning4 Nov 2025
- 2025Object-Centric World Model for Language-Guided Manipulation8 Mar 2025
- 2025SENSEI: Semantic Exploration Guided by Foundation Models to Learn Versatile World Models3 Mar 2025
- 2024Dream to Manipulate: Compositional World Models Empowering Robot Imitation Learning with Imagination19 Dec 2024
- 2023Relational Object-Centric Actor-Critic26 Oct 2023
- 2023Learning Dynamic Attribute-factored World Models for Efficient Multi-object Reinforcement Learning18 Jul 2023
- 2023Intrinsic Motivation in Model-based Reinforcement Learning: A Brief Review24 Jan 2023
- 2025Object-Centric Pretraining via Target Encoder Bootstrapping19 Mar 2025
- 2022Bridging the Gap to Real-World Object-Centric Learningcited by 219
- 2023Object-Centric Learning for Real-World Videos by Predicting Temporal Feature Similaritiescited by 78
- 2023An Investigation into Pre-Training Object-Centric Representations for Reinforcement Learningcited by 49
- 2024Temporally Consistent Object-Centric Learning by Contrasting Slotscited by 30
- 2024Entity-Centric Reinforcement Learning for Object Manipulation from Pixelscited by 27
- 2022Policy Architectures for Compositional Generalization in Controlcited by 23
- 2021Towards Deeper Deep Reinforcement Learning with Spectral Normalizationcited by 19
- 2022Compositional Multi-Object Reinforcement Learning with Linear Relation Networkscited by 19
- 2024Zero-Shot Object-Centric Representation Learningcited by 19
- 2021Self-supervised Reinforcement Learning with Independently Controllable Subgoalscited by 16
- 2022Learning to Generalize with Object-centric Agents in the Open World Survival Game Craftercited by 13
- 2025CTRL-O: Language-Controllable Object-Centric Visual Representation Learningcited by 12
12 most cited of 30 All 30 (JSON)
- 2026Semantic Slots for Video Object-Centric Learning21 Aug 2026
- 2026OCRA: Object-Centric Learning with 3D and Tactile Priors for Human-to-Robot Action Transfer15 Mar 2026
- 2026Hierarchical Entity-centric Reinforcement Learning with Factored Subgoal Diffusion2 Feb 2026
- 2026QASA: Quality-Guided K-Adaptive Slot Attention for Unsupervised Object-Centric Learning19 Jan 2026
- 2025CTRL-O: Language-Controllable Object-Centric Visual Representation Learning27 Mar 2025
- 2024EC-Diffuser: Multi-Object Manipulation via Entity-Centric Behavior Generation25 Dec 2024
- 2024Temporally Consistent Object-Centric Learning by Contrasting Slots18 Dec 2024
- 2024Object-centric proto-symbolic behavioural reasoning from pixels26 Nov 2024
- 2024SkiLD: Unsupervised Skill Discovery Guided by Factor Interactions24 Oct 2024
- 2024Object-Centric Temporal Consistency via Conditional Autoregressive Inductive Biases21 Oct 2024
- 2024Zero-Shot Object-Centric Representation Learning17 Aug 2024
- 2024Fine-Grained Causal Dynamics Learning with Quantization for Improving Robustness in Reinforcement Learning5 Jun 2024
12 newest of 30 All 30 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2026More Structure, Not More Capacity: Object-Centric Representations for Visuomotor Imitation Learning
- 2024Learning to Compose: Improving Object Centric Learning by Injecting Compositionality
- 2024Efficient Exploration and Discriminative World Model Learning with an Object-Centric Abstraction
- 2025Object-Centric Action-Enhanced Representations for Robot Visuo-Motor Policy Learning
- 2024Learning from 10 Demos: Generalisable and Sample-Efficient Policy Learning with Oriented Affordance Frames
- 2024Vision-Language Models Provide Promptable Representations for Reinforcement Learning
- 2025Iterative Compositional Data Generation for Robot Control
- 2019Supervise Thyself: Examining Self-Supervised Representations in Interactive Environments