A modification of convolutional neural network layer to increase images classification accuracy

Wahyudi Agustiono, Mohammad Imam Utoyo, Riries Rulaningtyas, Budi Dwi Satoto

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

3 Citations (Scopus)

Abstract

Image classification is one of the fundamental steps in digital image processing. Research in this area has received considerable attention, with photos shared on social media which are sometimes similar but have different identities. There are various classification methods proposed in the literature to improve accuracy. One important strategy is Convolutional Neural Networks (CNN). Although CNN is superior in pattern recognition, it has limitations inaccuracy. It requires additional training time, especially when dealing with variants in data generated from a large number of images but of similar properties. Therefore, this study aims to overcome this problem by proposing a modification of the CNN layer to increase the accuracy of the multi-class image classification. This research used four different flower species with similar patterns added from a public database. Each category consists of 400 colour images with different angles, backgrounds, and lighting conditions that provide different variations to the training process. Through experiments using 1, 600 of the four flower species, this study shows that the 18-34 layer modification produces the most optimal accuracy in the training process ranging from 99.3% with misclassification of MSE =0.0025, RMSE = 0.1606, and MAE = 0. 0133. Meanwhile, the computation time required to compile the data set is 3 minutes, 18 seconds. This result is 50% faster when compared to computation time using existing architecture such as Alexnet model with a similar number of layers.

Original languageEnglish
Title of host publicationProceeding - 6th Information Technology International Seminar, ITIS 2020
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages274-279
Number of pages6
ISBN (Electronic)9781728177267
DOIs
Publication statusPublished - 14 Oct 2020
Event6th Information Technology International Seminar, ITIS 2020 - Virtual, Surabaya, Indonesia
Duration: 14 Oct 202016 Oct 2020

Publication series

NameProceeding - 6th Information Technology International Seminar, ITIS 2020

Conference

Conference6th Information Technology International Seminar, ITIS 2020
Country/TerritoryIndonesia
CityVirtual, Surabaya
Period14/10/2016/10/20

Keywords

  • Convolutional neural nenvork
  • Data augmentation
  • Flower classification
  • Multi-class image

Fingerprint

Dive into the research topics of 'A modification of convolutional neural network layer to increase images classification accuracy'. Together they form a unique fingerprint.

Cite this