arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2406.09196cs.CVcs.LG

自适应槽位注意力:具有动态槽位数量的对象发现

Adaptive Slot Attention: Object Discovery with Dynamic Slot Number

Fudan University(复旦大学);National University of Singapore(新加坡国立大学);Amazon Web Services(亚马逊网络服务);Institute of Science and Technology Austria(奥地利科技学院)

Ke Fan, Zechen Bai, Tianjun Xiao, Tong He, Max Horn, Yanwei Fu, Francesco Locatello, Zheng Zhang

AI总结:

针对槽位注意力需预设槽位数的问题,本文提出复杂度感知对象自动编码器AdaSlot,通过离散槽位采样与掩码解码器动态选择槽位数,并在多数据集对象发现任务上达到或超过固定槽位模型。

摘要:

以对象为中心的学习(OCL)使用槽位提取对象的表示,为抽象低层感知特征提供了灵活性与可解释性的出色结合。OCL中被广泛采用的一种方法是slot attention(槽位注意力),它利用注意力机制迭代地细化槽位表示。然而,包括slot attention在内的大多数以对象为中心的模型的一个主要缺点,是它们依赖于预先定义槽位数量。这不仅需要事先了解数据集,还忽视了每个实例中所存在对象数量的固有变化。为克服这一根本性限制,我们提出了一种新颖的复杂度感知对象自动编码器框架。在该框架中,我们引入了自适应槽位注意力(AdaSlot)机制,可根据数据内容动态确定最优槽位数量。这是通过提出一个离散槽位采样模块实现的,该模块负责从候选列表中选择适当数量的槽位。此外,我们引入了掩码槽位解码器,在解码过程中抑制未被选择的槽位。我们的框架在多个数据集的对象发现任务上进行了广泛测试,表现出与顶尖固定槽位模型相当或更优的性能。此外,我们的分析证实,该方法能够根据每个实例的复杂度动态调整槽位数量,为slot attention研究中的进一步探索提供了潜力。项目将在 https://kfan21.github.io/AdaSlot/ 提供。

英文摘要:

Object-centric learning (OCL) extracts the representation of objects with slots, offering an exceptional blend of flexibility and interpretability for abstracting low-level perceptual features. A widely adopted method within OCL is slot attention, which utilizes attention mechanisms to iteratively refine slot representations. However, a major drawback of most object-centric models, including slot attention, is their reliance on predefining the number of slots. This not only necessitates prior knowledge of the dataset but also overlooks the inherent variability in the number of objects present in each instance. To overcome this fundamental limitation, we present a novel complexity-aware object auto-encoder framework. Within this framework, we introduce an adaptive slot attention (AdaSlot) mechanism that dynamically determines the optimal number of slots based on the content of the data. This is achieved by proposing a discrete slot sampling module that is responsible for selecting an appropriate number of slots from a candidate list. Furthermore, we introduce a masked slot decoder that suppresses unselected slots during the decoding process. Our framework, tested extensively on object discovery tasks with various datasets, shows performance matching or exceeding top fixed-slot models. Moreover, our analysis substantiates that our method exhibits the capability to dynamically adapt the slot number according to each instance's complexity, offering the potential for further exploration in slot attention research. Project will be available at https://kfan21.github.io/AdaSlot/

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑