A Study on Encodings for Neural Architecture Search
Colin White, Willie Neiswanger, Sam Nolen, Yash Savani
Abstract
Neural architecture search (NAS) has been extensively studied in the past few years. A popular approach is to represent each neural architecture in the search space as a directed acyclic graph (DAG), and then search over all DAGs by encoding the adjacency matrix and list of operations as a set of hyperparameters. Recent work has demonstrated that even small changes to the way each architecture is encoded can have a significant effect on the performance of NAS algorithms. In this work, we present the first formal study on the effect of architecture encodings for NAS, including a theoretical grounding and an empirical study. First we formally define architecture encodings and give a theoretical characterization on the scalability of the encodings we study Then we identify the main encoding-dependent subroutines which NAS algorithms employ, running experiments to show which encodings work best with each subroutine for many popular algorithms. The experiments act as an ablation study for prior work, disentangling the algorithmic and encoding-based contributions, as well as a guideline for future work. Our results demonstrate that NAS encodings are an important design decision which can have a significant impact on overall performance. Our code is available at this https URL.
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 papers20
- BANANAS: Bayesian Optimization with Neural Architectures for Neural Architecture SearchColin White, Willie Neiswanger, Yash SavaniAAAI 2021 · 401 citations
- How Powerful are Performance Predictors in Neural Architecture Search?Colin White, Arber Zela, Robin Ru, Yang Liu et al.NeurIPS 2021 · 168 citations
- Does Unsupervised Architecture Representation Learning Help Neural Architecture Search?Shen Yan, Yu Zheng, Wei Ao, Xiao Zeng et al.NeurIPS 2020 · 129 citations
- 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
- Evaluating Efficient Performance Estimators of Neural ArchitecturesXuefei Ning, Changcheng Tang, Wenshuo Li, Zixuan Zhou et al.NeurIPS 2021 · 99 citations
Builds on2
Related papers
- Encodings for Prediction-based Neural Architecture SearchYash Akhauri, Mohamed S. AbdelfattahICML 2024 · 8 citations
- Neural Graph Embedding for Neural Architecture SearchWei Li, Shaogang Gong, Xiatian ZhuAAAI 2020 · 31 citations
- CATE: Computation-aware Neural Architecture Encoding with TransformersShen Yan, Kaiqiang Song, Fei Liu, Mi ZhangICML 2021 · 35 citations
- TA-GATES: An Encoding Scheme for Neural Network ArchitecturesXuefei Ning, Zixuan Zhou, Junbo Zhao, Tianchen Zhao et al.NeurIPS 2022 · 19 citations
- PACE: A Parallelizable Computation Encoder for Directed Acyclic GraphsZehao Dong, Muhan Zhang, Fuhai Li, Yixin ChenICML 2022 · 24 citations
