Naslov
Image Classification by Optimized Convolution Neural Networks

Autori
Tuba, Eva ; Tuba, Ira ; Capor Hrošik, Romana ; Alihodzic, Adis ; Tuba, Milan

Vrsta, podvrsta i kategorija rada
Poglavlja u knjigama, znanstveni

Knjiga
Lecture Notes in Networks and Systems

Urednik/ci
Rathore, V.S. ; Sharma, S.C. ; Tavares, J.M.R.S. ; Moreira, C. ; Surendiran, B.

Izdavač
Springer

Grad
Berlin

Godina
2022

Raspon stranica
447-454

ISBN
978-981-19-1121-7

Ključne riječi
convolutional neural networks ; hyperparameter tuning ; image classification ; metaheuristic ; optimization ; swarm intelligence

Sažetak
Considering the fact that digital images are used in almost all scientific areas and they are a big part of everyday life, it is obvious that the importance of good methods for processing and analyzing them is great. One of the most frequent tasks in various applications that use digital images is image classification. A revolutionized improvement in this area was achieved with convolutional neural networks (CNN). The convolutional neural networks managed to achieve classification accuracy significantly better compared to previously proposed and used methods. Even better results can be obtained by tuning CNN hyperparameters. Since this is a hard optimization problem, swarm intelligence algorithms can be successfully used. In this paper, we propose bare bones fireworks algorithm for tuning a selected subset of hyperparameters and it was tested on the benchmark dataset for handwritten digit recognition, MNIST. The proposed method achieved higher classification accuracy compared to the methods from the literature.

Izvorni jezik
Engleski

Znanstvena područja
Matematika, Računarstvo

POVEZANOST RADA