arXiv reaDer
Zen-NAS:高性能の深部画像認識のためのゼロショットNAS
Zen-NAS: A Zero-Shot NAS for High-Performance Deep Image Recognition
精度予測子は、アーキテクチャをランク付けするためのニューラルアーキテクチャ検索(NAS)の重要なコンポーネントです。高品質の精度予測子を構築するには、通常、膨大な計算が必要です。この問題に対処するために、精度予測子を使用する代わりに、Zen-Scoreと呼ばれる新しいゼロショットインデックスを提案して、アーキテクチャをランク付けします。 Zen-Scoreはネットワークの表現力を表し、モデルの精度と正の相関関係があります。 Zen-Scoreの計算では、ネットワークパラメータをトレーニングせずに、ランダムに初期化されたネットワークを介していくつかの前方推論を行うだけです。 Zen-Scoreに基づいて構築され、与えられた推論バジェットの下でターゲットネットワークのZen-Scoreを最大化することにより、Zen-NASと呼ばれる新しいNASアルゴリズムをさらに提案します。 Zen-NASは、GPUの半日以内に、データのないスタイルで高性能アーキテクチャを直接検索できます。以前のNAS手法と比較すると、提案されたZen-NASは、複数のサーバー側およびモバイル側のGPUプラットフォームで、ImageNetの最先端の精度で、数倍高速です。ソースコードと事前トレーニング済みモデルは、https://github.com/idstcv/ZenNASでリリースされています。
Accuracy predictor is a key component in Neural Architecture Search (NAS) for ranking architectures. Building a high-quality accuracy predictor usually costs enormous computation. To address this issue, instead of using an accuracy predictor, we propose a novel zero-shot index dubbed Zen-Score to rank the architectures. The Zen-Score represents the network expressivity and positively correlates with the model accuracy. The calculation of Zen-Score only takes a few forward inferences through a randomly initialized network, without training network parameters. Built upon the Zen-Score, we further propose a new NAS algorithm, termed as Zen-NAS, by maximizing the Zen-Score of the target network under given inference budgets. Within less than half GPU day, Zen-NAS is able to directly search high performance architectures in a data-free style. Comparing with previous NAS methods, the proposed Zen-NAS is magnitude times faster on multiple server-side and mobile-side GPU platforms with state-of-the-art accuracy on ImageNet. Our source code and pre-trained models are released on https://github.com/idstcv/ZenNAS.
updated: Thu Mar 25 2021 23:33:28 GMT+0000 (UTC)
published: Mon Feb 01 2021 18:53:40 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アソシエイト