Hyperparameter Optimization Using Hyperband in Convolutional Neural Network for Image Classification of Indonesian Snacks

Abstract

Abstract. Indonesia is known for its traditional food both domestically and abroad. Several cakes are included in favorite traditional foods. Of the many types of cakes that exist, it is visually easy to recognize by humans, but computer vision requires special techniques in identifying image objects to types of cakes. Therefore, to recognize objects in the form of images of cakes as one of Indonesian specialties, a deep learning algorithm technique, namely the Convolutional Neural Network (CNN) can be used. Purpose: This study aims to find out how the Convolutional Neural Network (CNN) works by optimizing the hyperband hyperparameter in the classification process and knowing the accuracy value when hyperband is applied to the optimal hyperparameter selection process for classifying Indonesian snack images. Methods/Study design/approach: This study optimizes the hyperparameter Convolutional Neural Network (CNN) using Hyperband on the Indonesian cake dataset. The dataset is 1845 images of Indonesian snacks which consists of 1523 training data, 162 validation data and 160 testing data with 8 classes. In training data, the dataset is divided by 82% on training data, 9% validation, and 9% testing. Result/Findings: The best hyperparameter value produced is 480 for the number of dense neurons 2 and 0.0001 for the learning rate. The proposed method succeeded in achieving a training value of 87.53%, for the validation process it was obtained 66.8%, the testing process was obtained 79.37%. Results obtained from model training of 50 epochs. Novelty/Originality/Value: Previous research focused on the application and development of algorithms for the classification of Indonesian snacks. Therefore, optimizing hyperparameters in a Convolutional Neural Network (CNN) using Hyperband can be an alternative in selecting the optimal architecture and hyperparameters.