Enhancing coffee bean classification: a comparative analysis of pre-trained deep learning models

Esraa Hassan

DOI: https://doi.org/10.1007/s00521-024-09623-z

2024-04-01

Neural Computing and Applications

Abstract:Abstract Coffee bean production can encounter challenges due to fluctuations in global coffee prices, impacting the economic stability of some countries that heavily depend on coffee production. The primary objective is to evaluate how effectively various pre-trained models can predict coffee types using advanced deep learning techniques. The selection of an optimal pre-trained model is crucial, given the growing popularity of specialty coffee and the necessity for precise classification. We conducted a comprehensive comparison of several pre-trained models, including AlexNet, LeNet, HRNet, Google Net, Mobile V2 Net, ResNet (50), VGG, Efficient, Darknet, and DenseNet, utilizing a coffee-type dataset. By leveraging transfer learning and fine-tuning, we assess the generalization capabilities of the models for the coffee classification task. Our findings emphasize the substantial impact of the pre-trained model choice on the model's performance, with certain models demonstrating higher accuracy and faster convergence than conventional alternatives. This study offers a thorough evaluation of pre-trained architectural models regarding their effectiveness in coffee classification. Through the evaluation of result metrics, including sensitivity (1.0000), specificity (0.9917), precision (0.9924), negative predictive value (1.0000), accuracy (1.0000), and F1 score (0.9962), our analysis provides nuanced insights into the intricate landscape of pre-trained models.

computer science, artificial intelligence

What problem does this paper attempt to address?

The paper focuses on addressing the challenge of accurately classifying different types of coffee beans, which is important for ensuring the quality and flavor profile of the final coffee product. The primary objective of the study is to evaluate the effectiveness of various pre-trained deep learning models in predicting coffee types. The authors conduct a comprehensive comparison of different pre-trained models, including AlexNet, LeNet, HRNet, GoogleNet, MobileNetV2, ResNet-50, VGG, EfficientNet, Darknet, and DenseNet, using a coffee type dataset. The study employs transfer learning and fine-tuning techniques to assess the generalization capabilities of these models for the specific task of coffee classification. The choice of an optimal pre-trained model is highlighted as crucial, particularly given the growing demand for specialty coffee and the need for precise classification. The findings of the study emphasize the significant impact of the pre-trained model choice on model performance, with certain models demonstrating higher accuracy and faster convergence than others. Key contributions of the study include: 1. **Exploring the Importance of Model Selection:** Emphasizing the need for precise classification in specialty coffee types and the importance of selecting the right pre-trained deep learning model. 2. **Comparative Analysis:** Comparing various pre-trained models to understand their strengths and weaknesses for specific coffee classification tasks. 3. **Transfer Learning and Fine-Tuning:** Utilizing transfer learning and fine-tuning techniques to improve the performance of pre-trained models for coffee classification.

Enhancing coffee bean classification: a comparative analysis of pre-trained deep learning models

The Evaluation of Deep Learning Using Convolutional Neural Network (CNN) Approach for Identifying Arabica and Robusta Coffee Plants

Coffee Type Classification Using Backpropagation Artificial Neural Network

Machine learning techniques for coffee classification: a comprehensive review of scientific research

Machine Learning Classification of Roasted Arabic Coffee: Integrating Color, Chemical Compositions, and Antioxidants

PredCoffee: A binary classification approach specifically for coffee odor

Coffee Roast Intelligence

CNN-HOG based hybrid feature mining for classification of coffee bean varieties using image processing

Multiscale Defect Extraction Neural Network for Green Coffee Bean Defects Detection

Study of active food processing technology using computer vision and AI in coffee roasting

Novel authentication of African geographical coffee types (bean, roasted, powdered) by handheld NIR spectroscopic method

Intelligent classifier for various degrees of coffee roasts using smart multispectral vision system

Non-linear machine learning coupled near infrared spectroscopy enhanced model performance and insights for coffee origin traceability

Artificial intelligence-based solutions for coffee leaf disease classification

Discriminating Robusta coffee (Coffea canephora) cropping systems using leaf-level hyperspectral data

Automated Prediction of Brewing Methods for Coffee Flavors by Multiple Linear Regression, Support Vector Regression, Random Forest, and Neural Network

Rapid Identification of Coffee Bean Variety by Near Infrared Hyperspectral Imaging Technology

Determination of the Geographical Origin of Coffee Beans Using Terahertz Spectroscopy Combined With Machine Learning Methods

Tea grading, blending, and matching based on computer vision and deep learning

Development of a novel 2D single coffee bean model and comparison with a 3D model under varying heating profiles

Advanced Classification Of Coffee Beans With Fatty Acids Profiling To Block Information Loss