Transfer learning with multiple pre-trained network for fundus classification

Research output: Contribution to journalArticlepeer-review

7 Citations (Scopus)

Abstract

Transfer learning (TL) is a technique of reuse and modify a pre-trained network. It reuses feature extraction layer at a pre-trained network. A target domain in TL obtains the features knowledge from the source domain. TL modified classification layer at a pre-trained network. The target domain can do new tasks according to a purpose. In this article, the target domain is fundus image classification includes normal and neovascularization. Data consist of 100 patches. The comparison of training and validation data was 70:30. The selection of training and validation data is done randomly. Steps of TL i. e load pre-trained networks, replace final layers, train the network, and assess network accuracy. First, the pre-trained network is a layer configuration of the convolutional neural network architecture. Pre-trained network used are AlexNet, VGG16, VGG19, ResNet50, ResNet101, GoogLeNet, Inception-V3, InceptionResNetV2, and squeezenet. Second, replace the final layer is to replace the last three layers. They are fully connected layer, softmax, and output layer. The layer is replaced with a fully connected layer that classifies according to number of classes. Furthermore, it's followed by a softmax and output layer that matches with the target domain. Third, we trained the network. Networks were trained to produce optimal accuracy. In this section, we use gradient descent algorithm optimization. Fourth, assess network accuracy. The experiment results show a testing accuracy between 80% and 100%.

Original languageEnglish
Pages (from-to)1382-1388
Number of pages7
JournalTelkomnika (Telecommunication Computing Electronics and Control)
Volume18
Issue number3
DOIs
Publication statusPublished - 2020

Keywords

  • Classification
  • Convolutional neural network
  • Multiple pre-trained network
  • Neovascularization
  • Transfer learning

Fingerprint

Dive into the research topics of 'Transfer learning with multiple pre-trained network for fundus classification'. Together they form a unique fingerprint.

Cite this