Explicitly Disentangled Representations in Object-Centric Learning

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Majellaro, Riccardo, Collu, Jonathan, Plaat, Aske, Moerland, Thomas M.
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866929684251410432
author Majellaro, Riccardo
Collu, Jonathan
Plaat, Aske
Moerland, Thomas M.
author_facet Majellaro, Riccardo
Collu, Jonathan
Plaat, Aske
Moerland, Thomas M.
contents Extracting structured representations from raw visual data is an important and long-standing challenge in machine learning. Recently, techniques for unsupervised learning of object-centric representations have raised growing interest. In this context, enhancing the robustness of the latent features can improve the efficiency and effectiveness of the training of downstream tasks. A promising step in this direction is to disentangle the factors that cause variation in the data. Previously, Invariant Slot Attention disentangled position, scale, and orientation from the remaining features. Extending this approach, we focus on separating the shape and texture components. In particular, we propose a novel architecture that biases object-centric models toward disentangling shape and texture components into two non-overlapping subsets of the latent space dimensions. These subsets are known a priori, hence before the training process. Experiments on a range of object-centric benchmarks reveal that our approach achieves the desired disentanglement while also numerically improving baseline performance in most cases. In addition, we show that our method can generate novel textures for a specific object or transfer textures between objects with distinct shapes.
format Preprint
id arxiv_https___arxiv_org_abs_2401_10148
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Explicitly Disentangled Representations in Object-Centric Learning
Majellaro, Riccardo
Collu, Jonathan
Plaat, Aske
Moerland, Thomas M.
Computer Vision and Pattern Recognition
Artificial Intelligence
Machine Learning
Extracting structured representations from raw visual data is an important and long-standing challenge in machine learning. Recently, techniques for unsupervised learning of object-centric representations have raised growing interest. In this context, enhancing the robustness of the latent features can improve the efficiency and effectiveness of the training of downstream tasks. A promising step in this direction is to disentangle the factors that cause variation in the data. Previously, Invariant Slot Attention disentangled position, scale, and orientation from the remaining features. Extending this approach, we focus on separating the shape and texture components. In particular, we propose a novel architecture that biases object-centric models toward disentangling shape and texture components into two non-overlapping subsets of the latent space dimensions. These subsets are known a priori, hence before the training process. Experiments on a range of object-centric benchmarks reveal that our approach achieves the desired disentanglement while also numerically improving baseline performance in most cases. In addition, we show that our method can generate novel textures for a specific object or transfer textures between objects with distinct shapes.
title Explicitly Disentangled Representations in Object-Centric Learning
topic Computer Vision and Pattern Recognition
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2401.10148