NAS-Bench-Suite: NAS Evaluation is (Now) Surprisingly Easy
Yash Mehta, Colin White, Arber Zela, Arjun Krishnakumar, Guri Zabergja, Shakiba Moradian, Mahmoud Safari, Kaicheng Yu, Frank Hutter
Abstract
The release of tabular benchmarks, such as NAS-Bench-101 and NAS-Bench-201, has significantly lowered the computational overhead for conducting scientific research in neural architecture search (NAS). Although they have been widely adopted and used to tune real-world NAS algorithms, these benchmarks are limited to small search spaces and focus solely on image classification. Recently, several new NAS benchmarks have been introduced that cover significantly larger search spaces over a wide range of tasks, including object detection, speech recognition, and natural language processing. However, substantial differences among these NAS benchmarks have so far prevented their widespread adoption, limiting researchers to using just a few benchmarks. In this work, we present an in-depth analysis of popular NAS algorithms and performance prediction methods across 25 different combinations of search spaces and datasets, finding that many conclusions drawn from a few NAS benchmarks do not generalize to other benchmarks. To help remedy this problem, we introduce NAS-Bench-Suite, a comprehensive and extensible collection of NAS benchmarks, accessible through a unified interface, created with the aim to facilitate reproducible, generalizable, and rapid NAS research. Our code is available at https://github.com/automl/naslib .
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers13
- Generalized Global Ranking-Aware Neural Architecture Ranker for Efficient Image Classifier SearchBicheng Guo, Tao Chen, Shibo He, Haoyu Liu et al.ACM MM 2022 · 21 citations
- Sample complexity of data-driven tuning of model hyperparameters in neural networks with structured parameter-dependent dual functionMaria-Florina Balcan, Anh Nguyen, Dravyansh SharmaNeurIPS 2025 · 14 citations
- TabNAS: Rejection Sampling for Neural Architecture Search on Tabular DatasetsChengrun Yang, Gabriel Bender, Hanxiao Liu, Pieter-Jan Kindermans et al.NeurIPS 2022 · 9 citations
- Encodings for Prediction-based Neural Architecture SearchYash Akhauri, Mohamed S. AbdelfattahICML 2024 · 8 citations
- Shortest Edit Path Crossover: A Theory-driven Solution to the Permutation Problem in Evolutionary Neural Architecture SearchXin Qiu, Risto MiikkulainenICML 2023 · 4 citations
Builds on17
- NAS-Bench-201: Extending the Scope of Reproducible Neural Architecture SearchXuanyi Dong, Yi YangICLR 2020 · 825 citations
- Understanding and Robustifying Differentiable Architecture SearchArber Zela, Thomas Elsken, Tonmoy Saikia, Yassine Marrakchi et al.ICLR 2020 · 408 citations
- BANANAS: Bayesian Optimization with Neural Architectures for Neural Architecture SearchColin White, Willie Neiswanger, Yash SavaniAAAI 2021 · 401 citations
- Evaluating The Search Phase of Neural Architecture SearchKaicheng Yu, Christian Sciuto, Martin Jaggi, Claudiu Musat et al.ICLR 2020 · 370 citations
- NAS evaluation is frustratingly hardAntoine Yang, Pedro M. Esperança, Fabio Maria CarlucciICLR 2020 · 180 citations
Related papers
- Surrogate NAS Benchmarks: Going Beyond the Limited Search Spaces of Tabular NAS BenchmarksArber Zela, Julien Niklas Siems, Lucas Zimmer, Jovita Lukasik et al.ICLR 2022 · 100 citations
- NAS-Bench-ASR: Reproducible Neural Architecture Search for Speech RecognitionAbhinav Mehrotra, Alberto Gil C. P. Ramos, Sourav Bhattacharya, Lukasz Dudziak et al.ICLR 2021 · 69 citations
- NAS-Bench-1Shot1: Benchmarking and Dissecting One-shot Neural Architecture SearchArber Zela, Julien Siems, Frank HutterICLR 2020 · 156 citations
- Generic Neural Architecture Search via RegressionYuhong Li, Cong Hao, Pan Li, Jinjun Xiong et al.NeurIPS 2021 · 40 citations
- TransNAS-Bench-101: Improving Transferability and Generalizability of Cross-Task Neural Architecture SearchYawen Duan, Xin Chen, Hang Xu, Zewei Chen et al.CVPR 2021
