mlsanityData to 7 Oct 2026

Next state prediction gives rise to entangled, yet compositional representations of objects

arXiv 2410.04940 ↗First posted 7 Oct 2024cs.LG cs.CVcited by 0

Not on the map. Why?

Abstract (arXiv)

Compositional representations are thought to enable humans to generalize across combinatorially vast state spaces. Models with learnable object slots, which encode information about objects in separate latent codes, have shown promise for this type of generalization but rely on strong architectural priors. Models with distributed representations, on the other hand, use overlapping, potentially entangled neural codes, and their ability to support compositional generalization remains underexplored. In this paper we examine whether distributed models can develop linearly separable representations of objects, like slotted models, through unsupervised training on videos of object interactions. We show that, surprisingly, models with distributed representations often match or outperform models with object slots in downstream prediction tasks. Furthermore, we find that linearly separable object representations can emerge without object-centric priors, with auxiliary objectives like next-state prediction playing a key role. Finally, we observe that distributed models' object representations are never fully disentangled, even if they are linearly separable: Multiple objects can be encoded through partially overlapping neural populations while still being highly separable with a linear classifier. We hypothesize that maintaining partially shared codes enables distributed models to better compress object dynamics, potentially enhancing generalization.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 40 of its 62 references that are in the map. First posted on arXiv 7 Oct 2024.

Builds on

40 of 62 references.

References · 40

  1. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  3. 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 11,979
  4. 2018The Unreasonable Effectiveness of Deep Features as a Perceptual Metriccited by 11,029
  5. 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,339
  6. 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,893
  7. 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,712
  8. 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,635
  9. 2021Barlow Twins: Self-Supervised Learning via Redundancy Reductioncited by 1,818
  10. 2019Similarity of Neural Network Representations Revisitedcited by 1,695
  11. 2016CLEVR: A Diagnostic Dataset for Compositional Language and Elementary Visual Reasoningcited by 1,445
  12. 2018Challenging Common Assumptions in the Unsupervised Learning of Disentangled Representationscited by 929
28 more
  1. 2016Deep Variational Information Bottleneckcited by 915
  2. 2020Object-Centric Learning with Slot Attentioncited by 865
  3. 2020CURL: Contrastive Unsupervised Representations for Reinforcement Learningcited by 745
  4. 2018Disentangling by Factorisingcited by 614
  5. 2024The Platonic Representation Hypothesiscited by 362
  6. 2019Multi-Object Representation Learning with Iterative Variational Inferencecited by 359
  7. 2021Mastering Visual Continuous Control: Improved Data-Augmented Reinforcement Learningcited by 359
  8. 2020Data-Efficient Reinforcement Learning with Self-Predictive Representationscited by 309
  9. 2018Towards a Definition of Disentangled Representationscited by 294
  10. 2019Improving Sample Efficiency in Model-Free Reinforcement Learning from Imagescited by 268
  11. 2020On the Binding Problem in Artificial Neural Networkscited by 230
  12. 2022Bridging the Gap to Real-World Object-Centric Learningcited by 219
  13. 2019Contrastive Learning of Structured World Modelscited by 206
  14. 2019DeepMDP: Learning Continuous Latent Space Models for Representation Learningcited by 204
  15. 2020SPACE: Unsupervised Object-Oriented Scene Representation via Spatial Attention and Decompositioncited by 190
  16. 2022SAVi++: Towards End-to-End Object-Centric Learning from Real-World Videoscited by 165
  17. 2022SlotFormer: Unsupervised Visual Dynamics Simulation with Object-Centric Modelscited by 146
  18. 2018Learning Deep Disentangled Embeddings with the F-Statistic Losscited by 101
  19. 2021Generalization and Robustness Implications in Object-Centric Learningcited by 81
  20. 2023Object-Centric Learning for Real-World Videos by Predicting Temporal Feature Similaritiescited by 78
  21. 2020A Sober Look at the Unsupervised Learning of Disentangled Representations and their Evaluationcited by 52
  22. 2023Provably Learning Object-Centric Representationscited by 45
  23. 2023Provable Compositional Generalization for Object-Centric Learningcited by 41
  24. 2021MarioNette: Self-Supervised Sprite Learningcited by 35
  25. 2022Unsupervised Image Representation Learning with Deep Latent Particlescited by 19
  26. 2016Learning to generalize to new compositions in image understandingcited by 12
  27. 2023The Acquisition of Physical Knowledge in Generative Neural Networkscited by 3
  28. 2024Parallelized Spatiotemporal Bindingcited by 2

Built on by

Not cited yet.