arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.07544cs.CV
收藏

MUFASA: 一种用于槽注意力的多层框架

MUFASA: A Multi-Layer Framework for Slot Attention

TU Darmstadt(图宾根大学);Zuse School ELIZA(泽尼特学校ELIZA)

Sebastian Bock, Leonie Schüßler, Krishnakant Singh, Simone Schaub-Meyer, Stefan Roth

AI总结:

提出MUFASA,一种轻量级即插即用框架,通过跨ViT编码器多层计算槽注意力并融合,提升无监督对象中心学习的分割性能,达到新最优。

摘要:

无监督对象中心学习(OCL)将视觉场景分解为不同的实体。槽注意力是一种流行的方法,将单个对象表示为潜在向量,称为槽。当前方法仅从预训练视觉变换器(ViT)的最后一层获取这些槽表示,忽略了跨其他层编码的宝贵、语义丰富的信息。为了更好地利用这些潜在语义信息,我们引入了MUFASA,一种用于基于槽注意力的无监督对象分割方法的轻量级即插即用框架。我们的模型跨ViT编码器的多个特征层计算槽注意力,充分利用其语义丰富性。我们提出了一种融合策略,将在多个层上获得的槽聚合成统一的以对象为中心的表示。将MUFASA集成到现有的OCL方法中,提高了它们在多个数据集上的分割结果,在仅增加少量推理开销的同时,建立了新的最先进水平并改善了训练收敛性。

英文摘要:

Unsupervised object-centric learning (OCL) decomposes visual scenes into distinct entities. Slot attention is a popular approach that represents individual objects as latent vectors, called slots. Current methods obtain these slot representations solely from the last layer of a pre-trained vision transformer (ViT), ignoring valuable, semantically rich information encoded across the other layers. To better utilize this latent semantic information, we introduce MUFASA, a lightweight plug-and-play framework for slot-attention-based approaches to unsupervised object segmentation. Our model computes slot attention across multiple feature layers of the ViT encoder, fully leveraging their semantic richness. We propose a fusion strategy to aggregate slots obtained on multiple layers into a unified object-centric representation. Integrating MUFASA into existing OCL methods improves their segmentation results across multiple datasets, setting a new state of the art while simultaneously improving training convergence with only minor inference overhead.

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

↑