Enhancing Spectroscopy-Based Fruit Quality Control: A Knowledge-Guided Machine Learning Approach to Reduce Model Uncertainty

Jie Yang,Zhizhong Sun,Shijie Tian,Hao Jiang,Jintao Feng,K. C. Ting,Tao Lin,Yibin Ying
DOI: https://doi.org/10.1016/j.postharvbio.2024.113009
IF: 6.751
2024-01-01
Postharvest Biology and Technology
Abstract:Spectroscopy-based techniques have made remarkable advancements in their application to fruit quality control but encounter challenges of high model uncertainty arising from biological variability. Minor changes in spectral measurement orientations or positions significantly altered the model prediction for fruit quality, which severely affects the reliability of online fruit grading systems. This study assesses the influence of varying structural designs in deep learning models and develops a Knowledge-Guided Convolutional Neural Network (KGCNN) to mitigate predictive uncertainty caused by different orientations. A novel loss criterion that quantifies model uncertainty incorporating a grouped input strategy is introduced, enabling the implementation of a knowledgeguided approach. Experiments are conducted on Mandarin and Valencia orange datasets collected on an onsite grading system for soluble solid content assessment. The proposed KGCNN model effectively reduces 27.1 % and 25.5 % averaged prediction variance compared with CNN and PLS models on five detections of random orientations, demonstrating a significant mitigation of model uncertainty. This strategy operates within the framework of the developed model without the need for structural hyperparameter adjustments, with a minor impact on the test accuracy, evidenced by a 1.7 % increase in RMSE and a 0.2 % decrease in R2 averaged on two datasets. A hidden feature visualization technique is further employed to explain the working mechanism of the decreased model uncertainty by KGCNN. The visual evidence supports the effectiveness of the knowledge-guided approach in mitigating the nonlinear scatter within the distributions of learned features from input data collected at multiple orientations, attributed to the constraint effect of the introduced loss criterion. This research has the potential to alleviate the stringent requirements on fruit loading or measurement positions within onsite grading systems and enhance the reliability of deep learning models in practical applications.
What problem does this paper attempt to address?