Efficient NAS with FaDE on Hierarchical Spaces

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Neumeyer, Simon, Stier, Julian, Granitzer, Michael
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866914769909317632
author Neumeyer, Simon
Stier, Julian
Granitzer, Michael
author_facet Neumeyer, Simon
Stier, Julian
Granitzer, Michael
contents Neural architecture search (NAS) is a challenging problem. Hierarchical search spaces allow for cheap evaluations of neural network sub modules to serve as surrogate for architecture evaluations. Yet, sometimes the hierarchy is too restrictive or the surrogate fails to generalize. We present FaDE which uses differentiable architecture search to obtain relative performance predictions on finite regions of a hierarchical NAS space. The relative nature of these ranks calls for a memory-less, batch-wise outer search algorithm for which we use an evolutionary algorithm with pseudo-gradient descent. FaDE is especially suited on deep hierarchical, respectively multi-cell search spaces, which it can explore by linear instead of exponential cost and therefore eliminates the need for a proxy search space. Our experiments show that firstly, FaDE-ranks on finite regions of the search space correlate with corresponding architecture performances and secondly, the ranks can empower a pseudo-gradient evolutionary search on the complete neural architecture search space.
format Preprint
id arxiv_https___arxiv_org_abs_2404_16218
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Efficient NAS with FaDE on Hierarchical Spaces
Neumeyer, Simon
Stier, Julian
Granitzer, Michael
Neural and Evolutionary Computing
Artificial Intelligence
Machine Learning
I.2.6
Neural architecture search (NAS) is a challenging problem. Hierarchical search spaces allow for cheap evaluations of neural network sub modules to serve as surrogate for architecture evaluations. Yet, sometimes the hierarchy is too restrictive or the surrogate fails to generalize. We present FaDE which uses differentiable architecture search to obtain relative performance predictions on finite regions of a hierarchical NAS space. The relative nature of these ranks calls for a memory-less, batch-wise outer search algorithm for which we use an evolutionary algorithm with pseudo-gradient descent. FaDE is especially suited on deep hierarchical, respectively multi-cell search spaces, which it can explore by linear instead of exponential cost and therefore eliminates the need for a proxy search space. Our experiments show that firstly, FaDE-ranks on finite regions of the search space correlate with corresponding architecture performances and secondly, the ranks can empower a pseudo-gradient evolutionary search on the complete neural architecture search space.
title Efficient NAS with FaDE on Hierarchical Spaces
topic Neural and Evolutionary Computing
Artificial Intelligence
Machine Learning
I.2.6
url https://arxiv.org/abs/2404.16218