Inversion of heavy metal content in soil using hyperspectral characteristic bands-based machine learning method

Abstract

The global concern regarding the adverse effects of heavy metal pollution in soil has grown significantly. Accurate prediction of heavy metal content in soil is crucial for environmental protection. This study proposes an inversion analysis method for heavy metals (As, Cd, Cr, Cu, Ni, Pb) in soil based on hyperspectral and machine learning algorithms for 21 soil reference materials from multiple provinces in China. On this basis, an integrated learning model called Stacked RF (the base model is XGBoost, LightGBM, CatBoost, and the meta-model is RF) was established to perform soil heavy metal inversion. Specifically, three popular algorithms were initially employed to preprocess the spectral data, then Random Forest (RF) was used to select the best feature bands to reduce the impact of noise, finally Stacking and four basic machine learning algorithms were used to establish comparisons and analysis of inversion model. Compared with traditional machine learning methods, the stacking model showcases enhanced stability and superior accuracy. Research results indicate that machine learning algorithms, especially ensemble learning models, have better inversion effects on heavy metals in soil. Overall, the MF-RF-Stacking model performed best in the inversion of the six heavy metals. The research results will provide a new perspective on the ensemble learning model method for soil heavy metal content inversion using data of hyperspectral characteristic bands collected from soil reference materials.