arXiv reaDer
Arch-Graph:タスク転送可能なニューラルアーキテクチャ検索のための非巡回アーキテクチャ関係予測子
Arch-Graph: Acyclic Architecture Relation Predictor for Task-Transferable Neural Architecture Search
Neural Architecture Search(NAS)は、複数のタスクの効率的なモデルを見つけることを目的としています。単一のタスクのソリューションを探すだけでなく、ネットワーク設計の知識を複数のタスクに転送することへの関心が高まっています。この一連の研究では、タスクの相関関係を効果的にモデル化することが重要ですが、非常に無視されています。したがって、特定のタスクの埋め込みに関してタスク固有の最適なアーキテクチャを予測する転送可能なNASメソッドであるArch-Graphを提案します。高速適応のための予測子の入力の一部として埋め込みを使用することにより、複数のタスク間の相関関係を活用します。また、NASをアーキテクチャ関係グラフ予測問題として定式化し、候補アーキテクチャをノードとして扱い、それらのペアワイズ関係をエッジとして扱うことによって構築されたリレーショナルグラフを使用します。リレーショナルグラフの非周期性などのいくつかの基本的なプロパティを適用するために、最適化プロセスに追加の制約を追加し、NASを最大加重非周期サブグラフ(MWAS)を見つける問題に変換します。次に、私たちのアルゴリズムは、サイクルを排除し、ランクの結果が信頼できる場合にのみグラフのエッジを確立しようとします。 Arch-Graphは、MWASを通じて、わずかな予算で各タスクの候補モデルを効果的にランク付けして、予測子を微調整できます。 TransNAS-Bench-101での広範な実験により、Arch-Graphの転送可能性と多数のタスクにわたる高いサンプル効率を示し、シングルタスク検索とマルチタスク検索の両方用に設計された多くのNASメソッドを上回ります。わずか50モデルの予算で、2つの検索スペースで平均して上位0.16%と0.29%のアーキテクチャを見つけることができます。
Neural Architecture Search (NAS) aims to find efficient models for multiple tasks. Beyond seeking solutions for a single task, there are surging interests in transferring network design knowledge across multiple tasks. In this line of research, effectively modeling task correlations is vital yet highly neglected. Therefore, we propose Arch-Graph, a transferable NAS method that predicts task-specific optimal architectures with respect to given task embeddings. It leverages correlations across multiple tasks by using their embeddings as a part of the predictor's input for fast adaptation. We also formulate NAS as an architecture relation graph prediction problem, with the relational graph constructed by treating candidate architectures as nodes and their pairwise relations as edges. To enforce some basic properties such as acyclicity in the relational graph, we add additional constraints to the optimization process, converting NAS into the problem of finding a Maximal Weighted Acyclic Subgraph (MWAS). Our algorithm then strives to eliminate cycles and only establish edges in the graph if the rank results can be trusted. Through MWAS, Arch-Graph can effectively rank candidate models for each task with only a small budget to finetune the predictor. With extensive experiments on TransNAS-Bench-101, we show Arch-Graph's transferability and high sample efficiency across numerous tasks, beating many NAS methods designed for both single-task and multi-task search. It is able to find top 0.16% and 0.29% architectures on average on two search spaces under the budget of only 50 models.
updated: Tue Apr 12 2022 16:46:06 GMT+0000 (UTC)
published: Tue Apr 12 2022 16:46:06 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アソシエイト