Petrovska, Biserka and Stojanovic, Igor and Atanasova-Pacemska, Tatjana (2018) Classification of small data sets of images with transfer learning in convolutional neural networks. Balkan Journal of Applied Mathematics and Informatics, 1 (1). pp. 17-24. ISSN 2545-4803
Preview |
Text
Trud BJIAM, avgust 2018.pdf Download (3MB) | Preview |
Abstract
Nowadays the rise of the artificial intelligence is with high speed. Even we are far away from the moment when machines are going to make decisions instead of human beings, the development in the field of neural networks is remarkable. They are in a big expansion in a new millennium. Their application is wide: they are used in processing images, video, speech, audio and text. One special kind of neural networks, convolutional neural networks have been a point of interest of researches in the last decade. These networks have been widely applied to a variety of pattern recognition problems, such as computer vision. Convolutional neural networks were trained on millions of images and it is difficult to outperform the accuracies that have been achieved. Sometimes we have a small set of images to be classified and in those situations there is no success to train the network from a scratch. This article exploits the technique of transfer learning for classifying the images of small datasets. It transfers the knowledge of the pre-trained convolutional neural network and use it for the classification of those data sets. Fine-tuning of the network is done through optimization of hyper parameters, in order to maximize the classification accuracy. At the end, the directions have been proposed for the selection of the hyper parameters and of the pre-existing network which can be suitable for transfer learning.
Item Type: | Article |
---|---|
Subjects: | Natural sciences > Computer and information sciences Natural sciences > Matematics |
Divisions: | Faculty of Computer Science |
Depositing User: | Tatjana A. Pacemska |
Date Deposited: | 01 Nov 2018 12:22 |
Last Modified: | 01 Nov 2018 12:22 |
URI: | https://eprints.ugd.edu.mk/id/eprint/20641 |
Actions (login required)
![]() |
View Item |