arXiv reaDer
グラフの畳み込みネットワークによる重み共有NASの検索スペースの適合
Fitting the Search Space of Weight-sharing NAS with Graph Convolutional Networks
 ニューラルアーキテクチャの検索は、学界と産業界の両方で大きな注目を集めています。それを加速するために、研究者は最初にスーパーネットワークをトレーニングして異なるオペレーター間で計算を再利用する重み共有方法を提案しました。これらの方法は、計算コストの点で大きな利点がありますが、サンプリングされたサブネットワークは、個々のトレーニングプロセスが実行されない限り、正確に推定されるとは限りません。この論文は、そのような不正確さは、組み立てられたネットワーク層間の不可避のミスマッチによるものであり、そのため、各推定にランダムな誤差項が追加されています。ランダムなエラーの影響が最小限になるように、サンプリングされたサブネットワークのパフォーマンスに合わせてグラフのたたみ込みネットワークをトレーニングすることで、この問題を緩和します。この戦略を使用すると、選択した候補のセットでより高いランク相関係数が得られ、その結果、最終的なアーキテクチャのパフォーマンスが向上します。さらに、グラフ畳み込みネットワークは、検索スペース全体でのアーキテクチャのパフォーマンスの効率的なルックアップテーブルを提供しているため、さまざまなハードウェア制約の下で使用できるという柔軟性もあります。
Neural architecture search has attracted wide attentions in both academia and industry. To accelerate it, researchers proposed weight-sharing methods which first train a super-network to reuse computation among different operators, from which exponentially many sub-networks can be sampled and efficiently evaluated. These methods enjoy great advantages in terms of computational costs, but the sampled sub-networks are not guaranteed to be estimated precisely unless an individual training process is taken. This paper owes such inaccuracy to the inevitable mismatch between assembled network layers, so that there is a random error term added to each estimation. We alleviate this issue by training a graph convolutional network to fit the performance of sampled sub-networks so that the impact of random errors becomes minimal. With this strategy, we achieve a higher rank correlation coefficient in the selected set of candidates, which consequently leads to better performance of the final architecture. In addition, our approach also enjoys the flexibility of being used under different hardware constraints, since the graph convolutional network has provided an efficient lookup table of the performance of architectures in the entire search space.
updated: Tue Dec 15 2020 09:47:03 GMT+0000 (UTC)
published: Fri Apr 17 2020 19:12:39 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アソシエイト