arXiv reaDer
RANK-NOSH:不均一な連続半減による効率的な予測子ベースのアーキテクチャ検索
RANK-NOSH: Efficient Predictor-Based Architecture Search via Non-Uniform Successive Halving
予測子ベースのアルゴリズムは、ニューラルアーキテクチャ検索(NAS)タスクで驚くべきパフォーマンスを達成しました。ただし、パフォーマンスプレディクタをトレーニングするには、通常、何百ものアーキテクチャを最初からトレーニングして評価する必要があるため、これらの方法には高い計算コストがかかります。この方針に沿ったこれまでの作業は、主に、予測子を適合させるために必要なアーキテクチャの数を減らすことに焦点を当てています。この作業では、この課題に別の観点から取り組みます。アーキテクチャトレーニングの計算予算を削減することで、検索効率を向上させます。予算の浪費を回避するために、パフォーマンスの低いアーキテクチャのトレーニングを早期に終了する階層型スケジューリングアルゴリズムであるNOn-uniform Successive Halving(NOSH)を提案します。 NOSHによって生成された不均一な監視信号を効果的に活用するために、ペアワイズ比較でランク付けすることを学習するものとして、予測子ベースのアーキテクチャ検索を定式化します。結果として得られる方法であるRANK-NOSHは、さまざまなスペースやデータセットで以前の最先端の予測子ベースの方法よりも競争力のある、またはさらに優れたパフォーマンスを実現しながら、検索予算を最大5分の1に削減します。
Predictor-based algorithms have achieved remarkable performance in the Neural Architecture Search (NAS) tasks. However, these methods suffer from high computation costs, as training the performance predictor usually requires training and evaluating hundreds of architectures from scratch. Previous works along this line mainly focus on reducing the number of architectures required to fit the predictor. In this work, we tackle this challenge from a different perspective - improve search efficiency by cutting down the computation budget of architecture training. We propose NOn-uniform Successive Halving (NOSH), a hierarchical scheduling algorithm that terminates the training of underperforming architectures early to avoid wasting budget. To effectively leverage the non-uniform supervision signals produced by NOSH, we formulate predictor-based architecture search as learning to rank with pairwise comparisons. The resulting method - RANK-NOSH, reduces the search budget by ~5x while achieving competitive or even better performance than previous state-of-the-art predictor-based methods on various spaces and datasets.
updated: Wed Aug 18 2021 07:45:21 GMT+0000 (UTC)
published: Wed Aug 18 2021 07:45:21 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アソシエイト