A1 Refereed original research article in a scientific journal

A comparative study of state-of-the-art deep learning architectures for rice grain classification




AuthorsFarahnakian Farshad, Sheikh Javad, Farahnakian Fahimeh, Heikkonen Jukka

PublisherElsevier BV

Publication year2024

JournalJournal of agriculture and food research

Article number100890

Volume15

eISSN2666-1543

DOIhttps://doi.org/10.1016/j.jafr.2023.100890

Web address https://doi.org/10.1016/j.jafr.2023.100890

Self-archived copy’s web addresshttps://research.utu.fi/converis/portal/detail/Publication/181985653


Abstract

Accurate and efficient automated rice grain classification systems are vital for rice producers, distributors, and traders, offering improved quality control, cost optimization, and supply chain management. They also hold the potential to aid in the development of rice varieties that are more resistant to disease, pests, and environmental stress. While most existing studies in the rice classification domain rely on traditional machine-learning techniques that necessitate feature extraction engineering processes, our research explores the effectiveness of novel deep-learning models for this task. We evaluated the performance of various contemporary deep-learning models, including Residual Network (ResNet), Visual Geometry Group (VGG) network, EfficientNet, and MobileNet. These models were tested on a dataset comprising 75,000 images, classified into five different rice categories. We assessed each model using established evaluation metrics such as accuracy, F1 score, precision, recall, and per-class accuracy. Our findings showed that the EfficientNet-based model delivered the highest accuracy (99.67%), while the MobileNet-based model excelled in the speed of classification (2556 s). We concluded that, compared to traditional machine learning methods, the models employed in our study are highly scalable and capable of managing large volumes of complex data with millions of features and samples.


Downloadable publication

This is an electronic reprint of the original article.
This reprint may differ from the original in pagination and typographic detail. Please cite the original version.





Last updated on 2024-26-11 at 12:19