1. Image preprocessing and hyperparameter optimization on pretrained model MobileNetV2 in white blood cell image classification
- Author
-
Parmonangan R. Togatorop, Yohanssen Pratama, Astri Monica Sianturi, Mega Sari Pasaribu, and Pebri Sangmajadi Sinaga
- Subjects
Information Systems and Management ,Artificial Intelligence ,Control and Systems Engineering ,Image classification ,Image preprocessing ,MobileNetV2 ,Convolutional neural network ,Electrical and Electronic Engineering ,Hyperparameter optimization ,Transfer learning ,White blood cell - Abstract
White blood cells play a role in maintaining the immune system which consists of several types such as neutrophils, lymphocytes, monocytes, eosinophils and basophils. MobileNetV2 is one of the pretrained convolutional neural network (CNN) models that provides excellent advantages and performance in classifying images. In this research was conducted to find out how to apply optimization hyperparameters and the impact of image processing on white blood cell image classification using MobileNetV2, so that it is expected to find a combination of preprocessing and combination of hyperparameter values that can produce the highest accuracy value. To maximize the classification process, before classifying the image, several stages of image preprocessing are carried out, namely cropping, grayscale, resizing and augmentation. Hyperparameter tuning was carried out for an experiment to improve model performance. The three main parameters used in hyperparameter tuning are learning rate, batch size, and number of epochs. Performance optimization model performance will be measured using accuracy, sensitivity, specificity and using a confusion matrix. Based on the experimental results in this study, it shows that the best learning rate value is 0.00001, the best batch size value is 32, and the best epoch value is 250.
- Published
- 2023