arXiv reaDer
セルフスリムビジョントランスフォーマー
Self-slimmed Vision Transformer
ビジョントランスフォーマー(ViT)は人気のある構造になり、さまざまなビジョンタスクで畳み込みニューラルネットワーク(CNN)を上回りました。ただし、このような強力なトランスフォーマーは、トークン間の比較が大変なため、計算に多大な負担をかけます。以前の作品は、ViTの計算コストを削減するために重要でないトークンを削除することに焦点を当てています。しかし、ドロップ率が上がると、この難しい方法は必然的に重要なトークンを破棄し、その効率を制限します。この問題を解決するために、バニラViT、つまりSiTの一般的な自己スリム学習アプローチを提案します。具体的には、最初に、動的トークン集約によってViTの推論効率を高めることができる新しいトークンスリミングモジュール(TSM)を設計します。トークンのハードドロップの一般的な方法として、TSMは冗長なトークンをより情報量の少ないトークンにソフトに統合します。スリミング率が高くても、画像内の識別トークンの関係を切断することなく、視覚的な注意を動的にズームできます。さらに、簡潔なFeature Recalibration Distillation(FRD)フレームワークを導入します。このフレームワークでは、TSM(RTSM)の逆バージョンを設計して、柔軟なオートエンコーダー方式で非構造化トークンを再キャリブレーションします。教師と生徒の間の構造が類似しているため、FRDは構造の知識を効果的に活用して収束を改善できます。最後に、SiTを評価するために広範な実験を行います。これは、私たちの方法がViTを1.7倍高速化し、精度の低下を無視できることを示しています。さらに、パフォーマンスの97%を維持しながら、ViTを3.6倍高速化することもできます。驚いたことに、LV-ViTにSiTを装備するだけで、ImageNetで新しい最先端のパフォーマンスを実現できます。コードはhttps://github.com/Sense-X/SiTで入手できます。
Vision transformers (ViTs) have become the popular structures and outperformed convolutional neural networks (CNNs) on various vision tasks. However, such powerful transformers bring a huge computation burden, because of the exhausting token-to-token comparison. The previous works focus on dropping insignificant tokens to reduce the computational cost of ViTs. But when the dropping ratio increases, this hard manner will inevitably discard the vital tokens, which limits its efficiency. To solve the issue, we propose a generic self-slimmed learning approach for vanilla ViTs, namely SiT. Specifically, we first design a novel Token Slimming Module (TSM), which can boost the inference efficiency of ViTs by dynamic token aggregation. As a general method of token hard dropping, our TSM softly integrates redundant tokens into fewer informative ones. It can dynamically zoom visual attention without cutting off discriminative token relations in the images, even with a high slimming ratio. Furthermore, we introduce a concise Feature Recalibration Distillation (FRD) framework, wherein we design a reverse version of TSM (RTSM) to recalibrate the unstructured token in a flexible auto-encoder manner. Due to the similar structure between teacher and student, our FRD can effectively leverage structure knowledge for better convergence. Finally, we conduct extensive experiments to evaluate our SiT. It demonstrates that our method can speed up ViTs by 1.7x with negligible accuracy drop, and even speed up ViTs by 3.6x while maintaining 97% of their performance. Surprisingly, by simply arming LV-ViT with our SiT, we achieve new state-of-the-art performance on ImageNet. Code is available at https://github.com/Sense-X/SiT.
updated: Tue Jul 26 2022 08:39:28 GMT+0000 (UTC)
published: Wed Nov 24 2021 16:48:57 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アソシエイト