Visual Attention Network

Code

• github.com/Visual-Attention-Network/VAN-Classification
  Officialpytorch★ 821
• github.com/huggingface/transformers
  pytorch★ 158,292
• github.com/facebookresearch/xformers
  pytorch★ 10,384
• github.com/PaddlePaddle/PaddleClas
  paddle★ 5,788
• github.com/open-mmlab/mmclassification
  In paperpytorch★ 3,839
• github.com/lucasjinreal/yolov7_d2
  pytorch★ 3,114
• github.com/MenghaoGuo/Awesome-Vision-Attentions
  none★ 2,843
• github.com/chengtan9907/simvpv2
  pytorch★ 1,078
• github.com/sithu31296/semantic-segmentation
  pytorch★ 939
• github.com/Westlake-AI/openmixup
  pytorch★ 657

Abstract

While originally designed for natural language processing tasks, the self-attention mechanism has recently taken various computer vision areas by storm. However, the 2D nature of images brings three challenges for applying self-attention in computer vision. (1) Treating images as 1D sequences neglects their 2D structures. (2) The quadratic complexity is too expensive for high-resolution images. (3) It only captures spatial adaptability but ignores channel adaptability. In this paper, we propose a novel linear attention named large kernel attention (LKA) to enable self-adaptive and long-range correlations in self-attention while avoiding its shortcomings. Furthermore, we present a neural network based on LKA, namely Visual Attention Network (VAN). While extremely simple, VAN surpasses similar size vision transformers(ViTs) and convolutional neural networks(CNNs) in various tasks, including image classification, object detection, semantic segmentation, panoptic segmentation, pose estimation, etc. For example, VAN-B6 achieves 87.8% accuracy on ImageNet benchmark and set new state-of-the-art performance (58.2 PQ) for panoptic segmentation. Besides, VAN-B2 surpasses Swin-T 4% mIoU (50.1 vs. 46.1) for semantic segmentation on ADE20K benchmark, 2.6% AP (48.8 vs. 46.2) for object detection on COCO dataset. It provides a novel method and a simple yet strong baseline for the community. Code is available at https://github.com/Visual-Attention-Network.

Tasks

Benchmark Results

Reproductions