arXiv reaDer
最適なニューラルネットワークの抽出:多様な空間での迅速な検索
Distilling Optimal Neural Networks: Rapid Search in Diverse Spaces
現在の最先端のニューラルアーキテクチャ検索(NAS)メソッドは、複数のハードウェアプラットフォームに効率的に拡張することも、多様なアーキテクチャ検索スペースを処理することもできません。これを改善するために、DONNA(Distilling Optimal Neural Network Architectures)を紹介します。これは、多くのユーザーシナリオに対応できる、高速でスケーラブルで多様なNASの新しいパイプラインです。 DONNAは3つのフェーズで構成されています。最初に、参照モデルからのブロックごとの知識蒸留を使用して、精度予測子が構築されます。この予測子を使用すると、レイヤータイプやアテンションメカニズムなどのさまざまなマクロアーキテクチャパラメータ、およびブロックリピートや拡張率などのマイクロアーキテクチャパラメータを使用して、さまざまなネットワーク全体を検索できます。第2に、迅速な進化的検索により、精度予測子とデバイス上の測定値を使用して、あらゆるシナリオに最適なアーキテクチャのセットが見つかります。第三に、最適なモデルは、ゼロからのトレーニングの精度にすばやく微調整されます。 DONNAは、デバイス上で最先端のアーキテクチャを見つける点で、MNasNetよりも最大100倍高速です。 ImageNetを分類すると、DONNAアーキテクチャはNvidia V100 GPUのEfficientNet-B0およびMobileNetV2よりも20%速く、Samsung S20スマートフォンのMobileNetV2-1.4xよりも0.5%高い精度で10%高速です。 NASに加えて、DONNAは、検索スペースの拡張と探索、およびハードウェア対応のモデル圧縮に使用されます。
Current state-of-the-art Neural Architecture Search (NAS) methods neither efficiently scale to multiple hardware platforms, nor handle diverse architectural search-spaces. To remedy this, we present DONNA (Distilling Optimal Neural Network Architectures), a novel pipeline for rapid, scalable and diverse NAS, that scales to many user scenarios. DONNA consists of three phases. First, an accuracy predictor is built using blockwise knowledge distillation from a reference model. This predictor enables searching across diverse networks with varying macro-architectural parameters such as layer types and attention mechanisms, as well as across micro-architectural parameters such as block repeats and expansion rates. Second, a rapid evolutionary search finds a set of pareto-optimal architectures for any scenario using the accuracy predictor and on-device measurements. Third, optimal models are quickly finetuned to training-from-scratch accuracy. DONNA is up to 100x faster than MNasNet in finding state-of-the-art architectures on-device. Classifying ImageNet, DONNA architectures are 20% faster than EfficientNet-B0 and MobileNetV2 on a Nvidia V100 GPU and 10% faster with 0.5% higher accuracy than MobileNetV2-1.4x on a Samsung S20 smartphone. In addition to NAS, DONNA is used for search-space extension and exploration, as well as hardware-aware model compression.
updated: Fri Aug 27 2021 13:02:16 GMT+0000 (UTC)
published: Wed Dec 16 2020 11:00:19 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アソシエイト