Hyperparameter Optimization in a Convolutional Neural Network Using Metaheuristic Algorithms

The hyperparameters of a convolutional neural network always have been important, because the performance of the convolutional neural network depends largely on them. To find manually the optimal value requires a lot of work, experience, and time. Metaheuristic algorithms are approximate general-purpose search and optimization algorithms, they are used to find a good solution into search space. In the book chapter, the hyperparameters of a convolutional neural network are optimized. A metaheuristic algorithm is used to finds the best solution, that is, the best hyperparameter value for higher accuracy of the convolutional neural network (CNN). The results of four metaheuristic algorithms that have the same objective are compared, which are PSO (Particle Swarm Optimization), ABC (Artificial Bee Colony), ALO (Ant Lion Optimization), and BA (Bat Algorithm). Each algorithm uses the same convolutional neural network architecture as well as the same dataset (MNIST database) also each algorithm is run 30 times to perform the corresponding statistics. Finally, the results of the algorithms are shown.

[1]  Seyed Mohammad Mirjalili,et al.  The Ant Lion Optimizer , 2015, Adv. Eng. Softw..

[2]  Harish Garg Solving structural engineering design optimization problems using an artificial bee colony algorithm , 2013 .

[3]  Erik Cuevas,et al.  Learning classical and metaheuristic optimization techniques by using an educational platform based on LEGO robots , 2019, The International Journal of Electrical Engineering & Education.

[4]  Gonzalo Pajares,et al.  Parameter identification of solar cells using artificial bee colony optimization , 2014 .

[5]  Seung-Min Park,et al.  Optimal hyperparameter tuning of convolutional neural networks based on the parameter-setting-free harmony search algorithm , 2018, Optik.

[6]  Wilfrido Gómez-Flores,et al.  Automatic adjustment of the pulse-coupled neural network hyperparameters based on differential evolution and cluster validity index for image segmentation , 2020, Appl. Soft Comput..

[7]  Yoshua. Bengio,et al.  Learning Deep Architectures for AI , 2007, Found. Trends Mach. Learn..

[8]  Gonzalo Pajares,et al.  Cross entropy based thresholding for magnetic resonance brain images using Crow Search Algorithm , 2017, Expert Syst. Appl..

[9]  Yulong Wang,et al.  cPSO-CNN: An efficient PSO-based algorithm for fine-tuning hyper-parameters of convolutional neural networks , 2019, Swarm Evol. Comput..

[10]  Aboul Ella Hassanien,et al.  Parameter identification of two dimensional digital filters using electro-magnetism optimization , 2018, Multimedia Tools and Applications.

[11]  Jie Bai,et al.  A new hyperparameters optimization method for convolutional neural networks , 2019, Pattern Recognit. Lett..