arXiv reaDer
FNAS:不確実性を意識した高速ニューラルアーキテクチャ検索
FNAS: Uncertainty-Aware Fast Neural Architecture Search
強化学習(RL)ベースのニューラルアーキテクチャ検索(NAS)は、一般に、より良い収束を保証しますが、ロールアウトのボトルネック(プロキシタスクでサンプリングされた各世代の徹底的なトレーニング)のため、勾配ベースのアプローチと比較して膨大な計算リソースの要件に悩まされます。このホワイトペーパーでは、NASでのRLプロセスだけでなくロールアウトプロセスのコンバージェンスを加速するための一般的なパイプラインを提案します。アーキテクチャとパラメータの知識の両方が、異なる実験間、さらには異なるタスク間で転送できるという興味深い観察に動機付けられています。最初に、Proximal Policy Optimization(PPO)に不確実性を意識した批評家(値関数)を導入して、以前の実験のアーキテクチャ知識を活用します。これにより、トレーニングプロセスが安定し、検索時間が4分の1に短縮されます。さらに、パラメータ知識を利用し、検索時間を2分の1に短縮するために、ブロック類似性関数とともにアーキテクチャ知識プールを提案します。これは、RLベースのNASにブロックレベルの重み共有を導入した最初の製品です。ブロック類似性関数は、厳密な公平性で100%のヒット率を保証します。さらに、RL最適化の「再生バッファー」で使用される単純に設計されたポリシー外補正係数により、検索時間の半分をさらに短縮できることを示します。モバイルニューラルアーキテクチャ検索(MNAS)検索スペースでの実験は、提案された高速ニューラルアーキテクチャ検索(FNAS)が標準のRLベースのNASプロセスを最大10倍高速化することを示しています(例:約256 2x2 TPUv2x日/ 20,000 GPUx時間-> 2,000 GPU x MNASの場合は1時間)、さまざまなビジョンタスクでのパフォーマンスの向上を保証します。
Reinforcement learning (RL)-based neural architecture search (NAS) generally guarantees better convergence yet suffers from the requirement of huge computational resources compared with gradient-based approaches, due to the rollout bottleneck -- exhaustive training for each sampled generation on proxy tasks. In this paper, we propose a general pipeline to accelerate the convergence of the rollout process as well as the RL process in NAS. It is motivated by the interesting observation that both the architecture and the parameter knowledge can be transferred between different experiments and even different tasks. We first introduce an uncertainty-aware critic (value function) in Proximal Policy Optimization (PPO) to utilize the architecture knowledge in previous experiments, which stabilizes the training process and reduces the searching time by 4 times. Further, an architecture knowledge pool together with a block similarity function is proposed to utilize parameter knowledge and reduces the searching time by 2 times. It is the first to introduce block-level weight sharing in RLbased NAS. The block similarity function guarantees a 100% hitting ratio with strict fairness. Besides, we show that a simply designed off-policy correction factor used in "replay buffer" in RL optimization can further reduce half of the searching time. Experiments on the Mobile Neural Architecture Search (MNAS) search space show the proposed Fast Neural Architecture Search (FNAS) accelerates standard RL-based NAS process by ~10x (e.g. ~256 2x2 TPUv2 x days / 20,000 GPU x hour -> 2,000 GPU x hour for MNAS), and guarantees better performance on various vision tasks.
updated: Thu May 27 2021 07:53:59 GMT+0000 (UTC)
published: Tue May 25 2021 06:32:52 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アソシエイト