Assessment of Various Machine Learning Models for Peach Maturity Prediction Using Non-Destructive Sensor Data

Sensors (Basel). 2022 Aug 3;22(15):5791. doi: 10.3390/s22155791.

Abstract

To date, many machine learning models have been used for peach maturity prediction using non-destructive data, but no performance comparison of the models on these datasets has been conducted. In this study, eight machine learning models were trained on a dataset containing data from 180 'Suncrest' peaches. Before the models were trained, the dataset was subjected to dimensionality reduction using the least absolute shrinkage and selection operator (LASSO) regularization, and 8 input variables (out of 29) were chosen. At the same time, a subgroup consisting of the peach ground color measurements was singled out by dividing the set of variables into three subgroups and by using group LASSO regularization. This type of variable subgroup selection provided valuable information on the contribution of specific groups of peach traits to the maturity prediction. The area under the receiver operating characteristic curve (AUC) values of the selected models were compared, and the artificial neural network (ANN) model achieved the best performance, with an average AUC of 0.782. The second-best machine learning model was linear discriminant analysis with an AUC of 0.766, followed by logistic regression, gradient boosting machine, random forest, support vector machines, a classification and regression trees model, and k-nearest neighbors. Although the primary parameter used to determine the performance of the model was AUC, accuracy, F1 score, and kappa served as control parameters and ultimately confirmed the obtained results. By outperforming other models, ANN proved to be the most accurate model for peach maturity prediction on the given dataset.

Keywords: AUC; artificial neural networks; dimensionality reduction; fruit quality; group lasso; lasso regularization; machine learning; non-destructive measurements; peach maturity prediction.

MeSH terms

  • Logistic Models
  • Machine Learning
  • Neural Networks, Computer
  • Prunus persica*
  • Support Vector Machine