SPACE: Unsupervised Object-Oriented Scene Representation via Spatial Attention and Decomposition
Not on the map. Why?
Abstract (arXiv)
The ability to decompose complex multi-object scenes into meaningful abstractions like objects is fundamental to achieve higher-level cognition. Previous approaches for unsupervised object-oriented scene representation learning are either based on spatial-attention or scene-mixture approaches and limited in scalability which is a main obstacle towards modeling real-world scenes. In this paper, we propose a generative latent variable model, called SPACE, that provides a unified probabilistic modeling framework that combines the best of spatial-attention and scene-mixture approaches. SPACE can explicitly provide factorized object representations for foreground objects while also decomposing background segments of complex morphology. Previous models are good at either of these, but not both. SPACE also resolves the scalability problems of previous methods by incorporating parallel spatial-attention and thus is applicable to scenes with a large number of objects without performance degradations. We show through experiments on Atari and 3D-Rooms that SPACE achieves the above properties consistently in comparison to SPAIR, IODINE, and GENESIS. Results of our experiments can be found on our project website: https://sites.google.com/view/space-project-page
Key facts. Cited by 190 papers in the map (data through 7 Oct 2026). First posted on arXiv 8 Jan 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 36; 2022: 39; 2023: 44; 2024: 37; 2025: 14; 2026: 20.
190 papers.
Sort:
By area:
- 2021Conditional Object-Centric Learning from Videocited by 247
- 2022Bridging the Gap to Real-World Object-Centric Learningcited by 219
- 2021Towards Causal Representation Learningcited by 207
- 2022Deep Spectral Methods: A Surprisingly Strong Baseline for Unsupervised Semantic Segmentation and Localizationcited by 150
- 2021Illiterate DALL-E Learns to Composecited by 147
- 2022SlotFormer: Unsupervised Visual Dynamics Simulation with Object-Centric Modelscited by 146
- 2022Simple Unsupervised Object-Centric Learning for Complex and Naturalistic Videoscited by 139
- 2021GENESIS-V2: Inferring Unordered Object Representations without Iterative Refinementcited by 109
- 2021Unsupervised Discovery of Object Radiance Fieldscited by 100
- 2021ClevrTex: A Texture-Rich Benchmark for Unsupervised Multi-Object Segmentationcited by 93
- 2022Object Scene Representation Transformercited by 90
- 2023Object-Centric Slot Diffusioncited by 90
38 more
- 2021Decomposing 3D Scenes into Objects via Unsupervised Volume Segmentationcited by 88
- 2023SlotDiffusion: Object-Centric Generative Modeling with Diffusion Modelscited by 85
- 2021Generalization and Robustness Implications in Object-Centric Learningcited by 81
- 2023Object-Centric Learning for Real-World Videos by Predicting Temporal Feature Similaritiescited by 78
- 2021SIMONe: View-Invariant, Temporally-Abstracted Object Representations via Unsupervised Video Decompositioncited by 69
- 2022Improving Object-centric Learning with Query Optimizationcited by 60
- 2021Unsupervised Part Discovery from Contrastive Reconstructioncited by 57
- 2023Additive Decoders for Latent Variables Identification and Cartesian-Product Extrapolationcited by 56
- 2023Invariant Slot Attention: Object Discovery with Slot-Centric Reference Framescited by 51
- 2021Unsupervised Layered Image Decomposition into Object Prototypescited by 50
- 2023An Investigation into Pre-Training Object-Centric Representations for Reinforcement Learningcited by 49
- 2023Self-supervised Object-Centric Learning for Videoscited by 49
- 2023SPOT: Self-Training with Patch-Order Permutation for Object-Centric Learning with Autoregressive Transformerscited by 49
- 2023Read and Reap the Rewards: Learning to Play Atari with the Help of Instruction Manualscited by 45
- 2023Provably Learning Object-Centric Representationscited by 45
- 2023Interpretable and Explainable Logical Policies via Neurally Guided Symbolic Abstractioncited by 45
- 2022Discovering Objects that Can Movecited by 44
- 2022Guess What Moves: Unsupervised Video and Image Segmentation by Anticipating Motioncited by 42
- 2022Compositional Generalization in Unsupervised Compositional Representation Learning: A Study on Disentanglement and Emergent Languagecited by 42
- 2023Provable Compositional Generalization for Object-Centric Learningcited by 41
- 2022Complex-Valued Autoencoders for Object Discoverycited by 40
- 2021Learning Object-Centric Representations of Multi-Object Scenes from Multiple Viewscited by 37
- 2022Neural Systematic Bindercited by 37
- 2021Efficient Iterative Amortized Inference for Learning Symmetric and Disentangled Multi-Object Representationscited by 36
- 2021MarioNette: Self-Supervised Sprite Learningcited by 35
- 2022VAEL: Bridging Variational Autoencoders and Probabilistic Logic Programmingcited by 31
- 2023OCAtari: Object-Centric Atari 2600 Reinforcement Learning Environmentscited by 30
- 2021Unsupervised Object Learning via Common Fatecited by 29
- 2023Rotating Features for Object Discoverycited by 29
- 2022Unsupervised Discovery and Composition of Object Light Fieldscited by 28
- 2022Compositional Scene Representation Learning via Reconstruction: A Surveycited by 27
- 2021Unsupervised Foreground Extraction via Deep Region Competitioncited by 26
- 2022Reasoning with Scene Graphs for Robot Planning under Partial Observabilitycited by 25
- 2023Object Discovery from Motion-Guided Tokenscited by 25
- 2021Generative Video Transformer: Can Objects be the Words?cited by 24
- 2021Compositional Transformers for Scene Generationcited by 24
- 2021Interactive Disentanglement: Learning Concepts by Interacting with their Prototype Representationscited by 24
- 2021Adaptive Rational Activations to Boost Deep Reinforcement Learningcited by 23
50 most cited of 190 All 190 (JSON)
- 2026iSEE: Object Permanence Through Self-Supervision1 Oct 2026
- 2026ParticleSplat: Self-supervised Object-centric Latent Particle Splatting16 Sep 2026
- 2026SlotDiT: Object-Centric Representations for Diffusion Transformers15 Sep 2026
- 2026Bottom-up Modeling of Repeated Elements via Single Image Analysis-by-Synthesis7 Sep 2026
- 2026Unsupervised Learning of Cell Instances with Generative Routing Pyramids17 Aug 2026
- 20263D-DLP: Self-Supervised 3D Object-Centric Scene Representation Learning17 Jun 2026
- 2026BRICKS-WM: Building Reusability via Interface Composition Kinetics for Structured World Models15 Jun 2026
- 2026TSA: Temporal Slot Activation for Persistent Object-Centric Video Representation10 Jun 2026
- 2026Dual-State Slot Attention: Decoupling Appearance and Identity for Video Object-Centric Learning10 Jun 2026
- 2026Deep Sprite-based Image Models: An Analysis21 Apr 2026
- 2026GRAIL: Autonomous Concept Grounding for Neuro-Symbolic Reinforcement Learning18 Apr 2026
- 2026Reconstruction-Guided Slot Curriculum: Addressing Object Over-Fragmentation in Video Object-Centric Learning24 Mar 2026
38 more
- 2026OCRA: Object-Centric Learning with 3D and Tactile Priors for Human-to-Robot Action Transfer15 Mar 2026
- 2026Latent Particle World Models: Self-supervised Object-centric Stochastic Dynamics Modeling4 Mar 2026
- 2026ORGAN: Object-Centric Representation Learning using Cycle Consistent Generative Adversarial Networks2 Mar 2026
- 2026SIGMA: Selective-Interleaved Generation with Multi-Attribute Tokens7 Feb 2026
- 2026MUFASA: A Multi-Layer Framework for Slot Attention7 Feb 2026
- 2026Evaluating Object-Centric Models beyond Object Discovery7 Feb 2026
- 2026From Vicious to Virtuous Cycles: Synergistic Representation Learning for Unsupervised Video Object-Centric Learning3 Feb 2026
- 2026Hierarchical Entity-centric Reinforcement Learning with Factored Subgoal Diffusion2 Feb 2026
- 2025Disentangled Representation Learning via Modular Compositional Bias24 Oct 2025
- 2025Object-Centric Case-Based Reasoning via Argumentation30 Sep 2025
- 2025Motion-Refined DINOSAUR for Unsupervised Multi-Object Discovery2 Sep 2025
- 2025Identifiable Object Representations under Spatial Ambiguities9 Jun 2025
- 2025CoLa: Chinese Character Decomposition with Compositional Latent Components4 Jun 2025
- 2025unMORE: Unsupervised Multi-Object Segmentation via Center-Boundary Reasoning2 Jun 2025
- 2025Object Centric Concept Bottlenecks30 May 2025
- 2025Hierarchical Compact Clustering Attention (COCA) for Unsupervised Object-Centric Learning4 May 2025
- 2025Deep Reinforcement Learning via Object-Centric Attention3 Apr 2025
- 2025TextOCVP: Object-Centric Video Prediction with Language Guidance17 Feb 2025
- 2025A Compressive-Expressive Communication Framework for Compositional Representations31 Jan 2025
- 2025Slot-Guided Adaptation of Pre-trained Diffusion Models for Object-Centric Learning and Compositional Generation27 Jan 2025
- 2025Dreamweaver: Learning Compositional World Models from Pixels24 Jan 2025
- 2025Slot-BERT: Self-supervised Object Discovery in Surgical Video21 Jan 2025
- 2024DeSplat: Decomposed Gaussian Splatting for Distractor-Free Rendering29 Nov 2024
- 2024Unsupervised Object Discovery: A Comprehensive Survey and Unified Taxonomy30 Oct 2024
- 2024Causality-Driven Audits of Model Robustness30 Oct 2024
- 2024Learning Global Object-Centric Representations via Disentangled Slot Attention24 Oct 2024
- 2024Interpretable end-to-end Neurosymbolic Reinforcement Learning agents18 Oct 2024
- 2024BlendRL: A Framework for Merging Symbolic and Neural Policy Learning15 Oct 2024
- 2024Next state prediction gives rise to entangled, yet compositional representations of objects7 Oct 2024
- 2024Simplified priors for Object-Centric Learning1 Oct 2024
- 2024Zero-Shot Object-Centric Representation Learning17 Aug 2024
- 2024Historical Printed Ornaments: Dataset and Tasks16 Aug 2024
- 2024SlotLifter: Slot-guided Feature Lifting for Learning Object-centric Radiance Fields13 Aug 2024
- 2024Exploring the Effectiveness of Object-Centric Representations in Visual Question Answering: Comparative Insights with Foundation Models22 Jul 2024
- 2024Rethinking Image-to-Video Adaptation: An Object-centric Perspective9 Jul 2024
- 2024Attention Normalization Impacts Cardinality Generalization in Slot Attention4 Jul 2024
- 2024Learning Disentangled Representation in Object-Centric Models for Visual Dynamics Prediction via Transformers3 Jul 2024
- 2024OCALM: Object-Centric Assessment with Language Models24 Jun 2024
50 newest of 190 All 190 (JSON)
- 2021Conditional Object-Centric Learning from Videocited by 247
- 2021Structured World Belief for Reinforcement Learning in POMDPcited by 21
- 2026Latent Particle World Models: Self-supervised Object-centric Stochastic Dynamics Modelingcited by 11
- 2023Imagine the Unseen World: A Benchmark for Systematic Generalization in Visual World Modelscited by 6
- 2023Relational Object-Centric Actor-Criticcited by 4
- 2024Compete and Compose: Learning Independent Mechanisms for Modular World Modelscited by 4
- 2025Dreamweaver: Learning Compositional World Models from Pixelscited by 4
- 2024Transformers and Slot Encoding for Sample Efficient Physical World Modellingcited by 0
- 2026BRICKS-WM: Building Reusability via Interface Composition Kinetics for Structured World Modelscited by 0
- 2026BRICKS-WM: Building Reusability via Interface Composition Kinetics for Structured World Models15 Jun 2026
- 2026Latent Particle World Models: Self-supervised Object-centric Stochastic Dynamics Modeling4 Mar 2026
- 2025Dreamweaver: Learning Compositional World Models from Pixels24 Jan 2025
- 2024Transformers and Slot Encoding for Sample Efficient Physical World Modelling30 May 2024
- 2024Compete and Compose: Learning Independent Mechanisms for Modular World Models23 Apr 2024
- 2023Imagine the Unseen World: A Benchmark for Systematic Generalization in Visual World Models15 Nov 2023
- 2023Relational Object-Centric Actor-Critic26 Oct 2023
- 2021Conditional Object-Centric Learning from Video24 Nov 2021
- 2021Structured World Belief for Reinforcement Learning in POMDP19 Jul 2021
- 2022VAEL: Bridging Variational Autoencoders and Probabilistic Logic Programmingcited by 31
- 2024BlendRL: A Framework for Merging Symbolic and Neural Policy Learningcited by 14
- 2024Learning to Compose: Improving Object Centric Learning by Injecting Compositionalitycited by 13
- 2024Neural Language of Thought Modelscited by 9
- 2022Towards Unsupervised Visual Reasoning: Do Off-The-Shelf Features Know How to Reason?cited by 1
- 2025Object-Centric Case-Based Reasoning via Argumentationcited by 1
- 2026GRAIL: Autonomous Concept Grounding for Neuro-Symbolic Reinforcement Learningcited by 0
- 2026GRAIL: Autonomous Concept Grounding for Neuro-Symbolic Reinforcement Learning18 Apr 2026
- 2025Object-Centric Case-Based Reasoning via Argumentation30 Sep 2025
- 2024BlendRL: A Framework for Merging Symbolic and Neural Policy Learning15 Oct 2024
- 2024Learning to Compose: Improving Object Centric Learning by Injecting Compositionality1 May 2024
- 2024Neural Language of Thought Models2 Feb 2024
- 2022Towards Unsupervised Visual Reasoning: Do Off-The-Shelf Features Know How to Reason?20 Dec 2022
- 2022VAEL: Bridging Variational Autoencoders and Probabilistic Logic Programming7 Feb 2022
- 2023Object-Centric Slot Diffusioncited by 90
- 2023SlotDiffusion: Object-Centric Generative Modeling with Diffusion Modelscited by 85
- 2025Slot-Guided Adaptation of Pre-trained Diffusion Models for Object-Centric Learning and Compositional Generationcited by 13
- 2026SIGMA: Selective-Interleaved Generation with Multi-Attribute Tokenscited by 2
- 2026SlotDiT: Object-Centric Representations for Diffusion Transformerscited by 0
- 2026SlotDiT: Object-Centric Representations for Diffusion Transformers15 Sep 2026
- 2026SIGMA: Selective-Interleaved Generation with Multi-Attribute Tokens7 Feb 2026
- 2025Slot-Guided Adaptation of Pre-trained Diffusion Models for Object-Centric Learning and Compositional Generation27 Jan 2025
- 2023SlotDiffusion: Object-Centric Generative Modeling with Diffusion Models18 May 2023
- 2023Object-Centric Slot Diffusion20 Mar 2023
- 2024Slot State Space Modelscited by 17
- 2024Slot State Space Models18 Jun 2024
- 2022Bridging the Gap to Real-World Object-Centric Learningcited by 219
- 2021Towards Causal Representation Learningcited by 207
- 2022Deep Spectral Methods: A Surprisingly Strong Baseline for Unsupervised Semantic Segmentation and Localizationcited by 150
- 2021Illiterate DALL-E Learns to Composecited by 147
- 2022SlotFormer: Unsupervised Visual Dynamics Simulation with Object-Centric Modelscited by 146
- 2022Simple Unsupervised Object-Centric Learning for Complex and Naturalistic Videoscited by 139
- 2021GENESIS-V2: Inferring Unordered Object Representations without Iterative Refinementcited by 109
- 2021Unsupervised Discovery of Object Radiance Fieldscited by 100
- 2021ClevrTex: A Texture-Rich Benchmark for Unsupervised Multi-Object Segmentationcited by 93
- 2022Object Scene Representation Transformercited by 90
- 2021Decomposing 3D Scenes into Objects via Unsupervised Volume Segmentationcited by 88
- 2021Generalization and Robustness Implications in Object-Centric Learningcited by 81
12 most cited of 167 All 167 (JSON)
- 2026iSEE: Object Permanence Through Self-Supervision1 Oct 2026
- 2026ParticleSplat: Self-supervised Object-centric Latent Particle Splatting16 Sep 2026
- 2026Bottom-up Modeling of Repeated Elements via Single Image Analysis-by-Synthesis7 Sep 2026
- 2026Unsupervised Learning of Cell Instances with Generative Routing Pyramids17 Aug 2026
- 20263D-DLP: Self-Supervised 3D Object-Centric Scene Representation Learning17 Jun 2026
- 2026TSA: Temporal Slot Activation for Persistent Object-Centric Video Representation10 Jun 2026
- 2026Dual-State Slot Attention: Decoupling Appearance and Identity for Video Object-Centric Learning10 Jun 2026
- 2026Deep Sprite-based Image Models: An Analysis21 Apr 2026
- 2026Reconstruction-Guided Slot Curriculum: Addressing Object Over-Fragmentation in Video Object-Centric Learning24 Mar 2026
- 2026OCRA: Object-Centric Learning with 3D and Tactile Priors for Human-to-Robot Action Transfer15 Mar 2026
- 2026ORGAN: Object-Centric Representation Learning using Cycle Consistent Generative Adversarial Networks2 Mar 2026
- 2026MUFASA: A Multi-Layer Framework for Slot Attention7 Feb 2026
12 newest of 167 All 167 (JSON)