arXiv:2601.12936cs.CV2026-01

提出新方法让模型自动决定物体数量,且更准地识别每个物体。

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

  • 分离选槽与重建任务,避免目标冲突
  • 用质量评分动态筛选高质量槽,提升特征绑定准确率
  • 在真实数据上超越固定数量的基线方法,适合复杂场景建模

Slot Attention 是一种将场景中不同物体绑定到一组“槽”上的无监督对象中心学习方法。现有方法多假设槽数量 K 固定,虽有研究探索 K 自适应变体,但仍存在两大问题:一是未显式约束槽绑定质量,导致特征归属模糊;二是添加槽数惩罚项使减少槽数与保持重建保真度产生冲突优化目标,致使性能显著落后于强基线。为此,本文提出质量引导的 K 自适应槽注意力(QASA)。首先,解耦槽选择与重建过程,消除两目标间的相互制约。其次,设计无监督的槽质量度量,为细粒度的槽-物体绑定提供合理信号。基于此,构建质量引导的槽选择机制,在训练时仅将高质量槽输入门控解码器进行重建。推理时通过槽注意力中的逐标记竞争实现 K 自适应。实验表明,QASA 在真实和合成数据集上均显著优于现有 K 自适应方法,且在真实数据集上超越固定 K 方法。

原文摘要 · Abstract (English)

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.

无监督学习对象中心自适应槽

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。