Prediction of Potassium Content in Rice Leaves Based on Spectral Features and Random Forests

Abstract

The information acquisition about potassium, which affects the quality and yield of crops, is of great significance for crop nutrient management and intelligent decision making in smart agriculture. This article proposes a method for predicting the rice leaf potassium content (LKC) using spectral characteristics and random forests (RF). The method screens spectral characteristic variables based on the linear correlation analysis results of rice LKC and four transformed spectra (original reflectance (R), first derivative reflectance (FDR), continuum-removed reflectance (CRR), and normalized reflectance (NR)) of leaves and the PCA dimensionality reduction results of vegetation indices. Following a second screening of the correlated single band and vegetation index variables of the four transformed spectra, the RF is used to obtain the mixed variable (MV), and regression models are developed to achieve an accurate prediction of rice LKC. Additionally, the effect of potassium spectral sensitivity bands, indices, spectral transformation form, and different modeling methods on rice LKC prediction accuracy is assessed. The results showed that the mixed variable obtained with the second screening using the random forest feature selection method could effectively improve the prediction accuracy of rice LKC. The regression models based on the single band variables (BV) and the vegetation index variables (IV), FDR–RF and IV–RF, with R2 values of 0.62301 and 0.7387 and RMSE values of 0.24174 and 0.15045, respectively, are the best models. In comparison to the previous two models, the MV–RF validation had a higher R2 and a lower RMSE, reaching 0.77817 and 0.14913, respectively. It can be seen that the RF has a better processing ability for the MV that contains vegetation indices and IV than for the BV. Furthermore, the results of different variable screening and regression analyses also revealed that the single band’s range of 1402–1428 nm and 1871–1907 nm, as well as the vegetation indices constituted of reflectance 1799–1881 nm and 2276–2350 nm, are of great significance for predicting rice LKC. This conclusion can provide a reference for establishing a universal vegetation index related to potassium.