Optimization of convolutional neural networks for image classification using genetic algorithms and bayesian optimization

• Main Author: Rawat, Waseem
• Other Authors: Wang, Zenghui
• Format: Others
• Language: en
• Published: 2018
• Subjects: Deep learning; Artificial neural networks; Convolutional neural networks; Evolutionary algorithms; Genetic algorithms; Bayesian optimization; Computer vision; Image classification; Model selection; Hyperparameter optimization; 006.32; Neural networks (Computer science); Image processing > Digital techniques
• Online Access: Rawat, Waseem (2018) Optimization of convolutional neural networks for image classification using genetic algorithms and bayesian optimization, University of South Africa, Pretoria, <http://hdl.handle.net/10500/24977>; http://hdl.handle.net/10500/24977

Notwithstanding the recent successes of deep convolutional neural networks for classification tasks, they are sensitive to the selection of their hyperparameters, which impose an exponentially large search space on modern convolutional models. Traditional hyperparameter selection methods include manual, grid, or random search, but these require expert knowledge or are computationally burdensome. Divergently, Bayesian optimization and evolutionary inspired techniques have surfaced as viable alternatives to the hyperparameter problem. Thus, an alternative hybrid approach that combines the advantages of these techniques is proposed. Specifically, the search space is partitioned into discrete-architectural, and continuous and categorical hyperparameter subspaces, which are respectively traversed by a stochastic genetic search, followed by a genetic-Bayesian search. Simulations on a prominent image classification task reveal that the proposed method results in an overall classification accuracy improvement of 0.87% over unoptimized baselines, and a greater than 97% reduction in computational costs compared to a commonly employed brute force approach. === Electrical and Mining Engineering === M. Tech. (Electrical Engineering)