Organized Grouped Discrete Representation for Object-Centric Learning

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Zhao, Rongzhen, Wang, Vivienne, Kannala, Juho, Pajarinen, Joni
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866909435599781888
author Zhao, Rongzhen
Wang, Vivienne
Kannala, Juho
Pajarinen, Joni
author_facet Zhao, Rongzhen
Wang, Vivienne
Kannala, Juho
Pajarinen, Joni
contents Object-Centric Learning (OCL) represents dense image or video pixels as sparse object features. Representative methods utilize discrete representation composed of Variational Autoencoder (VAE) template features to suppress pixel-level information redundancy and guide object-level feature aggregation. The most recent advancement, Grouped Discrete Representation (GDR), further decomposes these template features into attributes. However, its naive channel grouping as decomposition may erroneously group channels belonging to different attributes together and discretize them as sub-optimal template attributes, which losses information and harms expressivity. We propose Organized GDR (OGDR) to organize channels belonging to the same attributes together for correct decomposition from features into attributes. In unsupervised segmentation experiments, OGDR is fully superior to GDR in augmentating classical transformer-based OCL methods; it even improves state-of-the-art diffusion-based ones. Codebook PCA and representation similarity analyses show that compared with GDR, our OGDR eliminates redundancy and preserves information better for guiding object representation learning. The source code is available in the supplementary material.
format Preprint
id arxiv_https___arxiv_org_abs_2409_03553
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Organized Grouped Discrete Representation for Object-Centric Learning
Zhao, Rongzhen
Wang, Vivienne
Kannala, Juho
Pajarinen, Joni
Computer Vision and Pattern Recognition
Object-Centric Learning (OCL) represents dense image or video pixels as sparse object features. Representative methods utilize discrete representation composed of Variational Autoencoder (VAE) template features to suppress pixel-level information redundancy and guide object-level feature aggregation. The most recent advancement, Grouped Discrete Representation (GDR), further decomposes these template features into attributes. However, its naive channel grouping as decomposition may erroneously group channels belonging to different attributes together and discretize them as sub-optimal template attributes, which losses information and harms expressivity. We propose Organized GDR (OGDR) to organize channels belonging to the same attributes together for correct decomposition from features into attributes. In unsupervised segmentation experiments, OGDR is fully superior to GDR in augmentating classical transformer-based OCL methods; it even improves state-of-the-art diffusion-based ones. Codebook PCA and representation similarity analyses show that compared with GDR, our OGDR eliminates redundancy and preserves information better for guiding object representation learning. The source code is available in the supplementary material.
title Organized Grouped Discrete Representation for Object-Centric Learning
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2409.03553