A Multi-Target Regression Method to Predict Element Concentrations in Tomato Leaves Using Hyperspectral Imaging

Abstract

Recent years have seen the development of novel, rapid, and inexpensive techniques for collecting plant data to monitor the nutritional status of crops. These techniques include hyperspectral imaging, which has been widely used in combination with machine learning models to predict element concentrations in plants. When there are multiple elements, the machine learning models are trained with spectral features to predict individual element concentrations; this type of single-target prediction is known as single-target regression. Although this method can achieve reliable accuracy for some elements, there are others that remain less accurate. We aimed to improve the accuracy of element concentration predictions by using a multi-target regression method that sequentially augmented the original input features (hyperspectral imaging) by chaining the predicted element concentration values. To evaluate the multi-target method, the concentrations of 17 elements in tomato leaves were predicted and compared with the single-target regression results. We trained 5 machine learning models with hyperspectral data and predicted element concentration values and found a significant improvement in the prediction accuracy for 10 elements (Mg, P, S, Mn, Fe, Co, Cu, Sr, Mo, and Cd). Furthermore, our multi-target regression method outperformed single-target predictions by increasing the coefficient of determination (R2) for elements such as Mn, Cu, Co, Fe, and Mg by 12.5%, 10.3%, 11%, 10%, and 8.4%, respectively. Hence, our multi-target method can improve the accuracy of predicting 10-element concentrations compared to single-target regression.