arXiv reaDer
セルフスリムビジョントランスフォーマー
Self-slimmed Vision Transformer
ビジョントランスフォーマー(ViT)は人気のある構造になり、さまざまなビジョンタスクで畳み込みニューラルネットワーク(CNN)を上回りました。ただし、このような強力なトランスフォーマーは、計算に大きな負担をかけます。そして、この背後にある本質的な障壁は、トークン間の比較です。これを軽減するために、ViTのモデルプロパティを深く掘り下げ、ViTがトークンの類似性が高くスパースな注意を示すことを確認します。これにより、計算コストを削減するために、実行可能な構造に依存しない次元、トークン番号が直感的に導入されます。この調査に基づいて、バニラViT、つまりSiTの一般的な自己スリム学習アプローチを提案します。具体的には、最初に、動的トークン集約によってViTの推論効率を高めることができる新しいトークンスリミングモジュール(TSM)を設計します。トークンのハードドロップとは異なり、TSMは冗長なトークンをより情報量の少ないトークンにソフトに統合します。これにより、画像内の識別可能なトークンの関係を切断することなく、視覚的な注意を動的にズームできます。さらに、簡潔なDense Knowledge Distillation(DKD)フレームワークを導入します。このフレームワークは、組織化されていないトークン情報を柔軟なオートエンコーダー方式で高密度に転送します。教師と生徒の間の構造が類似しているため、私たちのフレームワークは構造の知識を効果的に活用して、より良い収束を実現できます。最後に、SiTを評価するために広範な実験を行います。これは、私たちの方法がViTを1.7倍高速化し、精度の低下を無視できることを示しています。さらに、パフォーマンスの97%を維持しながら、ViTを3.6倍高速化することもできます。驚いたことに、LV-ViTにSiTを装備するだけで、ImageNetで新しい最先端のパフォーマンスを実現し、最近の文献のすべてのCNNとViTを上回りました。
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. And the essential barrier behind this is the exhausting token-to-token comparison. To alleviate this, we delve deeply into the model properties of ViT and observe that ViTs exhibit sparse attention with high token similarity. This intuitively introduces us a feasible structure-agnostic dimension, token number, to reduce the computational cost. Based on this exploration, 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. Different from the token hard dropping, our TSM softly integrates redundant tokens into fewer informative ones, which can dynamically zoom visual attention without cutting off discriminative token relations in the images. Furthermore, we introduce a concise Dense Knowledge Distillation (DKD) framework, which densely transfers unorganized token information in a flexible auto-encoder manner. Due to the similar structure between teacher and student, our framework 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, surpassing all the CNNs and ViTs in the recent literature.
updated: Wed Nov 24 2021 16:48:57 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アソシエイト