FOCUS: Object-Centric World Models for Robotics Manipulation
On the map: World models → Embodied and robot world models
Abstract (arXiv)
Understanding the world in terms of objects and the possible interplays with them is an important cognition ability, especially in robotics manipulation, where many tasks require robot-object interactions. However, learning such a structured world model, which specifically captures entities and relationships, remains a challenging and underexplored problem. To address this, we propose FOCUS, a model-based agent that learns an object-centric world model. Thanks to a novel exploration bonus that stems from the object-centric representation, FOCUS can be deployed on robotics manipulation tasks to explore object interactions more easily. Evaluating our approach on manipulation tasks across different settings, we show that object-centric world models allow the agent to solve tasks more efficiently and enable consistent exploration of robot-object interactions. Using a Franka Emika robot arm, we also showcase how FOCUS could be adopted in real-world settings.
Key facts. Cited by 17 papers in the map (data through 7 Oct 2026). Placed in Embodied and robot world models, one of 11 directions in World models. Builds on 42 of its 61 references that are in the map. First posted on arXiv 5 Jul 2023.
Builds on
42 of 61 references.
Closest ancestors · 3
Same direction, newest first
- 2023Multi-View Masked World Models for Visual Robotic Manipulation5 Feb
- 2022Masked World Models for Visual Control28 Jun
- 2019Solving Rubik's Cube with a Robot Hand16 Oct
Other references · 39
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 11,979
- 2023Segment Anythingcited by 8,158
- 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,635
- 2018Addressing Function Approximation Error in Actor-Critic Methodscited by 2,302
- 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,598
- 2016Concrete Problems in AI Safetycited by 1,582
- 2018World Modelscited by 1,444
- 2019Mastering Atari, Go, Chess and Shogi by Planning with a Learned Modelcited by 1,327
- 2014On the Properties of Neural Machine Translation: Encoder-Decoder Approachescited by 1,303
- 2018Learning Latent Dynamics for Planning from Pixelscited by 1,221
- 2023Mastering Diverse Domains through World Modelscited by 1,193
- 2017Curiosity-driven Exploration by Self-supervised Predictioncited by 1,173
27 more
- 2020Mastering Atari with Discrete World Modelscited by 990
- 2020Object-Centric Learning with Slot Attentioncited by 865
- 2015End-to-End Training of Deep Visuomotor Policiescited by 847
- 2018QT-Opt: Scalable Deep Reinforcement Learning for Vision-Based Robotic Manipulationcited by 730
- 2020robosuite: A Modular Simulation Framework and Benchmark for Robot Learningcited by 484
- 2019MONet: Unsupervised Scene Decomposition and Representationcited by 388
- 2022Temporal Difference Learning for Model Predictive Controlcited by 375
- 2022DayDreamer: World Models for Physical Robot Learningcited by 372
- 2020Planning to Explore via Self-Supervised World Modelscited by 365
- 2019Multi-Object Representation Learning with Iterative Variational Inferencecited by 359
- 2018Large-Scale Study of Curiosity-Driven Learningcited by 277
- 2019Self-Supervised Exploration via Disagreementcited by 263
- 2019Contrastive Learning of Structured World Modelscited by 206
- 2023Segment and Track Anythingcited by 193
- 2021Behavior From the Void: Unsupervised Active Pre-Trainingcited by 173
- 2022SlotFormer: Unsupervised Visual Dynamics Simulation with Object-Centric Modelscited by 146
- 2021URLB: Unsupervised Reinforcement Learning Benchmarkcited by 138
- 2021State Entropy Maximization with Random Encoders for Efficient Explorationcited by 107
- 2021The Distracting Control Suite -- A Challenging Benchmark for Reinforcement Learning from Pixelscited by 97
- 2021Generalization and Robustness Implications in Object-Centric Learningcited by 81
- 2021DreamerPro: Reconstruction-Free Model-Based Reinforcement Learning with Prototypical Representationscited by 73
- 2021Beyond Pick-and-Place: Tackling Robotic Stacking of Diverse Shapescited by 70
- 2017Data-efficient Deep Reinforcement Learning for Dexterous Manipulationcited by 61
- 2021AW-Opt: Learning Robotic Skills with Imitation and Reinforcement at Scalecited by 43
- 2021Curiosity-Driven Exploration via Latent Bayesian Surprisecited by 22
- 2022Mastering the Unsupervised Reinforcement Learning Benchmark from Pixelscited by 20
- 2021Touch-based Curiosity for Sparse-Reward Taskscited by 6
Built on by
Citing papers per year: 2023: 1; 2024: 10; 2025: 4; 2026: 2.
17 papers.
Sort:
By area:
- 2023MUVO: A Multimodal Generative World Model for Autonomous Driving with Geometric Representationscited by 33
- 2024Entity-Centric Reinforcement Learning for Object Manipulation from Pixelscited by 27
- 2026OA-WAM: Object-Addressable World Action Model for Robust Robot Manipulationcited by 25
- 2024World Model-based Perception for Visual Legged Locomotioncited by 17
- 2024Composing Pre-Trained Object-Centric Representations for Robotics From "What" and "Where" Foundation Modelscited by 16
- 2025PlaySlot: Learning Inverse Latent Dynamics for Controllable Object-Centric Video Prediction and Planningcited by 13
- 2025Dyn-O: Building Structured World Models with Object-Centric Representationscited by 12
- 2024EC-Diffuser: Multi-Object Manipulation via Entity-Centric Behavior Generationcited by 8
- 2024Recasting Generic Pretrained Vision Transformers As Object-Centric Scene Encoders For Manipulation Policiescited by 6
- 2025Focusing on What Matters: Object-Agent-centric Tokenization for Vision Language Action modelscited by 6
- 2024Object-Centric Temporal Consistency via Conditional Autoregressive Inductive Biasescited by 5
- 2024Make the Pertinent Salient: Task-Relevant Reconstruction for Visual Control with Distractionscited by 4
5 more
- 2024Zero-Shot Generalization of Vision-Based RL Without Data Augmentationcited by 3
- 2024Physically Interpretable World Models via Weakly Supervised Representation Learningcited by 3
- 2026A Definition and Roadmap for World Modelscited by 3
- 2024Grouped Discrete Representation Guides Object-Centric Learningcited by 2
- 2025Learning to Grasp Anything by Playing with Random Toyscited by 0
- 2026A Definition and Roadmap for World Models7 Jul 2026
- 2026OA-WAM: Object-Addressable World Action Model for Robust Robot Manipulation7 May 2026
- 2025Learning to Grasp Anything by Playing with Random Toys14 Oct 2025
- 2025Focusing on What Matters: Object-Agent-centric Tokenization for Vision Language Action models28 Sep 2025
- 2025Dyn-O: Building Structured World Models with Object-Centric Representations4 Jul 2025
- 2025PlaySlot: Learning Inverse Latent Dynamics for Controllable Object-Centric Video Prediction and Planning11 Feb 2025
- 2024EC-Diffuser: Multi-Object Manipulation via Entity-Centric Behavior Generation25 Dec 2024
- 2024Physically Interpretable World Models via Weakly Supervised Representation Learning17 Dec 2024
- 2024Object-Centric Temporal Consistency via Conditional Autoregressive Inductive Biases21 Oct 2024
- 2024Make the Pertinent Salient: Task-Relevant Reconstruction for Visual Control with Distractions13 Oct 2024
- 2024Zero-Shot Generalization of Vision-Based RL Without Data Augmentation9 Oct 2024
- 2024World Model-based Perception for Visual Legged Locomotion25 Sep 2024
5 more
- 2024Grouped Discrete Representation Guides Object-Centric Learning1 Jul 2024
- 2024Recasting Generic Pretrained Vision Transformers As Object-Centric Scene Encoders For Manipulation Policies24 May 2024
- 2024Composing Pre-Trained Object-Centric Representations for Robotics From "What" and "Where" Foundation Models20 Apr 2024
- 2024Entity-Centric Reinforcement Learning for Object Manipulation from Pixels1 Apr 2024
- 2023MUVO: A Multimodal Generative World Model for Autonomous Driving with Geometric Representations20 Nov 2023
- 2023MUVO: A Multimodal Generative World Model for Autonomous Driving with Geometric Representationscited by 33
- 2026OA-WAM: Object-Addressable World Action Model for Robust Robot Manipulationcited by 25
- 2024World Model-based Perception for Visual Legged Locomotioncited by 17
- 2025Dyn-O: Building Structured World Models with Object-Centric Representationscited by 12
- 2024Physically Interpretable World Models via Weakly Supervised Representation Learningcited by 3
- 2026A Definition and Roadmap for World Modelscited by 3
- 2026A Definition and Roadmap for World Models7 Jul 2026
- 2026OA-WAM: Object-Addressable World Action Model for Robust Robot Manipulation7 May 2026
- 2025Dyn-O: Building Structured World Models with Object-Centric Representations4 Jul 2025
- 2024Physically Interpretable World Models via Weakly Supervised Representation Learning17 Dec 2024
- 2024World Model-based Perception for Visual Legged Locomotion25 Sep 2024
- 2023MUVO: A Multimodal Generative World Model for Autonomous Driving with Geometric Representations20 Nov 2023
- 2024Entity-Centric Reinforcement Learning for Object Manipulation from Pixelscited by 27
- 2024Composing Pre-Trained Object-Centric Representations for Robotics From "What" and "Where" Foundation Modelscited by 16
- 2025PlaySlot: Learning Inverse Latent Dynamics for Controllable Object-Centric Video Prediction and Planningcited by 13
- 2024EC-Diffuser: Multi-Object Manipulation via Entity-Centric Behavior Generationcited by 8
- 2024Recasting Generic Pretrained Vision Transformers As Object-Centric Scene Encoders For Manipulation Policiescited by 6
- 2024Object-Centric Temporal Consistency via Conditional Autoregressive Inductive Biasescited by 5
- 2024Make the Pertinent Salient: Task-Relevant Reconstruction for Visual Control with Distractionscited by 4
- 2024Zero-Shot Generalization of Vision-Based RL Without Data Augmentationcited by 3
- 2024Grouped Discrete Representation Guides Object-Centric Learningcited by 2
- 2025Learning to Grasp Anything by Playing with Random Toyscited by 0
- 2025Learning to Grasp Anything by Playing with Random Toys14 Oct 2025
- 2025PlaySlot: Learning Inverse Latent Dynamics for Controllable Object-Centric Video Prediction and Planning11 Feb 2025
- 2024EC-Diffuser: Multi-Object Manipulation via Entity-Centric Behavior Generation25 Dec 2024
- 2024Object-Centric Temporal Consistency via Conditional Autoregressive Inductive Biases21 Oct 2024
- 2024Make the Pertinent Salient: Task-Relevant Reconstruction for Visual Control with Distractions13 Oct 2024
- 2024Zero-Shot Generalization of Vision-Based RL Without Data Augmentation9 Oct 2024
- 2024Grouped Discrete Representation Guides Object-Centric Learning1 Jul 2024
- 2024Recasting Generic Pretrained Vision Transformers As Object-Centric Scene Encoders For Manipulation Policies24 May 2024
- 2024Composing Pre-Trained Object-Centric Representations for Robotics From "What" and "Where" Foundation Models20 Apr 2024
- 2024Entity-Centric Reinforcement Learning for Object Manipulation from Pixels1 Apr 2024
Similar, not cited
Close in wording, no citation either way
- 2024Representing Positional Information in Generative World Models for Object Manipulation
- 2025Object-Centric World Model for Language-Guided Manipulation
- 2025A Step Toward World Models: A Survey on Robotic Manipulation
- 2022Binding Actions to Objects in World Models
- 2026World Models for Robotic Manipulation: A Survey
- 2022Factored World Models for Zero-Shot Generalization in Robotic Manipulation
- 2025OmniManip: Towards General Robotic Manipulation via Object-Centric Interaction Primitives as Spatial Constraints
- 2026Unifying Object-Centric World Models and Diffusion Policy: A Hierarchical Framework for Multi-Stage Robotic Tasks