arXiv reaDer
多様な計算予算のためのパレート認識型ニューラル アーキテクチャの生成
Pareto-aware Neural Architecture Generation for Diverse Computational Budgets
さまざまなアプリケーション/デバイスによって発生するさまざまな計算予算の下で実行可能で効果的なアーキテクチャを設計することは、実際のアプリケーションで深いモデルを展開するために不可欠です。この目標を達成するために、既存の方法では多くの場合、ターゲット予算ごとに独立したアーキテクチャ検索プロセスが実行されますが、これは非常に非効率的で不要です。さらに重要なことに、これらの独立した検索プロセスは、学習した知識 (つまり、優れたアーキテクチャの配布) を互いに共有できないため、多くの場合、限られた検索結果しか得られません。これらの問題に対処するために、パレートを認識するニューラル アーキテクチャ ジェネレーター (PNAG) を提案します。PNAG は、一度トレーニングするだけで、推論によって任意の予算に対してパレート最適アーキテクチャを動的に生成します。 PNAG をトレーニングするために、さまざまな予算の下で複数のパレート最適アーキテクチャを共同で見つけることにより、パレート フロンティア全体を学習します。このような共同検索アルゴリズムは、全体的な検索コストを大幅に削減するだけでなく、検索結果も改善します。 3 つのハードウェア プラットフォーム (つまり、モバイル デバイス、CPU、および GPU) での広範な実験により、既存の方法に対する本方法の優位性が示されました。
Designing feasible and effective architectures under diverse computational budgets, incurred by different applications/devices, is essential for deploying deep models in real-world applications. To achieve this goal, existing methods often perform an independent architecture search process for each target budget, which is very inefficient yet unnecessary. More critically, these independent search processes cannot share their learned knowledge (i.e., the distribution of good architectures) with each other and thus often result in limited search results. To address these issues, we propose a Pareto-aware Neural Architecture Generator (PNAG) which only needs to be trained once and dynamically produces the Pareto optimal architecture for any given budget via inference. To train our PNAG, we learn the whole Pareto frontier by jointly finding multiple Pareto optimal architectures under diverse budgets. Such a joint search algorithm not only greatly reduces the overall search cost but also improves the search results. Extensive experiments on three hardware platforms (i.e., mobile device, CPU, and GPU) show the superiority of our method over existing methods.
updated: Fri Oct 14 2022 08:30:59 GMT+0000 (UTC)
published: Fri Oct 14 2022 08:30:59 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アソシエイト