arXiv reaDer
4 GPU時間でのImageNetでのニューラルアーキテクチャ検索:理論的に着想を得た視点
Neural Architecture Search on ImageNet in Four GPU Hours: A Theoretically Inspired Perspective
ニューラルアーキテクチャ検索(NAS)は、最高のパフォーマンスを発揮するニューラルネットワークの発見を自動化するために爆発的に研究されてきました。現在の作業では、スーパーネットの大量のトレーニングまたは集中的なアーキテクチャ評価が必要であるため、リソースの消費量が多く、トレーニングや近似の切り捨てにより検索バイアスが発生することがよくあります。トレーニングを必要とせずに最適なニューラルアーキテクチャを選択し、検索コストの大幅な部分を排除できますか?トレーニングフリーのニューラルアーキテクチャ検索(TE-NAS)と呼ばれる新しいフレームワークを提案することにより、肯定的な答えを提供します。 TE-NASは、ニューラルタンジェントカーネル(NTK)のスペクトルと入力空間の線形領域の数を分析することにより、アーキテクチャをランク付けします。どちらも、ディープネットワークにおける最近の理論の進歩に動機付けられており、トレーニングやラベルなしで計算できます。 (1)これら2つの測定値は、ニューラルネットワークのトレーニング可能性と表現力を意味します。 (2)それらはネットワークのテスト精度と強く相関します。さらに、プルーニングベースのNASメカニズムを設計して、検索中のトレーニング可能性と表現力の間のより柔軟で優れたトレードオフを実現します。 NAS-Bench-201およびDARTS検索スペースでは、TE-NASは高品質の検索を完了しますが、CIFAR-10およびImageNetでそれぞれ1つの1080Tiで0.5および4GPU時間しかかかりません。私たちの仕事が、実際のNASアプリケーションにおけるディープネットワークの理論的発見と実際的な影響を橋渡しするためのより多くの試みを刺激することを願っています。コードはhttps://github.com/VITA-Group/TENASで入手できます。
Neural Architecture Search (NAS) has been explosively studied to automate the discovery of top-performer neural networks. Current works require heavy training of supernet or intensive architecture evaluations, thus suffering from heavy resource consumption and often incurring search bias due to truncated training or approximations. Can we select the best neural architectures without involving any training and eliminate a drastic portion of the search cost? We provide an affirmative answer, by proposing a novel framework called training-free neural architecture search (TE-NAS). TE-NAS ranks architectures by analyzing the spectrum of the neural tangent kernel (NTK) and the number of linear regions in the input space. Both are motivated by recent theory advances in deep networks and can be computed without any training and any label. We show that: (1) these two measurements imply the trainability and expressivity of a neural network; (2) they strongly correlate with the network's test accuracy. Further on, we design a pruning-based NAS mechanism to achieve a more flexible and superior trade-off between the trainability and expressivity during the search. In NAS-Bench-201 and DARTS search spaces, TE-NAS completes high-quality search but only costs 0.5 and 4 GPU hours with one 1080Ti on CIFAR-10 and ImageNet, respectively. We hope our work inspires more attempts in bridging the theoretical findings of deep networks and practical impacts in real NAS applications. Code is available at: https://github.com/VITA-Group/TENAS.
updated: Wed Mar 03 2021 16:17:27 GMT+0000 (UTC)
published: Tue Feb 23 2021 07:50:44 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アソシエイト