TY - GEN
T1 - Application of Transfer Learning for Image Classification on Dataset with Not Mutually Exclusive Classes
AU - Fan, Jiayi
AU - Lee, Jang Hyeon
AU - Lee, Yong Keun
N1 - Publisher Copyright:
© 2021 IEEE.
PY - 2021/6/27
Y1 - 2021/6/27
N2 - Machine learning technologies, especially deep convolution neural network (CNN), play an important role in image classification tasks. However, performing image classification tasks using state-of-the-art deep learning models might suffer from the lack of available images for network training and requirement of computationally powerful machines to conduct the training. In order to classify new classes, in this paper, transfer learning models are built based on the pretrained AlexNet and the VGG16 to overcome the drawbacks of the deep CNN The models are used on a not well-classified image dataset, where classes of the images are not mutually exclusive, and an image could belong to more than one classes. Experimental results are given to evaluate the performance of the transfer learning approach on this not exclusive dataset, and the conventional CNN are used as the benchmark. It shows that the transfer learning models outperform the conventional CNN by a large margin in both coupled and decoupled datasets.
AB - Machine learning technologies, especially deep convolution neural network (CNN), play an important role in image classification tasks. However, performing image classification tasks using state-of-the-art deep learning models might suffer from the lack of available images for network training and requirement of computationally powerful machines to conduct the training. In order to classify new classes, in this paper, transfer learning models are built based on the pretrained AlexNet and the VGG16 to overcome the drawbacks of the deep CNN The models are used on a not well-classified image dataset, where classes of the images are not mutually exclusive, and an image could belong to more than one classes. Experimental results are given to evaluate the performance of the transfer learning approach on this not exclusive dataset, and the conventional CNN are used as the benchmark. It shows that the transfer learning models outperform the conventional CNN by a large margin in both coupled and decoupled datasets.
KW - Convolutional neural network
KW - deep learning
KW - transfer learning
UR - https://www.scopus.com/pages/publications/85113970241
U2 - 10.1109/ITC-CSCC52171.2021.9501424
DO - 10.1109/ITC-CSCC52171.2021.9501424
M3 - Conference contribution
AN - SCOPUS:85113970241
T3 - 2021 36th International Technical Conference on Circuits/Systems, Computers and Communications, ITC-CSCC 2021
BT - 2021 36th International Technical Conference on Circuits/Systems, Computers and Communications, ITC-CSCC 2021
PB - Institute of Electrical and Electronics Engineers Inc.
T2 - 36th International Technical Conference on Circuits/Systems, Computers and Communications, ITC-CSCC 2021
Y2 - 27 June 2021 through 30 June 2021
ER -