Classification of small data sets of images with transfer learning in convolutional neural networks

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

[thumbnail of Trud BJIAM, avgust 2018.pdf]
Trud BJIAM, avgust 2018.pdf

Download (3MB) | Preview


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

Actions (login required)

View Item View Item