GPNAS: A Neural Network Architecture Search Framework Based on Graphical Predictor

Abstract

In practice, the problems encountered in Neural Architecture Search (NAS) training are not simple problems, but often a series of difficult combinations (wrong compensation estimation, curse of dimension, overfitting, high complexity, etc.). In this paper, we propose a framework to decouple network structure from operator search space, and use two BOHBs to search alternatively. Considering that activation function and initialization are also important parts of neural network, the generalization ability of the model will be affected. We introduce an activation function and an initialization method domain, and add them into the operator search space to form a generalized search space, so as to improve the generalization ability of the child model. We then trained a GCN-based predictor using feedback from the child model. This can not only improve the search efficiency, but also solve the problem of dimension curse. Next, unlike other NAS studies, we used predictors to analyze the stability of different network structures. Finally, we applied our framework to neural structure search and achieved significant improvements on multiple datasets.

12 pages,15 figures

References in corpus (10)

• Neural Architecture Search with Reinforcement Learning
• Improved Regularization of Convolutional Neural Networks with Cutout
• Designing Neural Network Architectures using Reinforcement Learning
• Shake-Shake regularization
• NAS-Bench-101: Towards Reproducible Neural Architecture Search
• Peephole: Predicting Network Performance Before Training
• NAS-Bench-1Shot1: Benchmarking and Dissecting One-shot Neural Architecture Search
• Tabular Benchmarks for Joint Architecture and Hyperparameter Optimization
• APQ: Joint Search for Network Architecture, Pruning and Quantization Policy
• Differential Evolution for Neural Architecture Search