Logo image
Simplified swarm optimization for hyperparameters of convolutional neural networks
期刊文章   同儕審查

Simplified swarm optimization for hyperparameters of convolutional neural networks

Wei-Chang Yeh, Yi-Ping Lin, Yun-Chia Liang, Chyh-Ming LaiChia-Ling Huang
Computers and Industrial Engineering, 卷.177, 109076
03/2023

摘要

Convolutional neural networks Hyperparameter optimization Image recognition Machine learning Simplified swarm optimization Computer Science (all) Engineering (all)
Convolutional neural networks (CNNs) are widely used in image recognition. Numerous CNN models, such as LeNet, AlexNet, VGG, ResNet, and GoogLeNet, have been developed by increasing the number of layers to improve performance. However, performance deteriorates beyond a certain number of layers. Hence, hyperparameter optimization is a more efficient way to improve CNNs. To validate this concept, in the present study, an algorithm based on simplified swarm optimization was developed for optimizing the hyperparameters of the simplest CNN model: LeNet. The results of experiments involving the MNIST, Fashion-MNIST, and CIFAR-10 datasets indicated that the accuracy of the proposed algorithm was higher than those of LeNet and PSO-LeNet and that the proposed algorithm can be applied to more complex models such as AlexNet.

相關連結

指標

1 檢視次數

詳細資料

Logo image