Marta BISTROŃ, Michał PACZKOWSKI, Igor PAWLAK, Piotr ŚWIDZIŃSKI, Monika CIERPICKA and Zbigniew PIOTROWSKI
Military University of Technology, gen. Sylwestra Kaliskiego 2, 00-902 Warsaw, Poland
Volume 2022 (30),
Article ID 4043122,
Methods and Applications in Artificial Intelligence and Machine Learning: 40AI 2022
Abstract
This paper examines the impact of the size and quality of the learning dataset on the training efficiency of a convolutional network model. The transfer learning technique is characterized, and the results obtained for models trained from scratch for 40 and 100 epochs are compared with those obtained by transfer models built on 4 popular convolutional neural network architectures: AlexNet, VGG-16, ResNet-50, SqueezeNet. The impact of the quality of the dataset on the evaluation results of the selected model was also analyzed. To this end, neural network training using transfer learning was performed on 3 datasets: a baseline dataset, a dataset reduced by incorrect examples and a dataset reduced and then extended by additional correct examples.
Keywords: CNN, digital image classification, data selection, transfer learning