arXiv reaDer
効率的なニューラルアーキテクチャ検索のための滑らかな変分グラフ埋め込み
Smooth Variational Graph Embeddings for Efficient Neural Architecture Search
ニューラルアーキテクチャ検索(NAS)は、離散的なサンプリングベースの方法や効率的な微分可能アプローチなど、さまざまな方向から最近取り上げられています。前者は高額なことで有名ですが、後者は検索スペースに強い制約を課すことに悩まされています。学習した埋め込みスペースから、たとえばグラフニューラルネットワークベースの変分オートエンコーダを介したアーキテクチャの最適化は、中間点を構築し、両側からの利点を活用します。このようなアプローチは、最近、いくつかのベンチマークで良好なパフォーマンスを示しています。それでも、それらの安定性と予測力は、埋め込みスペースからネットワークを再構築する能力に大きく依存します。本論文では、様々な探索空間から神経構造をスムーズに符号化し、正確に再構築することを可能にする両面変分グラフオートエンコーダを提案する。 ENASアプローチ、NAS-Bench-101およびNAS-Bench-201検索スペースによって定義されたニューラルアーキテクチャで提案されたアプローチを評価し、スムーズな埋め込みスペースにより、実行可能領域外のアーキテクチャにパフォーマンス予測を直接外挿できることを示します(たとえば、より多くの操作で)。したがって、費用のかかるベイズ最適化や強化学習がなくても、優れたネットワークアーキテクチャを予測しやすくなります。
Neural architecture search (NAS) has recently been addressed from various directions, including discrete, sampling-based methods and efficient differentiable approaches. While the former are notoriously expensive, the latter suffer from imposing strong constraints on the search space. Architecture optimization from a learned embedding space for example through graph neural network based variational autoencoders builds a middle ground and leverages advantages from both sides. Such approaches have recently shown good performance on several benchmarks. Yet, their stability and predictive power heavily depends on their capacity to reconstruct networks from the embedding space. In this paper, we propose a two-sided variational graph autoencoder, which allows to smoothly encode and accurately reconstruct neural architectures from various search spaces. We evaluate the proposed approach on neural architectures defined by the ENAS approach, the NAS-Bench-101 and the NAS-Bench-201 search space and show that our smooth embedding space allows to directly extrapolate the performance prediction to architectures outside the seen domain (e.g. with more operations). Thus, it facilitates to predict good network architectures even without expensive Bayesian optimization or reinforcement learning.
updated: Wed May 12 2021 12:44:54 GMT+0000 (UTC)
published: Fri Oct 09 2020 17:05:41 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アソシエイト