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

超越 [cls]:探索掩码图像建模表示的真正潜力

Beyond [cls]: Exploring the true potential of Masked Image Modeling representations

Jagiellonian University(雅盖隆大学);Brown University(布朗大学);AGH University of Krakow(克拉科夫AGH大学)

Marcin Przewięźlikowski, Randall Balestriero, Wojciech Jasiński, Marek Śmieja, Bartosz Zieliński

AI总结:

本文发现MIM即用性能差因注意力均匀分布致[cls] token聚合无效,提出选择性聚合方法捕获patch token语义信息,显著提升性能。

摘要:

掩码图像建模(MIM)已成为视觉表示自监督学习(SSL)的一种有前景的方法。然而,MIM的即用性能通常劣于竞争方法。由于需要大量数据、高GPU消耗和专业知识,大多数用户无法承担微调。因此,MIM表示的实际应用受限。本文探讨MIM即用性能不佳的原因:是MIM模型产生的特征较弱,还是使用方式不佳?通过详细分析,我们发现MIM中的注意力几乎均匀分布在多个patch上,导致[cls] token聚合无效。基于此,我们提出选择性聚合以更好捕获patch token中保留的丰富语义信息,显著提升了MIM的即用性能。

英文摘要:

Masked Image Modeling (MIM) has emerged as a promising approach for Self-Supervised Learning (SSL) of visual representations. However, the out-of-the-box performance of MIMs is typically inferior to competing approaches. Most users cannot afford fine-tuning due to the need for large amounts of data, high GPU consumption, and specialized user knowledge. Therefore, the practical use of MIM representations is limited. In this paper we ask what is the reason for the poor out-of-the-box performance of MIMs. Is it due to weaker features produced by MIM models, or is it due to suboptimal usage? Through detailed analysis, we show that attention in MIMs is spread almost uniformly over many patches, leading to ineffective aggregation by the [cls] token. Based on this insight, we propose Selective Aggregation to better capture the rich semantic information retained in patch tokens, which significantly improves the out-of-the-box performance of MIM.

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

↑