arXiv reaDer
EfficientViT:高解像度低計算視覚認識のための強化された線形注意
EfficientViT: Enhanced Linear Attention for High-Resolution Low-Computation Visual Recognition
Vision Transformer(ViT)は、多くのビジョンタスクで卓越したパフォーマンスを実現しています。ただし、高解像度のモバイルビジョンアプリケーションを対象とする場合、ViTは畳み込みニューラルネットワーク(CNN)より劣ります。 ViTの主要な計算上のボトルネックは、入力解像度で2次計算の複雑さを持つsoftmaxアテンションモジュールです。 ViTをエッジデバイスに展開するには、ViTのコストを削減することが不可欠です。既存の方法(Swin、PVTなど)は、ローカルウィンドウ内のソフトマックスの注意を制限するか、キー/値テンソルの解像度を下げてコストを削減します。これにより、グローバルな特徴抽出におけるViTのコアの利点が犠牲になります。この作業では、高解像度の低計算視覚認識のための効率的なViTアーキテクチャであるEfficientViTを紹介します。ソフトマックス注意を制限する代わりに、深さ方向の畳み込みで局所的な特徴抽出能力を強化しながら、ソフトマックス注意を線形注意に置き換えることを提案します。 EfficientViTは、線形計算の複雑さを享受しながら、グローバルおよびローカルの特徴抽出機能を維持します。 COCOオブジェクト検出とCityscapesセマンティックセグメンテーションに関する広範な実験は、私たちの方法の有効性を示しています。 COCOデータセットでは、EfficientViTは4.4GMACで42.6APを達成し、MACが27.9%少ない一方で、EfficientDet-D1を2.4AP上回っています。 Cityscapesでは、EfficientViTは19.1GMACで78.7mIoUに達し、計算コストの3分の1未満で、SegFormerを2.5mIoU上回ります。 Qualcomm Snapdragon 855 CPUでは、EfficientViTはEfficientNetより3倍高速であり、ImageNetの精度も高くなっています。
Vision Transformer (ViT) has achieved remarkable performance in many vision tasks. However, ViT is inferior to convolutional neural networks (CNNs) when targeting high-resolution mobile vision applications. The key computational bottleneck of ViT is the softmax attention module which has quadratic computational complexity with the input resolution. It is essential to reduce the cost of ViT to deploy it on edge devices. Existing methods (e.g., Swin, PVT) restrict the softmax attention within local windows or reduce the resolution of key/value tensors to reduce the cost, which sacrifices ViT's core advantages on global feature extractions. In this work, we present EfficientViT, an efficient ViT architecture for high-resolution low-computation visual recognition. Instead of restricting the softmax attention, we propose to replace softmax attention with linear attention while enhancing its local feature extraction ability with depthwise convolution. EfficientViT maintains global and local feature extraction capability while enjoying linear computational complexity. Extensive experiments on COCO object detection and Cityscapes semantic segmentation demonstrate the effectiveness of our method. On the COCO dataset, EfficientViT achieves 42.6 AP with 4.4G MACs, surpassing EfficientDet-D1 by 2.4 AP while having 27.9% fewer MACs. On Cityscapes, EfficientViT reaches 78.7 mIoU with 19.1G MACs, outperforming SegFormer by 2.5 mIoU while requiring less than 1/3 the computational cost. On Qualcomm Snapdragon 855 CPU, EfficientViT is 3x faster than EfficientNet while achieving higher ImageNet accuracy.
updated: Sun Sep 04 2022 20:18:11 GMT+0000 (UTC)
published: Sun May 29 2022 20:07:23 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト