arXiv · 2411.02299
Grouped Discrete Representation for Object-Centric Learning
Abstract
Object-Centric Learning (OCL) aims to discover objects in images or videos by reconstructing the input. Representative methods achieve this by reconstructing the input as its Variational Autoencoder (VAE) discrete representations, which suppress (super-)pixel noise and enhance object separability. However, these methods treat features as indivisible units, overlooking their compositional attributes, and discretize features via scalar code indexes, losing attribute-level similarities and differences. We propose Grouped Discrete Representation (GDR) for OCL. For better generalization, features are decomposed into combinatorial attributes by organized channel grouping. For better convergence, features are quantized into discrete representations via tuple code indexes. Experiments demonstrate that GDR consistently improves both mainstream and state-of-the-art OCL methods across various datasets. Visualizations further highlight GDR's superior object separability and interpretability. The source code is available on https://github.com/Genera1Z/GroupedDiscreteRepresentation.
Explore related subjects
Keep this discovery
Rongzhen Zhao, Vivienne Wang, Juho Kannala, Joni Pajarinen. 2024-11-04. Grouped Discrete Representation for Object-Centric Learning. https://arxiv.org/abs/2411.02299
Cite the original work for its findings. Save a collection to share your selection of sources.