QASA: Quality-Guided K-Adaptive Slot Attention for Unsupervised Object-Centric Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Ouyang, Tianran, Dong, Xingping, Zhang, Jing, Ye, Mang, Chen, Jun, Du, Bo
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911384545001472
author Ouyang, Tianran
Dong, Xingping
Zhang, Jing
Ye, Mang
Chen, Jun
Du, Bo
author_facet Ouyang, Tianran
Dong, Xingping
Zhang, Jing
Ye, Mang
Chen, Jun
Du, Bo
contents Slot Attention, an approach that binds different objects in a scene to a set of "slots", has become a leading method in unsupervised object-centric learning. Most methods assume a fixed slot count K, and to better accommodate the dynamic nature of object cardinality, a few works have explored K-adaptive variants. However, existing K-adaptive methods still suffer from two limitations. First, they do not explicitly constrain slot-binding quality, so low-quality slots lead to ambiguous feature attribution. Second, adding a slot-count penalty to the reconstruction objective creates conflicting optimization goals between reducing the number of active slots and maintaining reconstruction fidelity. As a result, they still lag significantly behind strong K-fixed baselines. To address these challenges, we propose Quality-Guided K-Adaptive Slot Attention (QASA). First, we decouple slot selection from reconstruction, eliminating the mutual constraints between the two objectives. Then, we propose an unsupervised Slot-Quality metric to assess per-slot quality, providing a principled signal for fine-grained slot--object binding. Based on this metric, we design a Quality-Guided Slot Selection scheme that dynamically selects a subset of high-quality slots and feeds them into our newly designed gated decoder for reconstruction during training. At inference, token-wise competition on slot attention yields a K-adaptive outcome. Experiments show that QASA substantially outperforms existing K-adaptive methods on both real and synthetic datasets. Moreover, on real-world datasets QASA surpasses K-fixed methods.
format Preprint
id arxiv_https___arxiv_org_abs_2601_12936
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle QASA: Quality-Guided K-Adaptive Slot Attention for Unsupervised Object-Centric Learning
Ouyang, Tianran
Dong, Xingping
Zhang, Jing
Ye, Mang
Chen, Jun
Du, Bo
Computer Vision and Pattern Recognition
Slot Attention, an approach that binds different objects in a scene to a set of "slots", has become a leading method in unsupervised object-centric learning. Most methods assume a fixed slot count K, and to better accommodate the dynamic nature of object cardinality, a few works have explored K-adaptive variants. However, existing K-adaptive methods still suffer from two limitations. First, they do not explicitly constrain slot-binding quality, so low-quality slots lead to ambiguous feature attribution. Second, adding a slot-count penalty to the reconstruction objective creates conflicting optimization goals between reducing the number of active slots and maintaining reconstruction fidelity. As a result, they still lag significantly behind strong K-fixed baselines. To address these challenges, we propose Quality-Guided K-Adaptive Slot Attention (QASA). First, we decouple slot selection from reconstruction, eliminating the mutual constraints between the two objectives. Then, we propose an unsupervised Slot-Quality metric to assess per-slot quality, providing a principled signal for fine-grained slot--object binding. Based on this metric, we design a Quality-Guided Slot Selection scheme that dynamically selects a subset of high-quality slots and feeds them into our newly designed gated decoder for reconstruction during training. At inference, token-wise competition on slot attention yields a K-adaptive outcome. Experiments show that QASA substantially outperforms existing K-adaptive methods on both real and synthetic datasets. Moreover, on real-world datasets QASA surpasses K-fixed methods.
title QASA: Quality-Guided K-Adaptive Slot Attention for Unsupervised Object-Centric Learning
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2601.12936