Grounded Object Centric Learning

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Kori, Avinash, Locatello, Francesco, Ribeiro, Fabio De Sousa, Toni, Francesca, Glocker, Ben
Natura: Preprint
Pubblicazione: 2023
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866909082531659776
author Kori, Avinash
Locatello, Francesco
Ribeiro, Fabio De Sousa
Toni, Francesca
Glocker, Ben
author_facet Kori, Avinash
Locatello, Francesco
Ribeiro, Fabio De Sousa
Toni, Francesca
Glocker, Ben
contents The extraction of modular object-centric representations for downstream tasks is an emerging area of research. Learning grounded representations of objects that are guaranteed to be stable and invariant promises robust performance across different tasks and environments. Slot Attention (SA) learns object-centric representations by assigning objects to \textit{slots}, but presupposes a \textit{single} distribution from which all slots are randomly initialised. This results in an inability to learn \textit{specialized} slots which bind to specific object types and remain invariant to identity-preserving changes in object appearance. To address this, we present \emph{\textsc{Co}nditional \textsc{S}lot \textsc{A}ttention} (\textsc{CoSA}) using a novel concept of \emph{Grounded Slot Dictionary} (GSD) inspired by vector quantization. Our proposed GSD comprises (i) canonical object-level property vectors and (ii) parametric Gaussian distributions, which define a prior over the slots. We demonstrate the benefits of our method in multiple downstream tasks such as scene generation, composition, and task adaptation, whilst remaining competitive with SA in popular object discovery benchmarks.
format Preprint
id arxiv_https___arxiv_org_abs_2307_09437
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Grounded Object Centric Learning
Kori, Avinash
Locatello, Francesco
Ribeiro, Fabio De Sousa
Toni, Francesca
Glocker, Ben
Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
The extraction of modular object-centric representations for downstream tasks is an emerging area of research. Learning grounded representations of objects that are guaranteed to be stable and invariant promises robust performance across different tasks and environments. Slot Attention (SA) learns object-centric representations by assigning objects to \textit{slots}, but presupposes a \textit{single} distribution from which all slots are randomly initialised. This results in an inability to learn \textit{specialized} slots which bind to specific object types and remain invariant to identity-preserving changes in object appearance. To address this, we present \emph{\textsc{Co}nditional \textsc{S}lot \textsc{A}ttention} (\textsc{CoSA}) using a novel concept of \emph{Grounded Slot Dictionary} (GSD) inspired by vector quantization. Our proposed GSD comprises (i) canonical object-level property vectors and (ii) parametric Gaussian distributions, which define a prior over the slots. We demonstrate the benefits of our method in multiple downstream tasks such as scene generation, composition, and task adaptation, whilst remaining competitive with SA in popular object discovery benchmarks.
title Grounded Object Centric Learning
topic Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2307.09437