A Study on CNN Transfer Learning for Image Classification

Authors – Mahbub Hussain, Jordan J. Bird, and Diego R. Faria
Aston Lab for Intelligent Collectives Engineering (ALICE)
School of Engineering and Applied Science, Aston University, UK.

Abstract – Many image classification models have been introduced to help tackle the foremost issue of recognition accuracy. Image classification is one of the core problems in Computer Vision field with a large variety of practical applications. Examples include object recognition for robotic manipulation, pedestrian or obstacle detection for autonomous vehicles, among others. A lot of attention has been associated with Machine Learning, specifically neural networks such as the Convolutional Neural Network (CNN) winning image classification competitions. This work proposes the study and investigation of such a CNN architecture model (i.e. Inception-v3) to establish whether it would work best in terms of accuracy and efficiency with new image datasets via Transfer Learning. The retrained model is evaluated, and the results are compared to some state-of-the-art approaches.

View on ResearchGate Download PDF