Title
Auto-Keras: An Efficient Neural Architecture Search System
Abstract
Neural architecture search (NAS) has been proposed to automatically tune deep neural networks, but existing search algorithms, e.g., NASNet, PNAS, usually suffer from expensive computational cost. Network morphism, which keeps the functionality of a neural network while changing its neural architecture, could be helpful for NAS by enabling more efficient training during the search. In this paper, we propose a novel framework enabling Bayesian optimization to guide the network morphism for efficient neural architecture search. The framework develops a neural network kernel and a tree-structured acquisition function optimization algorithm to efficiently explores the search space. Intensive experiments on real-world benchmark datasets have been done to demonstrate the superior performance of the developed framework over the state-of-the-art methods. Moreover, we build an open-source AutoML system based on our method, namely Auto-Keras. The system runs in parallel on CPU and GPU, with an adaptive search strategy for different GPU memory limits.
Year
DOI
Venue
2019
10.1145/3292500.3330648
knowledge discovery and data mining
Keywords
Field
DocType
automated machine learning, automl, bayesian optimization, network morphis, neural architecture search
Kernel (linear algebra),Architecture,Search algorithm,Bayesian optimization,Function optimization,Artificial intelligence,Artificial neural network,Mathematics,Morphism,Deep neural networks,Machine learning
Conference
ISBN
Citations 
PageRank 
978-1-4503-6201-6
23
0.84
References 
Authors
0
3
Name
Order
Citations
PageRank
Haifeng Jin1365.87
Qingquan Song2907.77
Xia Hu32411110.07