基于LDV和可解释1D-CNN的皇冠梨硬度预测方法

    Crown pear firmness prediction based on LDV and interpretable 1D-CNN

    • 摘要: 为研究皇冠梨振动频率特征与果实硬度的关系,改善现有研究中常规预测模型精度较低而深度学习模型缺乏可解释性的问题。该研究通过激光多普勒测振仪(laser doppler vibrometer,LDV)采集皇冠梨振动数据,采用一维卷积神经网络(one dimensional-convolutional neural network,1D-CNN)算法建立基于振动数据的皇冠梨硬度预测模型,并使用深度沙普利加性解释(deep shapleyadditive explanations,Deep SHAP)框架对预测模型进行解释。通过与其他经典预测模型相比,1D-CNN预测模型可以利用特征频率实现对皇冠梨硬度的高精度预测(RP2=0.945,RMSEP=0.594 N/mm和RPDP=4.272)。基于SHAP框架解释的结果表明,皇冠梨300~700 Hz频率特征与硬度联系密切。该研究展示了1D-CNN模型在水果硬度预测应用中的优异性能和巨大潜力,为振动特征频率应用于硬度预测和分析的研究提供理论基础。

       

      Abstract: A Crown pear is one of the most primary fresh fruit varieties worldwide, due to the high juice content, sweetness, and rich nutritional value. Among them, the firmness can represent one of the most important quality indicators for pears. It can directly present how ripe the pear is and how well the pear can be stored after harvest. For these reasons, it is often required to accurately measure the firmness of the pear. The conventional measurement of fruit firmness can depend mainly on the destructive tests, including the Magness-Taylor puncture and compression. However, the large amount of food waste cannot fully meet the large-scale testing in recent years. Also, the current studies still share the two challenges, whether the regular prediction has low accuracy or the deep learning is difficult to interpret. This research aims to examine the connection between fruit firmness and vibration frequency in Crown pears. A recognition model was also established using a standard convolutional neural network (CNN). After that, an improved CNN was combined with Deep Shapley Additive Explanations. A series of experiments was carried out to verify the improved model. The 508 Crown pears were taken as the test samples. Five groups were divided for the samples, each of which was tested every thirteen days. Vibration data was gathered from the Crown pears using a Laser Doppler Vibrometer (LDV). An improved one-dimensional CNN was applied to construct a firmness prediction model, according to the characteristic frequencies. Deep Shapley Additive Explanations (Deep SHAP) structure was used to explain the function of the prediction model. Furthermore, a comparison was made of several standard prediction models, including the partial least squares regression, support vector regression, extreme gradient boosting, and adaptive gradient boosting. The results show that the improved one-dimensional CNN model was achieved in the high-precision prediction of the Crown pear hardness. Among them, the feature frequency of the RP2 was 0.945, the RMSEP was 0.594 N/mm, and the RPDP was 4.272. The performance also outperformed all conventional models over all metrics. The most outstanding performance was achieved, where the R² value was closest to the ideal value of 1, the RMSE was the lowest, was the next-best XGBoost model, and the RPD value was significantly higher than the rest of the models, indicating the superior generalization. The characteristic frequencies were utilized to predict the Crown pear firmness with high accuracy. Vibration frequency features between 300 and 700 Hz shared a strong relationship with the pear firmness. An interpretable deep learning model with the LDV vibration data was achieved in the high-precision prediction for interpretability and practicality. There was a strong correlation between characteristic frequency and firmness. The finding can provide an efficient and reliable non-destructive testing of the Crown pear firmness in the assessment of the fruit quality.

       

    /

    返回文章
    返回