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 Technical View on Neural Architecture Search

Advances in neural architecture search

A Comprehensive Survey of Neural Architecture Search: Challenges and Solutions

Neural Architecture Search Survey: A Computer Vision Perspective

Neural Architecture Search: Insights from 1000 Papers

A review of neural architecture search

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

Evaluating the Search Phase of Neural Architecture Search

Generalized Global Ranking-Aware Neural Architecture Ranker for Efficient Image Classifier Search

An Approach for Efficient Neural Architecture Search Space Definition

Understanding Architectures Learnt by Cell-based Neural Architecture Search

Understanding Neural Architecture Search Techniques

A Survey on Computationally Efficient Neural Architecture Search

Neural Architecture Search for Dense Prediction Tasks in Computer Vision

Data Aware Neural Architecture Search

Trends in Neural Architecture Search: Towards the Acceleration of Search

Optimization-inspired Manual Architecture Design and Neural Architecture Search

Producible neural architecture search

Automating Neural Architecture Design without Search

An Introduction to Neural Architecture Search for Convolutional Networks

AutoSpace: Neural Architecture Search with Less Human Interference