Abstract:Neural architecture search (NAS) has achieved breakthrough success in a great number of applications in the past few years. It could be time to take a step back and analyze the good and bad aspects in the field of NAS. A variety of algorithms search architectures under different search space. These searched architectures are trained using different setups, e.g., hyper-parameters, data augmentation, regularization. This raises a comparability problem when comparing the performance of various NAS algorithms. NAS-Bench-101 has shown success to alleviate this problem. In this work, we propose an extension to NAS-Bench-101: NAS-Bench201 with a different search space, results on multiple datasets, and more diagnostic information. NAS-Bench-201 has a fixed search space and provides a unified benchmark for almost any up-to-date NAS algorithms. The design of our search space is inspired from the one used in the most popular cell-based searching algorithms, where a cell is represented as a directed acyclic graph. Each edge here is associated with an operation selected from a predefined operation set. For it to be applicable for all NAS algorithms, the search space defined in NAS-Bench-201 includes all possible architectures generated by 4 nodes and 5 associated operation options, which results in 15,625 neural cell candidates in total. The training log using the same setup and the performance for each architecture candidate are provided for three datasets. This allows researchers to avoid unnecessary repetitive training for selected architecture and focus solely on the search algorithm itself. The training time saved for every architecture also largely improves the efficiency of most NAS algorithms and brings a more computational cost friendly NAS community for a broader range of researchers. We provide additional diagnostic information such as fine-grained loss and accuracy, which can give inspirations to new designs of NAS algorithms. In further support of the proposed NAS-Bench201, we have analyzed it from many aspects and benchmarked 10 recent NAS algorithms, which verify its applicability.

A review of neural architecture search

Advances in neural architecture search

A Comprehensive Survey of Neural Architecture Search: Challenges and Solutions

A Technical View on Neural Architecture Search

Evolution and Efficiency in Neural Architecture Search: Bridging the Gap Between Expert Design and Automated Optimization

Neural Architecture Search Survey: A Computer Vision Perspective

A Survey on Neural Architecture Search

Neural Architecture Search: Insights from 1000 Papers

Evaluating the Search Phase of Neural Architecture Search

A Survey on Evolutionary Neural Architecture Search

Automating Neural Architecture Design without Search

Trends in Neural Architecture Search: Towards the Acceleration of Search

A Survey on Computationally Efficient Neural Architecture Search

Understanding Architectures Learnt by Cell-based Neural Architecture Search

Data Aware Neural Architecture Search

Understanding Neural Architecture Search Techniques

An Introduction to Neural Architecture Search for Convolutional Networks

Architecture Self-Attention Mechanism: Nonlinear Optimization for Neural Architecture Search

An Approach for Efficient Neural Architecture Search Space Definition

Efficient Automation of Neural Network Design: A Survey on Differentiable Neural Architecture Search

Producible neural architecture search