Performance prediction based on neural architecture features

Duo Long, Shizhou Zhang, Yanning Zhang

Research output: Contribution to journalArticlepeer-review

8 Scopus citations

Abstract

Neural Architecture Search (NAS) usually requires to train quantities of candidate neural networks on a dataset for choosing a high-performance network architecture and optimising hyperparameters, which is very time consuming and computationally expensive. In order to resolve the issue, the authors try to use a performance prediction method to predict the model performance with little or even no training steps. They assume that the performance is determined once the architecture and other hyperparameters are chosen. So they first extract the sequence features of the chain-structured neural architecture by introducing the N-grams model to process architecture textual description. Subsequently, based on the extracted neural architecture features, they use the appropriate regression model to predict validation accuracies for a modelling learning curve. Through a series of experimental comparisons, they verify the effectiveness of the authors' proposed performance prediction method and the effective acceleration of the NAS process.

Original languageEnglish
Pages (from-to)80-83
Number of pages4
JournalCognitive Computation and Systems
Volume2
Issue number2
DOIs
StatePublished - 1 Jun 2020

Fingerprint

Dive into the research topics of 'Performance prediction based on neural architecture features'. Together they form a unique fingerprint.

Cite this