We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Computer Vision and Pattern Recognition

Title: Progressive Neural Architecture Search

Abstract: 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.
Comments: To appear in ECCV 2018 as oral. The code and checkpoint for PNASNet-5 trained on ImageNet (both Mobile and Large) can now be downloaded from this https URL Also see this https URL for refactored and simplified TensorFlow code; see this https URL for exact conversion to PyTorch
Subjects: Computer Vision and Pattern Recognition (cs.CV); Machine Learning (cs.LG); Machine Learning (stat.ML)
Cite as: arXiv:1712.00559 [cs.CV]
  (or arXiv:1712.00559v3 [cs.CV] for this version)

Submission history

From: Chenxi Liu [view email]
[v1] Sat, 2 Dec 2017 06:23:16 GMT (750kb,D)
[v2] Sat, 24 Mar 2018 00:39:27 GMT (2037kb,D)
[v3] Thu, 26 Jul 2018 19:51:26 GMT (1539kb,D)

Link back to: arXiv, form interface, contact.