arXiv reaDer
CoAtNet: すべてのデータ サイズに対する畳み込みと注意の結合
CoAtNet: Marrying Convolution and Attention for All Data Sizes
トランスフォーマーは、コンピューター ビジョンへの関心が高まっていますが、依然として最先端の畳み込みネットワークに遅れをとっています。この作業では、トランスフォーマーはモデル容量が大きい傾向にありますが、適切な誘導バイアスがないため、それらの一般化は畳み込みネットワークよりも悪化する可能性があることを示しています。両方のアーキテクチャの長所を効果的に組み合わせるために、CoAtNets (「コート」ネットと発音します) を紹介します。これは、2 つの重要な洞察から構築されたハイブリッド モデルのファミリーです。(1) 深さ方向の畳み込みと自己注意は、単純な相対的注意によって自然に統合できます。 (2) 畳み込み層とアテンション層を原則的に垂直に積み重ねることは、汎化、容量、効率の改善に驚くほど効果的です。実験によると、当社の CoAtNet は、さまざまなデータセットにわたるさまざまなリソース制約の下で最先端のパフォーマンスを達成しています。たとえば、CoAtNet は、追加のデータなしで 86.0% の ImageNet トップ 1 精度を達成し、追加の JFT データを使用して 89.77% の精度を達成し、畳み込みネットワークとトランスフォーマーの両方の従来技術を凌駕します。特に、ImageNet-21K の 1300 万画像で事前トレーニングした場合、当社の CoAtNet は 88.56% のトップ 1 精度を達成し、使用するデータが 23 分の 1 でありながら、JFT の 300M 画像で事前トレーニングされた ViT-huge と一致します。
Transformers have attracted increasing interests in computer vision, but they still fall behind state-of-the-art convolutional networks. In this work, we show that while Transformers tend to have larger model capacity, their generalization can be worse than convolutional networks due to the lack of the right inductive bias. To effectively combine the strengths from both architectures, we present CoAtNets(pronounced "coat" nets), a family of hybrid models built from two key insights:(1) depthwise Convolution and self-Attention can be naturally unified via simple relative attention; (2) vertically stacking convolution layers and attention layers in a principled way is surprisingly effective in improving generalization, capacity and efficiency. Experiments show that our CoAtNets achieve state-of-the-art performance under different resource constraints across various datasets. For example, CoAtNet achieves 86.0% ImageNet top-1 accuracy without extra data, and 89.77% with extra JFT data, outperforming prior arts of both convolutional networks and Transformers. Notably, when pre-trained with 13M images fromImageNet-21K, our CoAtNet achieves 88.56% top-1 accuracy, matching ViT-huge pre-trained with 300M images from JFT while using 23x less data.
updated: Wed Jun 09 2021 04:35:31 GMT+0000 (UTC)
published: Wed Jun 09 2021 04:35:31 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アソシエイト