Path: Top -> Journal -> Telkomnika -> 2020 -> Vol 18, No 3, June

Transfer learning with multiple pre-trained network for fundus classification

Journal from gdlhub / 2021-01-20 15:23:51
By : Wahyudi Setiawan, Moh. Imam Utoyo, Riries Rulaningtyas, Telkomnika
Created : 2021-01-12, with 1 files

Keyword : classification, convolutional neural network, multiple pre-trained network, neovascularization, transfer learning
Url : http://journal.uad.ac.id/index.php/TELKOMNIKA/article/view/14868
Document Source : Web

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%.

Give Comment ?#(0) | Bookmark

PropertyValue
Publisher IDgdlhub
OrganizationTelkomnika
Contact NameHerti Yani, S.Kom
AddressJln. Jenderal Sudirman
CityJambi
RegionJambi
CountryIndonesia
Phone0741-35095
Fax0741-35093
Administrator E-mailelibrarystikom@gmail.com
CKO E-mailelibrarystikom@gmail.com

Print ...

Contributor...

  • , Editor: Calvin

Downnload...