Progressive Neural Architecture Search
About
We propose a new method for learning the structure of convolutional neural networks (CNNs) that is more efficient than recent state-of-the-art methods based on reinforcement learning and evolutionary algorithms. Our approach uses a sequential model-based optimization (SMBO) strategy, in which we search for structures in order of increasing complexity, while simultaneously learning a surrogate model to guide the search through structure space. Direct comparison under the same search space shows that our method is up to 5 times more efficient than the RL method of Zoph et al. (2018) in terms of number of models evaluated, and 8 times faster in terms of total compute. The structures we discover in this way achieve state of the art classification accuracies on CIFAR-10 and ImageNet.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Classification | CIFAR-100 (test) | -- | 3518 | |
| Image Classification | CIFAR-10 (test) | Accuracy97.5 | 3381 | |
| Image Classification | ImageNet-1k (val) | Top-1 Accuracy82.9 | 1453 | |
| Image Classification | ImageNet (val) | Top-1 Acc82.9 | 1206 | |
| Image Classification | CIFAR-10 (test) | -- | 906 | |
| Image Classification | ImageNet 1k (test) | Top-1 Accuracy82.9 | 798 | |
| Image Classification | CIFAR-100 | -- | 622 | |
| Image Classification | CIFAR-10 | -- | 471 | |
| Image Classification | ImageNet | Top-1 Accuracy82.9 | 429 | |
| Image Classification | ImageNet ILSVRC-2012 (val) | Top-1 Accuracy82.9 | 405 |