arXiv reaDer
ランダムウェイト評価による多目的ニューラルアーキテクチャ検索の高速化
Accelerating Multi-Objective Neural Architecture Search by Random-Weight Evaluation
高性能の深層畳み込みニューラルネットワーク(CNN)の自動設計を目的として、ニューラルアーキテクチャ検索(NAS)手法は、学界と産業界の両方にとってますます重要になっています。パフォーマンスのためのCNNのコストのかかる確率的勾配降下(SGD)トレーニングのため評価では、ほとんどの既存のNASメソッドは、実際の展開では計算コストが高くなります。この問題に対処するために、最初に、ランダムウェイト評価(RWE)という名前の新しいパフォーマンス推定メトリックを導入して、コスト効率の高い方法でCNNの品質を定量化します。 CNN全体を完全にトレーニングする代わりに、RWEは最後のレイヤーのみをトレーニングし、残りをランダムに初期化された重みで残すため、数秒で単一のネットワーク評価が行われます。次に、多目的NASに複雑さの指標を採用して、モデルのバランスを取ります。サイズとパフォーマンス。全体として、提案された方法は、2つの実世界の検索空間で最先端のパフォーマンスを備えた効率的なモデルのセットを取得します。次に、CIFAR-10データセットで取得された結果がImageNetデータセットに転送され、提案されたアルゴリズムの実用性が検証されます。さらに、NAS-Bench-301データセットのアブレーション研究により、既存の方法と比較したパフォーマンスの推定における提案されたRWEの有効性が明らかになりました。
For the goal of automated design of high-performance deep convolutional neural networks (CNNs), Neural Architecture Search (NAS) methodology is becoming increasingly important for both academia and industries.Due to the costly stochastic gradient descent (SGD) training of CNNs for performance evaluation, most existing NAS methods are computationally expensive for real-world deployments. To address this issue, we first introduce a new performance estimation metric, named Random-Weight Evaluation (RWE) to quantify the quality of CNNs in a cost-efficient manner. Instead of fully training the entire CNN, the RWE only trains its last layer and leaves the remainders with randomly initialized weights, which results in a single network evaluation in seconds.Second, a complexity metric is adopted for multi-objective NAS to balance the model size and performance. Overall, our proposed method obtains a set of efficient models with state-of-the-art performance in two real-world search spaces. Then the results obtained on the CIFAR-10 dataset are transferred to the ImageNet dataset to validate the practicality of the proposed algorithm. Moreover, ablation studies on NAS-Bench-301 datasets reveal the effectiveness of the proposed RWE in estimating the performance compared with existing methods.
updated: Fri Oct 08 2021 06:35:20 GMT+0000 (UTC)
published: Fri Oct 08 2021 06:35:20 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アソシエイト