iSEC: An Optimized Deep Learning Model for Image Classification on Edge Computing
Endah Kristiani, Chao‐Tung Yang, Chin‐Yin Huang
Abstract
Optimization strategies in deep learning models require different techniques for different use cases. Besides, various phases of the model deployment life-cycle specify possible and particular optimization strategies. In this paper, an optimized deep learning model on the edge computing environment is proposed for image classification cases. For preparing the dataset, the image preprocessing and data augmentation methods are utilized to prepare the data for the training process. To accelerate the deep learning training process, this system implemented CPU optimization and hyperparameter tuning. Tensorflow is applied as a framework for the training model. InceptionV3, VGG16, and MobileNet are applied as topology implemented in the deep learning training comparison. In this case, InceptionV3 was used for modeling the deep learning applications on edge. To optimize the trained model, a Model Optimizer is used on the edge device. It can be seen in the experiments, MobileNet was the least accurate model (85%) and the longest time to load the model (71s). VGG16 was the most reliable (91%) and the shortest time to load the model (50s). InceptionV3 has median accuracy (87%) and the average time to load the model (52s).