Indonesian Journal of Electrical Engineering and Computer Science
Vol 33, No 3: March 2024

Image classification-based transfer learning framework for image detection of IoT devices

Yuris Mulya Saputra (Universitas Gadjah Mada)
Hanung Addi Chandra Utomo (Universitas Gadjah Mada)
Ganjar Alfian (Universitas Gadjah Mada)



Article Info

Publish Date
01 Mar 2024

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.

Copyrights © 2024