arXiv reaDer
ビジョントランスフォーマーの移植
Grafting Vision Transformers
ビジョン トランスフォーマー (ViT) は、最近、多くのコンピューター ビジョン タスクで最先端のものになりました。畳み込みネットワーク (CNN) とは対照的に、ViT は、ネットワークの浅いレイヤー内、つまり高解像度フィーチャ間でもグローバルな情報共有を可能にします。ただし、この利点は、パフォーマンスと複雑さのトレードオフが優れている Swin Transformer などのピラミッド アーキテクチャの成功により、後に見落とされました。このホワイトペーパーでは、高解像度と低解像度の両方の機能で、ネットワーク全体のグローバルな依存関係とマルチスケール情報を同様に考慮する、シンプルで効率的なアドオン コンポーネント (GrafT と呼ばれる) を紹介します。任意の深さで分岐できる柔軟性があり、バックボーンのほとんどのパラメーターと計算を共有します。 GrafT は、ハイブリッド型と純粋な Transformer 型、同種型とピラミッド型の両方の構造、およびさまざまな自己注意方法を含む、さまざまなよく知られたモデルに対して一貫したゲインを示しています。特に、高レベルのセマンティクスを提供することで、モバイル サイズのモデルに大きなメリットがあります。 ImageNet-1k データセットでは、GrafT は、DeiT-T、Swin-T、および MobileViT-XXS に対して、それぞれ +3.9%、+1.4%、および +1.9% のトップ 1 精度の向上を実現します。私たちのコードとモデルが利用可能になります。
Vision Transformers (ViTs) have recently become the state-of-the-art across many computer vision tasks. In contrast to convolutional networks (CNNs), ViTs enable global information sharing even within shallow layers of a network, i.e., among high-resolution features. However, this perk was later overlooked with the success of pyramid architectures such as Swin Transformer, which show better performance-complexity trade-offs. In this paper, we present a simple and efficient add-on component (termed GrafT) that considers global dependencies and multi-scale information throughout the network, in both high- and low-resolution features alike. It has the flexibility of branching out at arbitrary depths and shares most of the parameters and computations of the backbone. GrafT shows consistent gains over various well-known models which includes both hybrid and pure Transformer types, both homogeneous and pyramid structures, and various self-attention methods. In particular, it largely benefits mobile-size models by providing high-level semantics. On the ImageNet-1k dataset, GrafT delivers +3.9%, +1.4%, and +1.9% top-1 accuracy improvement to DeiT-T, Swin-T, and MobileViT-XXS, respectively. Our code and models will be made available.
updated: Mon Apr 03 2023 14:16:14 GMT+0000 (UTC)
published: Fri Oct 28 2022 07:07:13 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アソシエイト