Attend, Infer, Repeat: Fast Scene Understanding with Generative Models
Not on the map. Why?
Abstract (arXiv)
We present a framework for efficient inference in structured image models that explicitly reason about objects. We achieve this by performing probabilistic inference using a recurrent neural network that attends to scene elements and processes them one at a time. Crucially, the model itself learns to choose the appropriate number of inference steps. We use this scheme to learn to perform inference in partially specified 2D models (variable-sized variational auto-encoders) and fully specified 3D models (probabilistic renderers). We show that such models learn to identify multiple objects - counting, locating and classifying the elements of a scene - without any supervision, e.g., decomposing 3D images with various numbers of objects in a single forward pass of a neural network. We further show that the networks produce accurate inferences when compared to supervised counterparts, and that their structure leads to improved generalization.
Key facts. Cited by 209 papers in the map (data through 14 Oct 2026). First posted on arXiv 28 Mar 2016.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 67; 2022: 47; 2023: 42; 2024: 26; 2025: 12; 2026: 15.
209 papers.
Sort:
By area:
- 2021Dynamic Neural Networks: A Surveycited by 408
- 2021Conditional Object-Centric Learning from Videocited by 248
- 2022Bridging the Gap to Real-World Object-Centric Learningcited by 220
- 2021Illiterate DALL-E Learns to Composecited by 147
- 2022Simple Unsupervised Object-Centric Learning for Complex and Naturalistic Videoscited by 139
- 2021Generative Adversarial Transformerscited by 120
- 2021GENESIS-V2: Inferring Unordered Object Representations without Iterative Refinementcited by 109
- 2021Unsupervised Discovery of Object Radiance Fieldscited by 101
- 2022Deep Image Deblurring: A Surveycited by 94
- 2021ClevrTex: A Texture-Rich Benchmark for Unsupervised Multi-Object Segmentationcited by 93
- 2023Object-Centric Slot Diffusioncited by 90
- 2021Decomposing 3D Scenes into Objects via Unsupervised Volume Segmentationcited by 88
38 more
- 2023SlotDiffusion: Object-Centric Generative Modeling with Diffusion Modelscited by 85
- 2021Generalization and Robustness Implications in Object-Centric Learningcited by 81
- 2021SetVAE: Learning Hierarchical Composition for Generative Modeling of Set-Structured Datacited by 78
- 2021Is Attention Better Than Matrix Decomposition?cited by 77
- 2021Neural Production Systems: Learning Rule-Governed Visual Dynamicscited by 75
- 2021Unsupervised Learning of Compositional Energy Conceptscited by 71
- 2021SIMONe: View-Invariant, Temporally-Abstracted Object Representations via Unsupervised Video Decompositioncited by 69
- 2021Differentiable Patch Selection for Image Recognitioncited by 68
- 2022Improving Object-centric Learning with Query Optimizationcited by 60
- 2021SemanticStyleGAN: Learning Compositional Generative Priors for Controllable Image Synthesis and Editingcited by 58
- 2023Additive Decoders for Latent Variables Identification and Cartesian-Product Extrapolationcited by 58
- 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
- 2021Learning Disentangled Representations in the Imaging Domaincited by 46
- 2022Discovering Objects that Can Movecited by 44
- 2022Complex-Valued Autoencoders for Object Discoverycited by 40
- 2022Deep Learning Methods for Abstract Visual Reasoning: A Survey on Raven's Progressive Matricescited 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
- 2021From Machine Learning to Robotics: Challenges and Opportunities for Embodied Intelligencecited by 35
- 2022What to look at and where: Semantic and Spatial Refined Transformer for detecting human-object interactionscited by 35
- 2024Adaptive Slot Attention: Object Discovery with Dynamic Slot Numbercited by 34
- 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
- 20213DP3: 3D Scene Perception via Probabilistic Programmingcited by 26
- 2022OGC: Unsupervised 3D Object Segmentation from Rigid Dynamics of Point Cloudscited by 25
- 2021Variable-rate discrete representation learningcited by 24
- 2021Generative Video Transformer: Can Objects be the Words?cited by 24
- 2021Compositional Transformers for Scene Generationcited by 24
50 most cited of 209 All 209 (JSON)
- 2026iSEE: Object Permanence Through Self-Supervision1 Oct 2026
- 2026Bottom-up Modeling of Repeated Elements via Single Image Analysis-by-Synthesis7 Sep 2026
- 2026MOSH-WM: Mask-Grounded Soft-Hamiltonian Dynamics for Object-Centric World Models24 Aug 2026
- 2026HSA: Hierarchical Slot Attention for Multi-granularity Scene-Decomposition9 Jul 2026
- 2026Mask-supervised Object-centric Representation Learning with LeJEPA2 Jul 2026
- 2026TSA: Temporal Slot Activation for Persistent Object-Centric Video Representation10 Jun 2026
- 2026GenMatter: Perceiving Physical Objects with Generative Matter Models24 Apr 2026
- 2026Deep Sprite-based Image Models: An Analysis21 Apr 2026
- 2026When Slots Compete: Slot Merging in Object-Centric Learning11 Mar 2026
- 2026ORGAN: Object-Centric Representation Learning using Cycle Consistent Generative Adversarial Networks2 Mar 2026
- 2026Communication-Inspired Tokenization for Structured Image Representations24 Feb 2026
- 2026MUFASA: A Multi-Layer Framework for Slot Attention7 Feb 2026
38 more
- 2026Improving Set Function Approximation with Quasi-Arithmetic Neural Networks4 Feb 2026
- 2026SHED Light on Segmentation for Dense Prediction30 Jan 2026
- 2026QASA: Quality-Guided K-Adaptive Slot Attention for Unsupervised Object-Centric Learning19 Jan 2026
- 2025Soft Geometric Inductive Bias for Object Centric Dynamics17 Dec 2025
- 2025Towards 3D Object-Centric Feature Learning for Semantic Scene Completion17 Nov 2025
- 2025Motion-Refined DINOSAUR for Unsupervised Multi-Object Discovery2 Sep 2025
- 2025Compositional Video Synthesis by Temporal Object-Centric Learning28 Jul 2025
- 2025unMORE: Unsupervised Multi-Object Segmentation via Center-Boundary Reasoning2 Jun 2025
- 2025Object Centric Concept Bottlenecks30 May 2025
- 2025CTRL-O: Language-Controllable Object-Centric Visual Representation Learning27 Mar 2025
- 2025Object-centric Binding in Contrastive Language-Image Pretraining19 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
- 2025Bridge the Inference Gaps of Neural Processes via Expectation Maximization4 Jan 2025
- 2024Object-centric proto-symbolic behavioural reasoning from pixels26 Nov 2024
- 2024Improving Viewpoint-Independent Object-Centric Representations through Active Viewpoint Selection1 Nov 2024
- 2024Unsupervised Object Discovery: A Comprehensive Survey and Unified Taxonomy30 Oct 2024
- 2024Learning Global Object-Centric Representations via Disentangled Slot Attention24 Oct 2024
- 2024Interpretable end-to-end Neurosymbolic Reinforcement Learning agents18 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
- 2024GLASS: Guided Latent Slot Diffusion for Object-Centric Learning25 Jul 2024
- 2024Exploring the Effectiveness of Object-Centric Representations in Visual Question Answering: Comparative Insights with Foundation Models22 Jul 2024
- 2024Learning Disentangled Representation in Object-Centric Models for Visual Dynamics Prediction via Transformers3 Jul 2024
- 2024Probabilistic Programming with Programmable Variational Inference22 Jun 2024
- 2024Slot State Space Models18 Jun 2024
- 2024Adaptive Slot Attention: Object Discovery with Dynamic Slot Number13 Jun 2024
- 2024Transformers and Slot Encoding for Sample Efficient Physical World Modelling30 May 2024
- 2024Recurrent Complex-Weighted Autoencoders for Unsupervised Object Discovery27 May 2024
- 2024Learning to Compose: Improving Object Centric Learning by Injecting Compositionality1 May 2024
- 2024SPARO: Selective Attention for Robust and Compositional Transformer Encodings for Vision24 Apr 2024
- 2024HiKER-SGG: Hierarchical Knowledge Enhanced Robust Scene Graph Generation18 Mar 2024
- 2024Unsupervised Discovery of Object-Centric Neural Fields12 Feb 2024
- 2024Binding Dynamics in Rotating Features8 Feb 2024
- 2024Neural Language of Thought Models2 Feb 2024
- 2024VONet: Unsupervised Video Object Learning With Parallel U-Net Attention and Object-wise Sequential VAE20 Jan 2024
50 newest of 209 All 209 (JSON)
- 2021Conditional Object-Centric Learning from Videocited by 248
- 2021Structured World Belief for Reinforcement Learning in POMDPcited by 21
- 2023Learning Dynamic Attribute-factored World Models for Efficient Multi-object Reinforcement Learningcited by 15
- 2022Active Predictive Coding: A Unified Neural Framework for Learning Hierarchical World Models for Perception and Planningcited by 10
- 2023Imagine the Unseen World: A Benchmark for Systematic Generalization in Visual World Modelscited by 6
- 2025Dreamweaver: Learning Compositional World Models from Pixelscited by 4
- 2024Transformers and Slot Encoding for Sample Efficient Physical World Modellingcited by 0
- 2026MOSH-WM: Mask-Grounded Soft-Hamiltonian Dynamics for Object-Centric World Modelscited by 0
- 2026MOSH-WM: Mask-Grounded Soft-Hamiltonian Dynamics for Object-Centric World Models24 Aug 2026
- 2025Dreamweaver: Learning Compositional World Models from Pixels24 Jan 2025
- 2024Transformers and Slot Encoding for Sample Efficient Physical World Modelling30 May 2024
- 2023Imagine the Unseen World: A Benchmark for Systematic Generalization in Visual World Models15 Nov 2023
- 2023Learning Dynamic Attribute-factored World Models for Efficient Multi-object Reinforcement Learning18 Jul 2023
- 2022Active Predictive Coding: A Unified Neural Framework for Learning Hierarchical World Models for Perception and Planning23 Oct 2022
- 2021Conditional Object-Centric Learning from Video24 Nov 2021
- 2021Structured World Belief for Reinforcement Learning in POMDP19 Jul 2021
- 2022Deep Learning Methods for Abstract Visual Reasoning: A Survey on Raven's Progressive Matricescited by 37
- 2022VAEL: Bridging Variational Autoencoders and Probabilistic Logic Programmingcited by 31
- 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
- 2023OC-NMN: Object-centric Compositional Neural Module Network for Generative Visual Analogical Reasoningcited by 0
- 2024Learning to Compose: Improving Object Centric Learning by Injecting Compositionality1 May 2024
- 2024Neural Language of Thought Models2 Feb 2024
- 2023OC-NMN: Object-centric Compositional Neural Module Network for Generative Visual Analogical Reasoning28 Oct 2023
- 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
- 2022Deep Learning Methods for Abstract Visual Reasoning: A Survey on Raven's Progressive Matrices28 Jan 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
- 2026Communication-Inspired Tokenization for Structured Image Representationscited by 0
- 2026Communication-Inspired Tokenization for Structured Image Representations24 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
- 2021Dynamic Neural Networks: A Surveycited by 408
- 2022Bridging the Gap to Real-World Object-Centric Learningcited by 220
- 2021Illiterate DALL-E Learns to Composecited by 147
- 2022Simple Unsupervised Object-Centric Learning for Complex and Naturalistic Videoscited by 139
- 2021Generative Adversarial Transformerscited by 120
- 2021GENESIS-V2: Inferring Unordered Object Representations without Iterative Refinementcited by 109
- 2021Unsupervised Discovery of Object Radiance Fieldscited by 101
- 2022Deep Image Deblurring: A Surveycited by 94
- 2021ClevrTex: A Texture-Rich Benchmark for Unsupervised Multi-Object Segmentationcited by 93
- 2021Decomposing 3D Scenes into Objects via Unsupervised Volume Segmentationcited by 88
- 2021Generalization and Robustness Implications in Object-Centric Learningcited by 81
- 2021SetVAE: Learning Hierarchical Composition for Generative Modeling of Set-Structured Datacited by 78
12 most cited of 186 All 186 (JSON)
- 2026iSEE: Object Permanence Through Self-Supervision1 Oct 2026
- 2026Bottom-up Modeling of Repeated Elements via Single Image Analysis-by-Synthesis7 Sep 2026
- 2026HSA: Hierarchical Slot Attention for Multi-granularity Scene-Decomposition9 Jul 2026
- 2026Mask-supervised Object-centric Representation Learning with LeJEPA2 Jul 2026
- 2026TSA: Temporal Slot Activation for Persistent Object-Centric Video Representation10 Jun 2026
- 2026GenMatter: Perceiving Physical Objects with Generative Matter Models24 Apr 2026
- 2026Deep Sprite-based Image Models: An Analysis21 Apr 2026
- 2026When Slots Compete: Slot Merging in Object-Centric Learning11 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
- 2026Improving Set Function Approximation with Quasi-Arithmetic Neural Networks4 Feb 2026
- 2026SHED Light on Segmentation for Dense Prediction30 Jan 2026
12 newest of 186 All 186 (JSON)