arXiv reaDer
生成による検索:アーキテクチャジェネレータを備えた柔軟で効率的なワンショットNAS
Searching by Generating: Flexible and Efficient One-Shot NAS with Architecture Generator
ワンショットNASでは、さまざまなハードウェアの制約を満たすために、スーパーネットからサブネットワークを検索する必要があります。ただし、検索コストは高く、N個の異なる制約に対してN回の検索が必要です。この作業では、サブネットワークを生成して検索するアーキテクチャジェネレータと呼ばれる新しい検索戦略を提案します。これにより、検索プロセスがはるかに効率的かつ柔軟になります。トレーニング済みのアーキテクチャジェネレータを使用すると、入力としてターゲットハードウェア制約が与えられ、再検索やスーパーネットの再トレーニングなしで、1回のフォワードパスでN個の制約に対してN個の優れたアーキテクチャを生成できます。さらに、検索効率をさらに向上させ、アーキテクチャジェネレータのGPUメモリ消費を削減するために、ユニファイドスーパーネットと呼ばれる新しいシングルパススーパーネットを提案します。アーキテクチャジェネレータと統合スーパーネットを使用して、Searching by Generating NAS(SGNAS)と呼ばれる柔軟で効率的なワンショットNASフレームワークを提案します。事前にトレーニングされたsuperntを使用すると、N個の異なるハードウェア制約に対するSGNASの検索時間はわずか5 GPU時間であり、これは以前のSOTAシングルパス方式よりも4N倍高速です。ゼロからトレーニングした後、ImageNetでのSGNASのトップ1の精度は77.1%であり、これはSOTAに匹敵します。コードはhttps://github.com/eric8607242/SGNASで入手できます。
In one-shot NAS, sub-networks need to be searched from the supernet to meet different hardware constraints. However, the search cost is high and N times of searches are needed for N different constraints. In this work, we propose a novel search strategy called architecture generator to search sub-networks by generating them, so that the search process can be much more efficient and flexible. With the trained architecture generator, given target hardware constraints as the input, N good architectures can be generated for N constraints by just one forward pass without re-searching and supernet retraining. Moreover, we propose a novel single-path supernet, called unified supernet, to further improve search efficiency and reduce GPU memory consumption of the architecture generator. With the architecture generator and the unified supernet, we propose a flexible and efficient one-shot NAS framework, called Searching by Generating NAS (SGNAS). With the pre-trained supernt, the search time of SGNAS for N different hardware constraints is only 5 GPU hours, which is 4N times faster than previous SOTA single-path methods. After training from scratch, the top1-accuracy of SGNAS on ImageNet is 77.1%, which is comparable with the SOTAs. The code is available at: https://github.com/eric8607242/SGNAS.
updated: Fri Mar 12 2021 14:04:50 GMT+0000 (UTC)
published: Fri Mar 12 2021 14:04:50 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アソシエイト