ますます高まる精度を達成するために、通常、大規模で複雑なニューラル ネットワークが開発されます。このようなモデルは大量の計算リソースを必要とするため、エッジ デバイスには展開できません。リソース効率の高い汎用ネットワークを構築することは、いくつかのアプリケーション分野で有用であるため、非常に興味深いものです。この作業では、CNN モデルと Transformer モデルの両方の長所を効果的に組み合わせ、新しい効率的なハイブリッド アーキテクチャ EdgeNeXt を提案するよう努めています。特に EdgeNeXt では、入力テンソルを複数のチャネル グループに分割し、深さ方向の畳み込みとチャネル次元全体のセルフアテンションを利用して、受容野を暗黙的に増加させ、マルチスケール機能をエンコードする分割深さ方向転置注意 (STDA) エンコーダーを導入します。 .分類、検出、およびセグメンテーションのタスクに関する広範な実験により、提案されたアプローチのメリットが明らかになり、計算要件が比較的低い最先端の方法よりも優れています。 1.3M パラメータの EdgeNeXt モデルは、ImageNet-1K で 71.2% のトップ 1 精度を達成し、絶対ゲイン 2.2% で MobileViT を上回り、FLOP を 28% 削減します。さらに、560 万のパラメーターを持つ EdgeNeXt モデルは、ImageNet-1K で 79.4% のトップ 1 精度を達成します。コードとモデルは https://t.ly/_Vu9 で入手できます。
In the pursuit of achieving ever-increasing accuracy, large and complex neural networks are usually developed. Such models demand high computational resources and therefore cannot be deployed on edge devices. It is of great interest to build resource-efficient general purpose networks due to their usefulness in several application areas. In this work, we strive to effectively combine the strengths of both CNN and Transformer models and propose a new efficient hybrid architecture EdgeNeXt. Specifically in EdgeNeXt, we introduce split depth-wise transpose attention (STDA) encoder that splits input tensors into multiple channel groups and utilizes depth-wise convolution along with self-attention across channel dimensions to implicitly increase the receptive field and encode multi-scale features. Our extensive experiments on classification, detection and segmentation tasks, reveal the merits of the proposed approach, outperforming state-of-the-art methods with comparatively lower compute requirements. Our EdgeNeXt model with 1.3M parameters achieves 71.2% top-1 accuracy on ImageNet-1K, outperforming MobileViT with an absolute gain of 2.2% with 28% reduction in FLOPs. Further, our EdgeNeXt model with 5.6M parameters achieves 79.4% top-1 accuracy on ImageNet-1K. The code and models are available at https://t.ly/_Vu9.