arXiv reaDer
すべての画像が16x16ワードに値するわけではありません:効率的な画像認識のための動的トランスフォーマー
Not All Images are Worth 16x16 Words: Dynamic Transformers for Efficient Image Recognition
Vision Transformers(ViT)は、大規模な画像認識で目覚ましい成功を収めています。それらはすべての2D画像を固定数のパッチに分割し、それぞれがトークンとして扱われます。一般に、より多くのトークンで画像を表現すると、予測精度が高くなりますが、計算コストも大幅に増加します。精度と速度の間で適切なトレードオフを実現するために、トークンの数は経験的に16x16または14x14に設定されています。この論文では、すべての画像に独自の特性があり、理想的にはトークン番号は個々の入力ごとに条件付けられる必要があると主張します。実際、4x4トークンの数だけで正確に予測できる「簡単な」画像がかなりの数存在する一方で、「難しい」画像のごく一部だけがより細かい表現を必要とすることがわかりました。この現象に触発されて、入力画像ごとに適切な数のトークンを自動的に構成するダイナミックトランスフォーマーを提案します。これは、トークンの数を増やしながら複数のトランスフォーマーをカスケードすることによって実現されます。トークンは、テスト時に適応的に順次アクティブ化されます。つまり、十分に信頼できる予測が生成されると、推論は終了します。さらに、Dynamic Transformerのさまざまなコンポーネント間で効率的な機能の再利用と関係の再利用のメカニズムを設計し、冗長な計算を削減します。 ImageNet、CIFAR-10、およびCIFAR-100での広範な経験的結果は、理論的な計算効率と実際の推論速度の両方の点で、私たちの方法が競合するベースラインを大幅に上回っていることを示しています。コードおよび事前トレーニング済みモデル(PyTorchおよびMindSporeに基づく)は、https://github.com/blackfeather-wang/Dynamic-Vision-Transformerおよびhttps://github.com/blackfeather-wang/Dynamic-Vision-で入手できます。トランスフォーマー-MindSpore。
Vision Transformers (ViT) have achieved remarkable success in large-scale image recognition. They split every 2D image into a fixed number of patches, each of which is treated as a token. Generally, representing an image with more tokens would lead to higher prediction accuracy, while it also results in drastically increased computational cost. To achieve a decent trade-off between accuracy and speed, the number of tokens is empirically set to 16x16 or 14x14. In this paper, we argue that every image has its own characteristics, and ideally the token number should be conditioned on each individual input. In fact, we have observed that there exist a considerable number of "easy" images which can be accurately predicted with a mere number of 4x4 tokens, while only a small fraction of "hard" ones need a finer representation. Inspired by this phenomenon, we propose a Dynamic Transformer to automatically configure a proper number of tokens for each input image. This is achieved by cascading multiple Transformers with increasing numbers of tokens, which are sequentially activated in an adaptive fashion at test time, i.e., the inference is terminated once a sufficiently confident prediction is produced. We further design efficient feature reuse and relationship reuse mechanisms across different components of the Dynamic Transformer to reduce redundant computations. Extensive empirical results on ImageNet, CIFAR-10, and CIFAR-100 demonstrate that our method significantly outperforms the competitive baselines in terms of both theoretical computational efficiency and practical inference speed. Code and pre-trained models (based on PyTorch and MindSpore) are available at https://github.com/blackfeather-wang/Dynamic-Vision-Transformer and https://github.com/blackfeather-wang/Dynamic-Vision-Transformer-MindSpore.
updated: Tue Oct 26 2021 14:08:00 GMT+0000 (UTC)
published: Mon May 31 2021 16:04:10 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アソシエイト