arXiv reaDer
Convolution Neural Network Hyperparameter Optimization Using Simplified Swarm Optimization
畳み込みニューラルネットワーク(CNN)は、画像認識で広く使用されています。 LeNet、AlexNet、VGG、ResNet、GoogLeNetなどの多数のCNNモデルが、CNNのパフォーマンスを向上させるために、レイヤーの数を増やすことによって提案されています。ただし、特定のレイヤー数を超えるとパフォーマンスが低下します。したがって、ハイパーパラメータの最適化は、CNNを改善するためのより効率的な方法です。この概念を検証するために、簡略化されたスウォーム最適化に基づく新しいアルゴリズムを提案して、最も単純なCNNモデルであるLeNetのハイパーパラメーターを最適化します。 MNIST、Fashion MNIST、およびCifar10データセットで実施された実験の結果は、提案されたアルゴリズムの精度が元のLeNetモデルおよびPSO-LeNetよりも高く、次のようなより複雑なモデルに拡張される可能性が高いことを示しました。 AlexNetとして。
Convolutional neural networks (CNNs) are widely used in image recognition. Numerous CNN models, such as LeNet, AlexNet, VGG, ResNet, and GoogLeNet, have been proposed by increasing the number of layers, to improve the performance of CNNs. However, performance deteriorates beyond a certain number of layers. Hence, hyperparameter optimisation is a more efficient way to improve CNNs. To validate this concept, a new algorithm based on simplified swarm optimisation is proposed to optimise the hyperparameters of the simplest CNN model, which is LeNet. The results of experiments conducted on the MNIST, Fashion MNIST, and Cifar10 datasets showed that the accuracy of the proposed algorithm is higher than the original LeNet model and PSO-LeNet and that it has a high potential to be extended to more complicated models, such as AlexNet.
updated: Mon Aug 09 2021 00:44:21 GMT+0000 (UTC)
published: Sat Mar 06 2021 00:23:27 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト