Hyper-parameter selection in deep neural networks using parallel particle swarm optimization

Pablo Ribalta Lorenzo,Jakub Nalepa,Luciano Sanchez Ramos,José Ranilla Pastor
DOI: https://doi.org/10.1145/3067695.3084211
2017-07-15
Abstract:The need of manual hyper-parameter selection can seriously hamper the model optimization of Deep Neural Networks (DNNs). Conventional automated approaches tackling this problem suffer from poor scalability or fail in certain scenarios. In this paper, we introduce a parallel method that applies Particle Swarm Optimization (PSO) for the hyper-parameter selection in DNNs. To estimate the best hyper-parameters, a population of particles is evolved, with their fitness calculated in parallel. The experimental results demonstrate very desirable scalability properties for different DNNs. We show that the parallel PSO can further optimize existent models designed by experts in an affordable amount of time.
What problem does this paper attempt to address?