用超像素替代均匀分块,让视觉令牌更专注单一语义。
Superpixel Tokenization for Vision Transformers: Preserving Semantic Integrity in Visual Tokens
- 用超像素生成视觉令牌,避免一个令牌包含多个概念。
- 在多个下游任务中提升ViT准确率与鲁棒性。
- 适合追求语义清晰的视觉模型研究者。
Transformer架构在自然语言处理领域取得突破后,也成功应用于计算机视觉。其成功关键在于注意力机制对令牌间关系的建模。然而,视觉Transformer(ViT)采用均匀划分的方形图像块作为令牌,可能导致一个令牌内混合多种视觉概念。本文提出用超像素代替网格化令牌化,使每个令牌仅包含单一视觉概念。由于超像素形状、大小和位置各异,将其融入ViT存在挑战。我们设计了预聚合提取与超像素感知聚合的流水线,有效克服这些困难。大量实验表明,该方法兼容现有框架,在多个下游任务中显著提升ViT的精度与鲁棒性。
原文摘要 · Abstract (English)
Transformers, a groundbreaking architecture proposed for Natural Language Processing (NLP), have also achieved remarkable success in Computer Vision. A cornerstone of their success lies in the attention mechanism, which models relationships among tokens. While the tokenization process in NLP inherently ensures that a single token does not contain multiple semantics, the tokenization of Vision Transformer (ViT) utilizes tokens from uniformly partitioned square image patches, which may result in an arbitrary mixing of visual concepts in a token. In this work, we propose to substitute the grid-based tokenization in ViT with superpixel tokenization, which employs superpixels to generate a token that encapsulates a sole visual concept. Unfortunately, the diverse shapes, sizes, and locations of superpixels make integrating superpixels into ViT tokenization rather challenging. Our tokenization pipeline, comprised of pre-aggregate extraction and superpixel-aware aggregation, overcomes the challenges that arise in superpixel tokenization. Extensive experiments demonstrate that our approach, which exhibits strong compatibility with existing frameworks, enhances the accuracy and robustness of ViT on various downstream tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。