Neural Architecture Search (NAS) は、1G FLOP を超えるビジョン トランスフォーマー (ViT) の自動設計において有望な性能を示しています。ただし、さまざまなモバイル デバイス向けに軽量で低レイテンシの ViT モデルを設計することは、依然として大きな課題です。この作業では、幅広いモバイル デバイスをサポートする非常に大きな検索スペースで高品質の ViT スーパーネットをトレーニングし、最適なサブネットワーク (サブネット) を直接検索する 2 段階の NAS アプローチである ElasticViT を提案します。展開。ただし、均一なサンプリングに依存する以前のスーパーネット トレーニング方法には、勾配の競合の問題があります。サンプリングされたサブネットのモデル サイズが大きく異なる場合があり (たとえば、50M と 2G FLOP)、最適化の方向が異なり、パフォーマンスが低下します。この課題に対処するために、複雑さを意識したサンプリングとパフォーマンスを意識したサンプリングという 2 つの新しいサンプリング手法を提案します。複雑さを意識したサンプリングは、検索空間内の異なるサイズのサブネットをカバーしながら、隣接するトレーニング ステップでサンプリングされたサブネット間の FLOP の差を制限します。パフォーマンスを意識したサンプリングにより、精度の高いサブネットがさらに選択されるため、勾配の競合が減少し、スーパーネットの品質が向上します。私たちが発見したモデルである ElasticViT モデルは、追加の再トレーニングなしで 60M から 800M FLOP の ImageNet で 67.2% から 80.0% のトップ 1 精度を達成し、精度とレイテンシーの点で以前のすべての CNN と ViT を上回っています。私たちの小さなモデルと小さなモデルは、最先端の CNN を凌駕する最初の ViT モデルでもあり、モバイル デバイスでの待ち時間が大幅に短縮されます。たとえば、ElasticViT-S1 は EfficientNet-B0 よりも 2.62 倍速く、0.1% 高い精度で実行されます。
Neural Architecture Search (NAS) has shown promising performance in the automatic design of vision transformers (ViT) exceeding 1G FLOPs. However, designing lightweight and low-latency ViT models for diverse mobile devices remains a big challenge. In this work, we propose ElasticViT, a two-stage NAS approach that trains a high-quality ViT supernet over a very large search space that supports a wide range of mobile devices, and then searches an optimal sub-network (subnet) for direct deployment. However, prior supernet training methods that rely on uniform sampling suffer from the gradient conflict issue: the sampled subnets can have vastly different model sizes (e.g., 50M vs. 2G FLOPs), leading to different optimization directions and inferior performance. To address this challenge, we propose two novel sampling techniques: complexity-aware sampling and performance-aware sampling. Complexity-aware sampling limits the FLOPs difference among the subnets sampled across adjacent training steps, while covering different-sized subnets in the search space. Performance-aware sampling further selects subnets that have good accuracy, which can reduce gradient conflicts and improve supernet quality. Our discovered models, ElasticViT models, achieve top-1 accuracy from 67.2% to 80.0% on ImageNet from 60M to 800M FLOPs without extra retraining, outperforming all prior CNNs and ViTs in terms of accuracy and latency. Our tiny and small models are also the first ViT models that surpass state-of-the-art CNNs with significantly lower latency on mobile devices. For instance, ElasticViT-S1 runs 2.62x faster than EfficientNet-B0 with 0.1% higher accuracy.