Application of Transfer Learning for Image Classification on Dataset with Not Mutually Exclusive Classes

Jiayi Fan, Jang Hyeon Lee, Yong Keun Lee

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

4 Scopus citations

Abstract

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.

Original languageEnglish
Title of host publication2021 36th International Technical Conference on Circuits/Systems, Computers and Communications, ITC-CSCC 2021
PublisherInstitute of Electrical and Electronics Engineers Inc.
ISBN (Electronic)9781665435536
DOIs
StatePublished - 27 Jun 2021
Event36th International Technical Conference on Circuits/Systems, Computers and Communications, ITC-CSCC 2021 - Jeju, Korea, Republic of
Duration: 27 Jun 202130 Jun 2021

Publication series

Name2021 36th International Technical Conference on Circuits/Systems, Computers and Communications, ITC-CSCC 2021

Conference

Conference36th International Technical Conference on Circuits/Systems, Computers and Communications, ITC-CSCC 2021
Country/TerritoryKorea, Republic of
CityJeju
Period27/06/2130/06/21

Keywords

  • Convolutional neural network
  • deep learning
  • transfer learning

Fingerprint

Dive into the research topics of 'Application of Transfer Learning for Image Classification on Dataset with Not Mutually Exclusive Classes'. Together they form a unique fingerprint.

Cite this