Transfer Learning Methods as a New Approach in Computer Vision Tasks with Small Datasets
Andrzej Brodzicki, Michał Piekarski, Dariusz Kucharski, Joanna Jaworek-Korjakowska, M. Gorgoń
Abstract
Abstract Deep learning methods, used in machine vision challenges, often face the problem of the amount and quality of data. To address this issue, we investigate the transfer learning method. In this study, we briefly describe the idea and introduce two main strategies of transfer learning. We also present the widely-used neural network models, that in recent years performed best in ImageNet classification challenges. Furthermore, we shortly describe three different experiments from computer vision field, that confirm the developed algorithms ability to classify images with overall accuracy 87.2-95%. Achieved numbers are state-of-the-art results in melanoma thickness prediction, anomaly detection and Clostridium di cile cytotoxicity classification problems.