From Pixels to Spectra: Predicting Wine Colorimetric Characteristics Through Machine Learning Models

Abstract

This study employs digital image processing and machine learning techniques to predict all colorimetric characteristics (color intensity, density, tonality, and color index percentages) of colored wines in a novel, cost-effective, and rapid manner. To determine the values of colorimetric characteristics, ultraviolet-visible (UV-Vis) absorbance was measured at three key wavelengths (A420, A520, and A620) using UV-Vis spectrophotometry, corresponding to the yellow, red, and blue color percentages, respectively. Simultaneously, the pictures of 86 wine samples were acquired, and the corresponding RGB and HSV color values were extracted from the images to serve as input features for multiple regression models. The models developed included principal component regression, k-nearest neighbors, linear regression, decision tree, random forest, and partial least squares (PLS). Among the models, random forest outperformed PLS in predicting A620 absorbance value due to its ability to capture non-linear patterns, whereas PLS demonstrated greater accuracy (R2 > 0.95) in predicting the A420 and A520 absorbance values. According to feature selection, hue and saturation had the biggest impact on prediction accuracy. By determining absorbance values using the developed models, the complete colorimetric characteristics of the wine samples can be calculated, enabling the evaluation of their physicochemical parameters during the fermentation process or post-fermentation. As a result, all the models, improved, offer a promising alternative for quick, easy, and scalable prediction methods by reducing measurement time, eliminating the need for laboratory instruments, and introducing a new methodology to complement conventional spectroscopic techniques, with potential applications in consumer-level analysis and the process of wine quality control.