In plain words: A neural network predicts how accurate and costly candidate designs will be, so only promising ones get trained while hunting the best accuracy-versus-complexity trade-off. On two image datasets it found near-ideal trade-offs while testing only a small slice of the design space, rather than hand-tuning.
Abstract · Neural Networks Designing Neural Networks: Multi-Objective Hyper-Parameter Optimization
Artificial neural networks have gone through a recent rise in popularity, achieving state-of-the-art results in various fields, including image classification, speech recognition, and automated control. Both the performance and computational complexity of such models are heavily dependant on the design of characteristic hyper-parameters (e.g., number of hidden layers, nodes per layer, or choice of activation functions), which have traditionally been optimized manually. With machine learning penetrating low-power mobile and embedded areas, the need to optimize not only for performance (accuracy), but also for implementation complexity, becomes paramount. In this work, we present a multi-objective design space exploration method that reduces the number of solution networks trained and evaluated through response surface modelling. Given spaces which can easily exceed 1020 solutions, manually designing a near-optimal architecture is unlikely as opportunities to reduce network complexity, while maintaining performance, may be overlooked. This problem is exacerbated by the fact that hyper-parameters which perform well on specific datasets may yield sub-par results on others, and must therefore be designed on a per-application basis. In our work, machine learning is leveraged by training an artificial neural network to predict the performance of future candidate networks. The method is evaluated on the MNIST and CIFAR-10 image datasets, optimizing for both recognition accuracy and computational complexity. Experimental results demonstrate that the proposed method can closely approximate the Pareto-optimal front, while only exploring a small fraction of the design space.
Sean C. Smithson, Guang Yang, Warren J. Gross, Brett H. Meyer
arXiv:1611.02120 · cs.NE, cs.LG · submitted Nov 7, 2016
abstract · pdf · html · To appear in ICCAD'16. The authoritative version will appear in the ACM Digital Library