arXiv · 2103.03875
Genetic Algorithm based hyper-parameters optimization for transfer Convolutional Neural Network
Abstract
Hyperparameter optimization is a challenging problem in developing deep neural networks. Decision of transfer layers and trainable layers is a major task for design of the transfer convolutional neural networks (CNN). Conventional transfer CNN models are usually manually designed based on intuition. In this paper, a genetic algorithm is applied to select trainable layers of the transfer model. The filter criterion is constructed by accuracy and the counts of the trainable layers. The results show that the method is competent in this task. The system will converge with a precision of 97% in the classification of Cats and Dogs datasets, in no more than 15 generations. Moreover, backward inference according the results of the genetic algorithm shows that our method can capture the gradient features in network layers, which plays a part on understanding of the transfer AI models.
Explore related subjects
Keep this discovery
Chen Li, JinZhe Jiang, YaQian Zhao, RenGang Li, EnDong Wang, Xin Zhang, Kun Zhao. 2021-02-26. Genetic Algorithm based hyper-parameters optimization for transfer Convolutional Neural Network. https://arxiv.org/abs/2103.03875
Cite the original work for its findings. Save a collection to share your selection of sources.