Image Classification with Transfer Learning and FastAI

Ujwal Gullapalli, Lei Chen, Jinbo Xiong

Research output: Contribution to book or proceedingConference articlepeer-review

Abstract

Today deep learning has provided us with endless possibilities for solving problems in many domains. Diagnosing diseases, speech recognition, image classification, and targeted advertising are a few of its applications. Starting this process from scratch requires using large amounts of labeled data and significant cloud processing usage. Transfer learning is a deep learning technique that solves this problem by making use of a model that is pre-trained for a certain task and using it on a different task of a related problem. Therefore, the goal of the project is to utilize transfer learning and achieve near-perfect results using a limited amount of data and computation power. To demonstrate, an image classifier using FastAI that detects three types of birds with up to 94% accuracy is implemented. This approach can be applied to solve tasks that are limited by labeled data and would gain by knowledge learned from a related task.

Original languageEnglish
Title of host publicationMobile Multimedia Communications - 14th EAI International Conference, Mobimedia 2021, Proceedings
EditorsJinbo Xiong, Shaoen Wu, Changgen Peng, Youliang Tian
PublisherSpringer Science and Business Media Deutschland GmbH
Pages796-806
Number of pages11
ISBN (Print)9783030898137
DOIs
StatePublished - 2021
Event14th EAI International Conference on Mobile Multimedia Communications, Mobimedia 2021 - Virtual, Online
Duration: Jul 23 2021Jul 25 2021

Publication series

NameLecture Notes of the Institute for Computer Sciences, Social-Informatics and Telecommunications Engineering, LNICST
Volume394 LNICST
ISSN (Print)1867-8211
ISSN (Electronic)1867-822X

Conference

Conference14th EAI International Conference on Mobile Multimedia Communications, Mobimedia 2021
CityVirtual, Online
Period07/23/2107/25/21

Scopus Subject Areas

  • Computer Networks and Communications

Keywords

  • Deep learning
  • Image classification
  • Transfer learning

Fingerprint

Dive into the research topics of 'Image Classification with Transfer Learning and FastAI'. Together they form a unique fingerprint.

Cite this