Klasifikasi citra makanan/non makanan menggunakan metode Transfer Learning dengan model Residual Network

Authors

  • Gustavo Thiodorus Universitas Brawijaya
  • Anugrah Prasetia Universitas Brawijaya
  • Luthfi Afrizal Ardhani Universitas Brawijaya
  • Novanto Yudistira Universitas Brawijaya

DOI:

https://doi.org/10.26594/teknologi.v11i2.2402

Abstract

People's activities that often upload photos of food before eating are common in popular social media, such as Facebook, Instagram, Twitter, and Pinterest. The food image data circulating on social media can be used for business purposes, such as analyzing customer behavior patterns. However, not all of the uploaded images are food images, so that before the analysis is carried out, it is necessary to do a classification task between food images and non-food images. Therefore, the researcher proposes automatic food image classification using transfer learning method using the Residual Network model version of 18 (ResNet-18). Residual Network model is used because it has a residual connection mechanism to solve the vanishing gradient problem. In addition, transfer learning was chosen because this method leverages the features and weights that have been generated in the previous training process on large and more general data (Imagenet) and thus reduce computation time and increase accuracy. The test was carried out by comparing the capabilities of the ResNet18 model with AlexNet. In addition, the fine tuning and freeze layer methods used to improve the quality of the model were also carried out in this study. In the experiment, the data set was divided into 3,000 images for training data and 1,000 images for test data, while the evaluation used was correctness accuracy. The results obtained in the ResNet18 model, namely the fine tuning training method, produced an accuracy value of 0.981 while the freeze layer resulted in the best accuracy value of 0.988. The AlexNet model that uses the fine tuning training method produces an accuracy value of 0.970 while the freeze layer produces the best accuracy value of 0.978. It can be concluded that the mechanism with the best accuracy is found in the RestNet18 architecture using the freeze layer 1-3 with an accuracy of 0.988.

Author Biographies

Gustavo Thiodorus, Universitas Brawijaya

Teknik Informatika

Anugrah Prasetia, Universitas Brawijaya

Teknik Informatika

Luthfi Afrizal Ardhani, Universitas Brawijaya

Teknik Informatika

Novanto Yudistira, Universitas Brawijaya

Teknik Informatika

References

Atanasova, A. (2016, Nov. 9). The Psychology of Foodstagramming. Retrieved from Social Media Today: https://www.socialmediatoday.com/social-networks/psychology-foodstagramming

Bestari, K. (2014). Fenomena Mengunggah Foto Makanan pada Pengguna Media Sosial. Depok: Universitas Indonesia.

Gao, H. (2017, Aug. 8). A Walk-through of AlexNet. Retrieved from Hao Gao: https://medium.com/@smallfishbigsea/a-walk-through-of-alexnet-6cbd137a5637

Gultom, Y., Arymurthy, A. M., & Masikome, R. J. (2018). Batik Classification using Deep Convolutional Network Transfer Learning. Jurnal Ilmu Komputer dan Informasi (Journal of Computer Science and Information), 11(2), 59-66.

He, K., Zhang, X., Ren, S., & Sun, J. (2016). Deep Residual Learning for Image Recognition. IEEE Conference on Computer Vision and Pattern Recognition (CVPR) (pp. 770-778). Las Vegas, NV, USA: IEEE.

Islam, K. T., Wijewickrema, S., Pervez, M., & O'Leary, S. (2018). An Exploration of Deep Transfer Learning for Food Image Classification. Digital Image Computing: Techniques and Applications (DICTA). Canberra, ACT, Australia: IEEE.

Islam, M. T., Siddique, B. M., Rahman, S., & Jabid, T. (2018). Food Image Classification with Convolutional Neural Network. International Conference on Intelligent Informatics and Biomedical Sciences (ICIIBMS) (pp. 257-262). Bangkok, Thailand: IEEE.

Janocha, K., & Czarnecki, W. M. (2016). On Loss Functions for Deep Neural Networks in Classification. Retrieved from http://arxiv.org/abs/1702.05659

Krizhevsky, A., Sutskever, I., & Hinton, G. E. (2017). ImageNet Classification with Deep Convolutional Neural Networks. Communications of the ACM, 60(6), 84–90.

Rohim, A., Sari, Y. A., & Tibyani, T. (2019). Convolution Neural Network (CNN) Untuk Pengklasifikasian Citra Makanan Tradisional. JPTIIK (Jurnal Pengembangan Teknologi Informasi dan Ilmu Komputer), 3(7), 7037-7042.

Ruder, S. (2016). An overview of gradient descent optimization algorithms. Retrieved from https://arxiv.org/abs/1609.04747

Singla, A., Yuan, L., & Ebrahimi, T. (2016). Food/Non-food Image Classification and Food Categorization using Pre-Trained GoogLeNet Model. MADiMa '16: Proceedings of the 2nd International Workshop on Multimedia Assisted Dietary Management (pp. 3-11). ACM.

Swastika, W. (2020). Studi Awal Deteksi COVID-19 Menggunakan Citra CT Berbasis Deep Learning. Jurnal Teknologi Informasi dan Ilmu Komputer, 7(3), 629-634.

Yudistira, N., Widodo, A. W., & Rahayudi, B. (2020). Deteksi COVID-19 pada Citra Sinar-X Dada menggunakan Deep Learning yang Efisien. Jurnal Teknologi Informasi dan Ilmu Komputer, 7(6), 1289-1296.

Downloads

Published

2021-06-09

Issue

Section

Articles