UPM Institutional Repository

Herbal plant image classification using transfer learning and fine-tuning deep learning model


Citation

Khalid, Fatimah and Romle, Amirul Azuani (2024) Herbal plant image classification using transfer learning and fine-tuning deep learning model. Journal of Advanced Research in Applied Sciences and Engineering Technology, 35 (1). pp. 16-25. ISSN 2462-1943

Abstract

Herbal plants are highly significant to the local community in Malaysia, as the country's fertile land is rich in diverse species that are widely used for various purposes, including traditional medicine, culinary, aromatherapy, and even in the cosmetic industries. This situation demands numerous applications and activities, including plant species identification, medicinal plant research, agriculture research, and environmental monitoring, which makes image classification of herbal plants a substantial task. However, this task is complicated by the complex nature of the plants, particularly includes plants variations in appearance, close similarities between species, and limited availability of labeled data and images. Motivated to mitigate the issues, this paper investigated the use of transfer learning and fine- tuning the deep learning neural network to classify different herbal plant species. Transfer learning is an algorithm that learns to recognize image features in one domain and having the capability to generalize the learnt knowledge to a new domain with a smaller dataset. Additionally, fine-tuning can be used to further improve the performance of the model on the new task with less training time and fewer training data. The authors performed experiments on ResNet-50 which been previously trained with ImageNet dataset. The experiments were carried out on a subset of the MYLPherbs-1 dataset, which consisted of two local perennial herbs plant species. Different hyperparameters were used across the various experiment settings, and the authors observed the behavior and relationships of the distinct models, datasets, and hyperparameters toward the classification task's accuracy. The authors also employed two different transfer learning approaches: (i) using pre-trained models as feature extractors with different classifiers, (ii) fine-tuning the pre-trained model. Based on the results and discussion, fine-tuning the ResNet-50 model on the MYLPherb-1 dataset demonstrated the best overall performance.


Download File

Full text not available from this repository.

Additional Metadata

Item Type: Article
Divisions: Faculty of Computer Science and Information Technology
DOI Number: https://doi.org/10.37934/araset.34.3.1625
Publisher: Semarak Ilmu
Keywords: Herbal plant; ResNet-50; Transfer learning; Fine tuning; Deep learning; Image classification
Depositing User: Ms. Zaimah Saiful Yazan
Date Deposited: 13 Jun 2024 04:17
Last Modified: 13 Jun 2024 04:17
Altmetrics: http://www.altmetric.com/details.php?domain=psasir.upm.edu.my&doi=10.37934/araset.34.3.1625
URI: http://psasir.upm.edu.my/id/eprint/108167
Statistic Details: View Download Statistic

Actions (login required)

View Item View Item