1. 首页
  2. 人工智能
  3. 论文/代码
  4. GreedyNAS:通过贪婪超级网络实现快速一键式NAS

GreedyNAS:通过贪婪超级网络实现快速一键式NAS

上传者: 2021-01-22 15:11:02上传 .PDF文件 1.36 MB 热度 31次

训练超级网络对于一次性神经体系结构搜索(NAS)方法很重要,因为它充当不同体系结构(路径)的基本性能估算器。当前的方法主要假设超网应该在所有路径上给出合理的排名。..

GreedyNAS: Towards Fast One-Shot NAS with Greedy Supernet

Training a supernet matters for one-shot neural architecture search (NAS) methods since it serves as a basic performance estimator for different architectures (paths). Current methods mainly hold the assumption that a supernet should give a reasonable ranking over all paths.They thus treat all paths equally, and spare much effort to train paths. However, it is harsh for a single supernet to evaluate accurately on such a huge-scale search space (e.g., $7^{21}$). In this paper, instead of covering all paths, we ease the burden of supernet by encouraging it to focus more on evaluation of those potentially-good ones, which are identified using a surrogate portion of validation data. Concretely, during training, we propose a multi-path sampling strategy with rejection, and greedily filter the weak paths. The training efficiency is thus boosted since the training space has been greedily shrunk from all paths to those potentially-good ones. Moreover, we further adopt an exploration and exploitation policy by introducing an empirical candidate path pool. Our proposed method GreedyNAS is easy-to-follow, and experimental results on ImageNet dataset indicate that it can achieve better Top-1 accuracy under same search space and FLOPs or latency level, but with only $\sim$60\% of supernet training cost. By searching on a larger space, our GreedyNAS can also obtain new state-of-the-art architectures.

下载地址
用户评论