Slot Structured World Models
Abstract
The ability to perceive and reason about individual objects and their interactions is a goal to be achieved for building intelligent artificial systems. State-of-the-art approaches use a feedforward encoder to extract object embeddings and a latent graph neural network to model the interaction between these object embeddings. However, the feedforward encoder can not extract {\it object-centric} representations, nor can it disentangle multiple objects with similar appearance. To solve these issues, we introduce {\it Slot Structured World Models} (SSWM), a class of world models that combines an {\it object-centric} encoder (based on Slot Attention) with a latent graph-based dynamics model. We evaluate our method in the Spriteworld benchmark with simple rules of physical interaction, where Slot Structured World Models consistently outperform baselines on a range of (multi-step) prediction tasks with action-conditional object interactions. All code to reproduce paper experiments is available from \url{https://github.com/JonathanCollu/Slot-Structured-World-Models}.
Authors 4
-
Affiliation as printed
Leiden Institute of Advanced Computer Science , Leiden University , The Netherlands
-
Affiliation as printed
Leiden Institute of Advanced Computer Science , Leiden University , The Netherlands
-
Affiliation as printed
Leiden Institute of Advanced Computer Science , Leiden University , The Netherlands
-
Affiliation as printed
Leiden Institute of Advanced Computer Science , Leiden University , The Netherlands
Cited by 0 stored of 0
No patents citing this paper on Lens.org (checked 2026-10-11).