arXiv reaDer
コンビナトリアル多腕バンディットによるニューラルアーキテクチャ検索
Neural Architecture Search via Combinatorial Multi-Armed Bandit
ニューラルアーキテクチャ検索(NAS)は、高性能のディープニューラルネットワーク(DNN)を設計するための効果的なツールとして大きな人気を博しています。 NASは、ポリシー勾配、進化的アルゴリズム、微分可能なアーキテクチャ検索、またはツリー検索方法を介して実行できます。ポリシー勾配と微分可能なアーキテクチャ検索の両方で大きな進歩が見られましたが、ツリー検索方法では、これまでのところ、同等の精度や検索効率を達成できていません。この論文では、NASを組み合わせ多腕バンディット(CMAB)問題(CMAB-NAS)として定式化します。これにより、大きな探索空間を小さなブロックに分解して、ツリー探索法をより効果的かつ効率的に適用できます。さらに、ネストされたモンテカルロ検索と呼ばれるツリーベースの方法を活用して、CMAB-NASの問題に取り組みます。 CIFAR-10で、私たちのアプローチは、現在のツリー検索方法より20倍速いわずか0.58 GPU日を使用して、最先端技術に匹敵する低いエラー率を達成するセル構造を発見します。さらに、発見された構造は、ImageNetなどの大規模なデータセットにうまく転送されます。
Neural Architecture Search (NAS) has gained significant popularity as an effective tool for designing high performance deep neural networks (DNNs). NAS can be performed via policy gradient, evolutionary algorithms, differentiable architecture search or tree-search methods. While significant progress has been made for both policy gradient and differentiable architecture search, tree-search methods have so far failed to achieve comparable accuracy or search efficiency. In this paper, we formulate NAS as a Combinatorial Multi-Armed Bandit (CMAB) problem (CMAB-NAS). This allows the decomposition of a large search space into smaller blocks where tree-search methods can be applied more effectively and efficiently. We further leverage a tree-based method called Nested Monte-Carlo Search to tackle the CMAB-NAS problem. On CIFAR-10, our approach discovers a cell structure that achieves a low error rate that is comparable to the state-of-the-art, using only 0.58 GPU days, which is 20 times faster than current tree-search methods. Moreover, the discovered structure transfers well to large-scale datasets such as ImageNet.
updated: Sat Apr 24 2021 14:13:15 GMT+0000 (UTC)
published: Fri Jan 01 2021 23:29:33 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アソシエイト