Investigation of transfer learning for image classification and impact on training sample size
Document Type
Article
Publication Date
4-15-2021
Abstract
Recent developments in deep learning have brought huge breakthroughs in the image processing area, which triggered numerous successful applications and positively impacted the current big data context of Industry 4.0. On the other hand, it is widely known that large amounts of training data are required to train a deep learning model with millions of parameters, which limits its application in many industrial applications where sufficient data resources are lacking. Transfer learning is one of the practical solutions to reduce the data required for training, which tries to reuse learned knowledge for similar tasks. Nevertheless, many technical details of transfer learning implementation are not well documented. Therefore, in this work, two datasets collected from plastics manufacturing processes were studied to investigate different transfer learning approaches and implementation details for high-performance model building under the constraint of limited available training data. Different transfer learning implementations are compared and important technical details are also discussed in this study. Through this study, the minimum number of training samples can be estimated. Transfer learning is compared with the newly developed few-shot learning approach as a brief comparative study. Finally, this work summarizes practical guidelines for the development of image classification models with limited data resources.
Publication Source (Journal or Book title)
Chemometrics and Intelligent Laboratory Systems
Recommended Citation
Zhu, W., Braun, B., Chiang, L., & Romagnoli, J. (2021). Investigation of transfer learning for image classification and impact on training sample size. Chemometrics and Intelligent Laboratory Systems, 211 https://doi.org/10.1016/j.chemolab.2021.104269