IAES International Journal of Artificial Intelligence (IJ-AI)
Vol 12, No 3: September 2023

Image preprocessing and hyperparameter optimization on pretrained model MobileNetV2 in white blood cell image classification

Parmonangan R. Togatorop (Institut Teknologi Del)
Yohanssen Pratama (Institut Teknologi Del)
Astri Monica Sianturi (Institut Teknologi Del)
Mega Sari Pasaribu (Institut Teknologi Del)
Pebri Sangmajadi Sinaga (Institut Teknologi Del)



Article Info

Publish Date
01 Sep 2023

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.

Copyrights © 2023






Journal Info

Abbrev

IJAI

Publisher

Subject

Computer Science & IT Engineering

Description

IAES International Journal of Artificial Intelligence (IJ-AI) publishes articles in the field of artificial intelligence (AI). The scope covers all artificial intelligence area and its application in the following topics: neural networks; fuzzy logic; simulated biological evolution algorithms (like ...