arXivDaily arXiv每日学术速递 周一至周五更新
稍安勿躁,官方尚未更新
arXiv 2406.09415cs.CVcs.LG
收藏

一张图像的价值不止于16x16补丁:探索基于单个像素的Transformer

An Image is Worth More Than 16x16 Patches: Exploring Transformers on Individual Pixels

FAIR, Meta AI(FAIR,Meta人工智能);University of Amsterdam(阿姆斯特丹大学)

Duy-Kien Nguyen, Mahmoud Assran, Unnat Jain, Martin R. Oswald, Cees G. M. Snoek, Xinlei Chen

AI总结:

研究发现普通Transformer可直接将单个像素作为token,在分类回归、掩码自编码和扩散生成任务上取得高性能,从而质疑视觉架构中局部性归纳偏置的必要性。

摘要:

本工作并未提出一种新方法。相反,我们呈现了一个有趣的发现,对现代计算机视觉架构中局部性归纳偏置的必要性提出质疑。具体而言,我们发现普通Transformer可以直接将每个单独像素视为一个token来运行,并取得高性能结果。这与Vision Transformer中的流行设计有本质不同,后者保留了来自ConvNets的、面向局部邻域的归纳偏置(例如通过将每个16x16补丁视为一个token)。我们在三个已被充分研究的计算机视觉任务上展示了“像素即token”的有效性:用于分类和回归的监督学习、通过掩码自编码进行的自监督学习,以及使用扩散模型的图像生成。尽管直接对单个像素进行运算在计算上不太实用,但我们认为,在为计算机视觉设计下一代神经网络架构时,必须让社区了解这一令人惊讶的认知。

英文摘要:

This work does not introduce a new method. Instead, we present an interesting finding that questions the necessity of the inductive bias of locality in modern computer vision architectures. Concretely, we find that vanilla Transformers can operate by directly treating each individual pixel as a token and achieve highly performant results. This is substantially different from the popular design in Vision Transformer, which maintains the inductive bias from ConvNets towards local neighborhoods (e.g. by treating each 16x16 patch as a token). We showcase the effectiveness of pixels-as-tokens across three well-studied computer vision tasks: supervised learning for classification and regression, self-supervised learning via masked autoencoding, and image generation with diffusion models. Although it's computationally less practical to directly operate on individual pixels, we believe the community must be made aware of this surprising piece of knowledge when devising the next generation of neural network architectures for computer vision.

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

↑