Neural networks designing neural networks: Multi-objective hyper-parameter optimization

By Sean C. Smithson, Guang Yang, Warren J. Gross, B. Meyer. 2016

104 citations recorded by the source corpus.

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.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Diana Marculescu
  2. Andrés Camero
  3. Enrique Alba
  4. Dimitrios Stamoulis
  5. Tejaswini Pedapati
  6. Jamal Toutouh
  7. Ting-Wu Chin
  8. Ambrish Rawat

In Computer Vision and Pattern Recognition

  1. Diana Marculescu
  2. Tejaswini Pedapati
  3. Dimitrios Stamoulis
  4. Andrés Camero
  5. Enrique Alba
  6. Jamal Toutouh
  7. Ting-Wu Chin
  8. Ambrish Rawat

Citations in the corpus