arXiv · 1906.07528
Prune and Replace NAS
Abstract
While recent NAS algorithms are thousands of times faster than the pioneering works, it is often overlooked that they use fewer candidate operations, resulting in a significantly smaller search space. We present PR-DARTS, a NAS algorithm that discovers strong network configurations in a much larger search space and a single day. A small candidate operation pool is used, from which candidates are progressively pruned and replaced with better performing ones. Experiments on CIFAR-10 and CIFAR-100 achieve 2.51% and 15.53% test error, respectively, despite searching in a space where each cell has 150 times as many possible configurations than in the DARTS baseline. Code is available at https://github.com/cogsys-tuebingen/prdarts
Explore related subjects
Keep this discovery
Kevin Alexander Laube, Andreas Zell. 2019-06-18. Prune and Replace NAS. https://doi.org/10.1109/icmla.2019.00158
Cite the original work for its findings. Save a collection to share your selection of sources.