Relational Neural Expectation Maximization: Unsupervised Discovery of Objects and their Interactions
Not on the map. Why?
Abstract (arXiv)
Common-sense physical reasoning is an essential ingredient for any intelligent agent operating in the real-world. For example, it can be used to simulate the environment, or to infer the state of parts of the world that are currently unobserved. In order to match real-world conditions this causal knowledge must be learned without access to supervised data. To address this problem we present a novel method that learns to discover objects and model their physical interactions from raw visual images in a purely unsupervised fashion. It incorporates prior knowledge about the compositional nature of human perception to factor interactions between object-pairs and learn efficiently. On videos of bouncing balls we show the superior modelling capabilities of our method compared to other unsupervised neural approaches that do not incorporate such prior knowledge. We demonstrate its ability to handle occlusion and show that it can extrapolate learned knowledge to scenes with different numbers of objects.
Key facts. Cited by 100 papers in the map (data through 7 Oct 2026). First posted on arXiv 28 Feb 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 43; 2022: 21; 2023: 21; 2024: 10; 2025: 4; 2026: 1.
100 papers.
Sort:
By area:
- 2021Conditional Object-Centric Learning from Videocited by 247
- 2021Towards Causal Representation Learningcited by 207
- 2021PredRNN: A Recurrent Neural Network for Spatiotemporal Predictive Learningcited by 181
- 2022SAVi++: Towards End-to-End Object-Centric Learning from Real-World Videoscited by 165
- 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
- 2021Decomposing 3D Scenes into Objects via Unsupervised Volume Segmentationcited by 88
- 2023SlotDiffusion: Object-Centric Generative Modeling with Diffusion Modelscited by 85
- 2021Coordination Among Neural Modules Through a Shared Global Workspacecited by 83
- 2021Generalization and Robustness Implications in Object-Centric Learningcited by 81
- 2023Object-Centric Learning for Real-World Videos by Predicting Temporal Feature Similaritiescited by 78
38 more
- 2021Neural Production Systems: Learning Rule-Governed Visual Dynamicscited by 75
- 2021Is Attention Better Than Matrix Decomposition?cited by 75
- 2021Unsupervised Learning of Compositional Energy Conceptscited by 71
- 2021Systematic Evaluation of Causal Discovery in Visual Model Based Reinforcement Learningcited by 53
- 2022Iso-Dream: Isolating and Leveraging Noncontrollable Visual Dynamics in World Modelscited by 53
- 2023Invariant Slot Attention: Object Discovery with Slot-Centric Reference Framescited by 51
- 2021Unsupervised Layered Image Decomposition into Object Prototypescited by 50
- 2022Object Representations as Fixed Points: Training Iterative Refinement Algorithms with Implicit Differentiationcited by 46
- 2023Provably Learning Object-Centric Representationscited by 45
- 2021iPOKE: Poking a Still Image for Controlled Stochastic Video Synthesiscited by 42
- 2021Discrete-Valued Neural Communicationcited by 39
- 2021Understanding Object Dynamics for Interactive Image-to-Video Synthesiscited by 38
- 2022Learning to Accelerate Partial Differential Equations via Latent Global Evolutioncited by 38
- 2022Neural Systematic Bindercited by 37
- 2021Efficient Iterative Amortized Inference for Learning Symmetric and Disentangled Multi-Object Representationscited by 36
- 2021RAIN: Reinforced Hybrid Attention Inference Network for Motion Forecastingcited by 36
- 2022PartAfford: Part-level Affordance Discovery from 3D Objectscited by 35
- 2022VAEL: Bridging Variational Autoencoders and Probabilistic Logic Programmingcited by 31
- 2024Temporally Consistent Object-Centric Learning by Contrasting Slotscited by 30
- 2021Unsupervised Object Learning via Common Fatecited by 29
- 2021Conditional Neural Relational Inference for Interacting Systemscited by 28
- 2022Dynamic-Group-Aware Networks for Multi-Agent Trajectory Prediction with Relational Reasoningcited by 28
- 2022Compositional Scene Representation Learning via Reconstruction: A Surveycited by 27
- 2021Unsupervised Foreground Extraction via Deep Region Competitioncited by 26
- 2021Generative Video Transformer: Can Objects be the Words?cited by 24
- 2021Dynamic Inference with Neural Interpreterscited by 24
- 2023Towards Unsupervised Object Detection From LiDAR Point Cloudscited by 22
- 2021Structured World Belief for Reinforcement Learning in POMDPcited by 21
- 2021Capturing the objects of vision with neural networkscited by 21
- 2023Object-centric architectures enable efficient causal representation learningcited by 21
- 2021Unsupervised Object-Based Transition Models for 3D Partially Observable Environmentscited by 19
- 2022Unsupervised Multi-object Segmentation by Predicting Probable Motion Patternscited by 19
- 2023Systematic Visual Reasoning through Object-Centric Relational Abstractioncited by 19
- 2023Shepherding Slots to Objects: Towards Stable and Robust Object-Centric Learningcited by 17
- 2024Slot State Space Modelscited by 17
- 2021Self-supervised Reinforcement Learning with Independently Controllable Subgoalscited by 16
- 2023Contrastive Training of Complex-Valued Autoencoders for Object Discoverycited by 16
- 2023Neural Relational Inference with Fast Modular Meta-learningcited by 16
50 most cited of 100 All 100 (JSON)
- 2026Disentangling Dynamical Systems: Causal Representation Learning Meets Local Sparse Attention15 Mar 2026
- 2025Physics-Inspired Spatial Temporal Graph Neural Networks for Predicting Industrial Chain Resilience22 Aug 2025
- 2025Predicting Video Slot Attention Queries from Random Slot-Feature Pairs2 Aug 2025
- 2025Identifiable Object Representations under Spatial Ambiguities9 Jun 2025
- 2025Dreamweaver: Learning Compositional World Models from Pixels24 Jan 2025
- 2024Temporally Consistent Object-Centric Learning by Contrasting Slots18 Dec 2024
- 2024SPARTAN: A Sparse Transformer World Model Attending to What Matters11 Nov 2024
- 2024Unsupervised Object Discovery: A Comprehensive Survey and Unified Taxonomy30 Oct 2024
- 2024Language Agents Meet Causality -- Bridging LLMs and Causal World Models25 Oct 2024
- 2024Learning Disentangled Representation in Object-Centric Models for Visual Dynamics Prediction via Transformers3 Jul 2024
- 2024Slot State Space Models18 Jun 2024
- 2024Compete and Compose: Learning Independent Mechanisms for Modular World Models23 Apr 2024
38 more
- 2024Explicitly Disentangled Representations in Object-Centric Learning18 Jan 2024
- 2024Slot Structured World Models8 Jan 2024
- 2024Unsupervised Object-Centric Learning from Multiple Unspecified Viewpoints3 Jan 2024
- 2023Unsupervised Musical Object Discovery from Audio13 Nov 2023
- 2023Object-Centric Learning with Slot Mixture Module8 Nov 2023
- 2023RigLSTM: Recurrent Independent Grid LSTM for Generalizable Sequence Learning3 Nov 2023
- 2023Towards Unsupervised Object Detection From LiDAR Point Clouds3 Nov 2023
- 2023Object-centric architectures enable efficient causal representation learning29 Oct 2023
- 2023OC-NMN: Object-centric Compositional Neural Module Network for Generative Visual Analogical Reasoning28 Oct 2023
- 2023Inferring Relational Potentials in Interacting Systems23 Oct 2023
- 2023Neural Relational Inference with Fast Modular Meta-learning10 Oct 2023
- 2023Object-Centric Learning for Real-World Videos by Predicting Temporal Feature Similarities7 Jun 2023
- 2023Systematic Visual Reasoning through Object-Centric Relational Abstraction4 Jun 2023
- 2023Revisiting Structured Variational Autoencoders25 May 2023
- 2023Contrastive Training of Complex-Valued Autoencoders for Object Discovery24 May 2023
- 2023Provably Learning Object-Centric Representations23 May 2023
- 2023SlotDiffusion: Object-Centric Generative Modeling with Diffusion Models18 May 2023
- 2023Shepherding Slots to Objects: Towards Stable and Robust Object-Centric Learning31 Mar 2023
- 2023Model-Based Reinforcement Learning with Isolated Imaginations27 Mar 2023
- 2023Neural Constraint Satisfaction: Hierarchical Abstraction for Combinatorial Generalization in Object Rearrangement20 Mar 2023
- 2023Reusable Slotwise Mechanisms21 Feb 2023
- 2023A Review of the Role of Causality in Developing Trustworthy AI Systems14 Feb 2023
- 2023Invariant Slot Attention: Object Discovery with Slot-Centric Reference Frames9 Feb 2023
- 2023The Construction of Reality in an AI: A Review3 Feb 2023
- 2022Dance of SNN and ANN: Solving binding problem by combining spike timing and reconstructive attention11 Nov 2022
- 2022Neural Systematic Binder2 Nov 2022
- 2022Unsupervised Multi-object Segmentation by Predicting Probable Motion Patterns21 Oct 2022
- 2022SlotFormer: Unsupervised Visual Dynamics Simulation with Object-Centric Models12 Oct 2022
- 2022Learning to Generalize with Object-centric Agents in the Open World Survival Game Crafter5 Aug 2022
- 2022Object Representations as Fixed Points: Training Iterative Refinement Algorithms with Implicit Differentiation2 Jul 2022
- 2022Dynamic-Group-Aware Networks for Multi-Agent Trajectory Prediction with Relational Reasoning27 Jun 2022
- 2022SAVi++: Towards End-to-End Object-Centric Learning from Real-World Videos15 Jun 2022
- 2022Learning to Accelerate Partial Differential Equations via Latent Global Evolution15 Jun 2022
- 2022ObPose: Leveraging Pose for Object-Centric Scene Inference and Generation in 3D7 Jun 2022
- 2022MS-RNN: A Flexible Multi-Scale Framework for Spatiotemporal Predictive Learning7 Jun 2022
- 2022Learning with Capsules: A Survey6 Jun 2022
- 2022Simple Unsupervised Object-Centric Learning for Complex and Naturalistic Videos27 May 2022
- 2022Iso-Dream: Isolating and Leveraging Noncontrollable Visual Dynamics in World Models27 May 2022
50 newest of 100 All 100 (JSON)
- 2021Conditional Object-Centric Learning from Videocited by 247
- 2022Iso-Dream: Isolating and Leveraging Noncontrollable Visual Dynamics in World Modelscited by 53
- 2021Structured World Belief for Reinforcement Learning in POMDPcited by 21
- 2024SPARTAN: A Sparse Transformer World Model Attending to What Matterscited by 10
- 2024Language Agents Meet Causality -- Bridging LLMs and Causal World Modelscited by 8
- 2024Slot Structured World Modelscited by 5
- 2023Model-Based Reinforcement Learning with Isolated Imaginationscited by 4
- 2024Compete and Compose: Learning Independent Mechanisms for Modular World Modelscited by 4
- 2025Dreamweaver: Learning Compositional World Models from Pixelscited by 4
- 2025Dreamweaver: Learning Compositional World Models from Pixels24 Jan 2025
- 2024SPARTAN: A Sparse Transformer World Model Attending to What Matters11 Nov 2024
- 2024Language Agents Meet Causality -- Bridging LLMs and Causal World Models25 Oct 2024
- 2024Compete and Compose: Learning Independent Mechanisms for Modular World Models23 Apr 2024
- 2024Slot Structured World Models8 Jan 2024
- 2023Model-Based Reinforcement Learning with Isolated Imaginations27 Mar 2023
- 2022Iso-Dream: Isolating and Leveraging Noncontrollable Visual Dynamics in World Models27 May 2022
- 2021Conditional Object-Centric Learning from Video24 Nov 2021
- 2021Structured World Belief for Reinforcement Learning in POMDP19 Jul 2021
- 2024Slot State Space Modelscited by 17
- 2024Slot State Space Models18 Jun 2024
- 2021Towards Causal Representation Learningcited by 207
- 2021PredRNN: A Recurrent Neural Network for Spatiotemporal Predictive Learningcited by 181
- 2022SAVi++: Towards End-to-End Object-Centric Learning from Real-World Videoscited by 165
- 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
- 2021Decomposing 3D Scenes into Objects via Unsupervised Volume Segmentationcited by 88
- 2021Coordination Among Neural Modules Through a Shared Global Workspacecited by 83
- 2021Generalization and Robustness Implications in Object-Centric Learningcited by 81
- 2023Object-Centric Learning for Real-World Videos by Predicting Temporal Feature Similaritiescited by 78
- 2021Neural Production Systems: Learning Rule-Governed Visual Dynamicscited by 75
- 2021Is Attention Better Than Matrix Decomposition?cited by 75
12 most cited of 85 All 85 (JSON)
- 2026Disentangling Dynamical Systems: Causal Representation Learning Meets Local Sparse Attention15 Mar 2026
- 2025Physics-Inspired Spatial Temporal Graph Neural Networks for Predicting Industrial Chain Resilience22 Aug 2025
- 2025Predicting Video Slot Attention Queries from Random Slot-Feature Pairs2 Aug 2025
- 2025Identifiable Object Representations under Spatial Ambiguities9 Jun 2025
- 2024Temporally Consistent Object-Centric Learning by Contrasting Slots18 Dec 2024
- 2024Unsupervised Object Discovery: A Comprehensive Survey and Unified Taxonomy30 Oct 2024
- 2024Learning Disentangled Representation in Object-Centric Models for Visual Dynamics Prediction via Transformers3 Jul 2024
- 2024Explicitly Disentangled Representations in Object-Centric Learning18 Jan 2024
- 2024Unsupervised Object-Centric Learning from Multiple Unspecified Viewpoints3 Jan 2024
- 2023Unsupervised Musical Object Discovery from Audio13 Nov 2023
- 2023Object-Centric Learning with Slot Mixture Module8 Nov 2023
- 2023RigLSTM: Recurrent Independent Grid LSTM for Generalizable Sequence Learning3 Nov 2023
12 newest of 85 All 85 (JSON)