In this study, particle swarm optimization (PSO) is applied for the hyperparameter optimization problem of Convolutional Neural Network (CNN) model. The design objective is to improve the prediction accuracy. Several PSO algorithms are compared in the classification problem of a CIFAR-10 dataset. The design variables of PSO are defined as continuous values. Hyperparameters of CNN model may take discrete or categorical variables, so the design variables defined as continuous values are converted into discrete or categorical variables if necessary. The results showed that SG-PSO exhibited the lowest loss value of all methods and that, in all algorithms, optimization process led to early convergence in about 15 generations.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Application of PSO for Hyperparameter Optimization of Convolutional Neural Network

  • Tetsuya Sato,
  • Kenta Shiomi,
  • Eisuke Kita

摘要

In this study, particle swarm optimization (PSO) is applied for the hyperparameter optimization problem of Convolutional Neural Network (CNN) model. The design objective is to improve the prediction accuracy. Several PSO algorithms are compared in the classification problem of a CIFAR-10 dataset. The design variables of PSO are defined as continuous values. Hyperparameters of CNN model may take discrete or categorical variables, so the design variables defined as continuous values are converted into discrete or categorical variables if necessary. The results showed that SG-PSO exhibited the lowest loss value of all methods and that, in all algorithms, optimization process led to early convergence in about 15 generations.