An Automatic-Segmentation- and Hyper-Parameter-Optimization-Based Artificial Rabbits Algorithm for Leaf Disease Classification

Biomimetics (Basel). 2023 Sep 19;8(5):438. doi: 10.3390/biomimetics8050438.

Abstract

In recent years, disease attacks have posed continuous threats to agriculture and caused substantial losses in the economy. Thus, early detection and classification could minimize the spread of disease and help to improve yield. Meanwhile, deep learning has emerged as the significant approach to detecting and classifying images. The classification performed using the deep learning approach mainly relies on large datasets to prevent overfitting problems. The Automatic Segmentation and Hyper Parameter Optimization Artificial Rabbits Algorithm (AS-HPOARA) is developed to overcome the above-stated issues. It aims to improve plant leaf disease classification. The Plant Village dataset is used to assess the proposed AS-HPOARA approach. Z-score normalization is performed to normalize the images using the dataset's mean and standard deviation. Three augmentation techniques are used in this work to balance the training images: rotation, scaling, and translation. Before classification, image augmentation reduces overfitting problems and improves the classification accuracy. Modified UNet employs a more significant number of fully connected layers to better represent deeply buried characteristics; it is considered for segmentation. To convert the images from one domain to another in a paired manner, the classification is performed by HPO-based ARA, where the training data get increased and the statistical bias is eliminated to improve the classification accuracy. The model complexity is minimized by tuning the hyperparameters that reduce the overfitting issue. Accuracy, precision, recall, and F1 score are utilized to analyze AS-HPOARA's performance. Compared to the existing CGAN-DenseNet121 and RAHC_GAN, the reported results show that the accuracy of AS-HPOARA for ten classes is high at 99.7%.

Keywords: Artificial Rabbits Algorithm; Automatic Segmentation; Hyper Parameter Optimization; leaf disease classification; synthetic images.