Hanung Addi Chandra Utomo
Universitas Gadjah Mada

Published : 1 Documents Claim Missing Document
Claim Missing Document
Check
Articles

Found 1 Documents
Search

Image classification-based transfer learning framework for image detection of IoT devices Yuris Mulya Saputra; Hanung Addi Chandra Utomo; Ganjar Alfian
Indonesian Journal of Electrical Engineering and Computer Science Vol 33, No 3: March 2024
Publisher : Institute of Advanced Engineering and Science

Show Abstract | Download Original | Original Source | Check in Google Scholar | DOI: 10.11591/ijeecs.v33.i3.pp1989-1997

Abstract

For artificial intelligence (AI) applications, centralized learning on a cloud server and local learning on an internet-of-things device may suffer from data privacy leakage due to data sharing and inaccurate prediction due to limited computing resources. Transfer learning has been proposed as one potential solution to the world's big data problems. Transfer learning eliminates the need for each internet-of-things device to share local data with the cloud server during the training process. Instead, it can go through the training process on its own, using a cloud server's pre-trained model with high accuracy. As a result, despite its limited computing resources, the internet of things device can still predict with high accuracy. This paper proposes a transfer learning model for improving image detection accuracy on IoT devices with restricted computation. To obtain accurate image classification, a deep learning approach based on convolutional neural networks is used. The proposed method with freeze and unfreeze approaches achieves a higher validation accuracy (up to 43.6%) and a lower validation loss (up to 6.5 times) than the non-transfer learning method, according to simulation results using three relevant internet-of-things datasets.