改进ViT的空间先验,提升模型性能与效率
Advancing Vision Transformer with Enhanced Spatial Priors

- 用欧氏距离衰减引入空间信息,替代曼哈顿距离
- 采用分组注意力机制,实现更高灵活性和更强表征能力
- 在图像分类等任务中表现优异,无需额外数据即达86.6%准确率
近年来,视觉变换器(ViT)在计算机视觉领域受到广泛关注。然而,其核心组件自注意力机制缺乏显式空间先验,且存在二次计算复杂度问题,限制了应用范围。为此,本文提出RMT,一种具备显式空间先验的通用视觉主干网络。RMT利用曼哈顿距离衰减引入空间信息,并采用水平与垂直分解注意力机制建模全局信息。在此基础上,本文进一步提出欧氏增强视觉变换器(EVT),通过更合理的欧氏距离衰减提升空间关系建模精度,取代RMT中的分解注意力机制,转而采用更简单的空间无关分组策略,使模型在每组令牌数量控制上更具灵活性。实验表明,EVT在图像分类、目标检测、实例分割和语义分割等任务中均表现出色,不依赖额外训练数据时,在ImageNet-1k上达到86.6%的Top1准确率。
原文摘要 · Abstract (English)
In recent years, the Vision Transformer (ViT) has garnered significant attention within the computer vision community. However, the core component of ViT, Self-Attention, lacks explicit spatial priors and suffers from quadratic computational complexity, limiting its applicability. To address these issues, we have proposed RMT, a robust vision backbone with explicit spatial priors for general purposes. RMT utilizes Manhattan distance decay to introduce spatial information and employs a horizontal and vertical decomposition attention method to model global information. Building on the strengths of RMT, Euclidean enhanced Vision Transformer (EVT) is an expanded version that incorporates several key improvements. Firstly, EVT uses a more reasonable Euclidean distance decay to enhance the modeling of spatial information, allowing for a more accurate representation of spatial relationships compared to the Manhattan distance used in RMT. Secondly, EVT abandons the decomposed attention mechanism featured in RMT and instead adopts a simpler spatially-independent grouping approach, providing the model with greater flexibility in controlling the number of tokens within each group. By addressing these modifications, EVT offers a more sophisticated and adaptable approach to incorporating spatial priors into the Self-Attention mechanism, thus overcoming some of the limitations associated with RMT and further enhancing its applicability in various computer vision tasks. Extensive experiments on Image Classification, Object Detection, Instance Segmentation, and Semantic Segmentation demonstrate that EVT exhibits exceptional performance. Without additional training data, EVT achieves 86.6% top1-acc on ImageNet-1k.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。