Journal of Beijing University of Posts and Telecommunications

  • EI核心期刊

Journal of Beijing University of Posts and Telecommunications ›› 2021, Vol. 44 ›› Issue (6): 83-88,95.doi: 10.13190/j.jbupt.2021-068

• PAPERS • Previous Articles     Next Articles

Hyperparameter Optimization of Convolutional Neural Network Based on Multi-Strategy MRFO Algorithm

LIU Yong-li, ZHU Ya-meng, CHAO Hao   

  1. College of Computer Science and Technology, Henan Polytechnic University, Jiaozuo 454000, China
  • Received:2021-04-20 Online:2021-12-28 Published:2021-12-28

Abstract: The performance of convolutional neural network is closely related to the configuration of hyperparameters. However, the selection of optimal hyperparameters is time-consuming and labor-consuming. In order to improve the efficiency of hyperparameter selection, a multi-strategy manta ray foraging optimization algorithm is proposed. On the one hand, half uniform initialization strategy is adopted to improve population diversity. On the other hand, it combines new weight factor update strategy and splitting strategy to improve the convergence speed and fitting accuracy respectively. According to the real coding strategy, this algorithm is applied to the research of convolutional neural network hyperparameter optimization, which can be iterated according to three foraging methods to obtain the optimal hyperparameter configuration. In order to evaluate the effectiveness of hyperparameter optimization, the proposed algorithm is compared with the mainstream convolutional neural network hyperparameter optimization algorithms on mixed national institute of standards and technology and CIFAR-10 datasets. Experimental results show that the proposed algorithm achieves higher accuracy with less resources.

Key words: convolutional neural network, manta ray foraging optimization, hyperparameter optimization

CLC Number: