Estimating low concentration heavy metals in water through hyperspectral analysis and genetic algorithm-partial least squares regression

Hyperspectral spectrum enables assessment of heavy metal content, but research on low concentration in water is limited. This study employed in situ hyperspectral data from Dalian Lake, Shanghai to develop a machine learning model for accurately determining heavy metal concentrations. Initially, we employed a combination of empirical analysis and algorithm-based analysis to identify the optimal features for retrieving Cu and Fe ions. Based on the correlation coefficients between heavy metals and water quality, the feature bands for TOC, Chl-a and TP were selected as empirical features. Algorithm-based feature selection was conducted by employing the random forest (RF) approach with the original spectrum (OR), first-order derivative reflectance (FDR), and second-order derivative reflectance (SDR). For the development of a prediction model, we utilized the Genetic Algorithm-Partial Least Squares Regression (GA-PLSR) approach for Cu and Fe ions inversion. Our findings demonstrated that the integration of both empirical features and algorithm-selected features resulted in superior performance compared to using algorithm-selected features alone. Importantly, the crucial wavelength data primarily located at 497, 665, 686, 831 and 935 nm showed superior results for Cu retrieval, while wavelengths of 700, 746, 801, 948, and 993 nm demonstrated better results for Fe retrieval. These results also displayed that the GA-PLSR model outperformed both the PLSR and RF models, exhibiting an R