Water Science and Technology (Jan 2024)

A method based on improved ant colony algorithm feature selection combined with GWO-SVR model for predicting chlorophyll-a concentration in Wuliangsu Lake

  • Chenhao Wu,
  • Xueliang Fu,
  • Honghui Li,
  • Hua Hu,
  • Xue Li,
  • Liqian Zhang

DOI
https://doi.org/10.2166/wst.2023.410
Journal volume & issue
Vol. 89, no. 1
pp. 20 – 37

Abstract

Read online

Chlorophyll-a (Chl-a) is an important parameter in water bodies. Due to the complexity of optics in water bodies, it is difficult to accurately predict Chl-a concentrations in water bodies by current traditional methods. In this paper, using Sentinel-2 remote sensing images as the data source combined with measured data, taking Wuliangsu Lake as the study area, a new intelligent algorithm is proposed for prediction of Chl-a concentration, which uses the adaptive ant colony exhaustive optimization algorithm (A-ACEO) for feature selection and the gray wolf optimization algorithm (GWO) to optimize support vector regression (SVR) to achieve Chl-a concentration prediction. The ant colony optimization algorithm is improved to select remote sensing feature bands for Chl-a concentration by introducing relevant optimization strategies. The GWO-SVR model is built by optimizing SVR using GWO with the selected feature bands as input and comparing it with the traditional SVR model. The results show that the usage of feature bands selected by the presented A-ACEO algorithm as inputs can effectively reduce complexity and improve the prediction performance of the model, under the condition of the same model, which can provide valuable references for monitoring the Chl-a concentration in Wuliangsu Lake. HIGHLIGHTS Second derivative as a pretreatment method can improve the prediction accuracy of machine learning.; Adaptive ant colony exhaustive optimization algorithm can more effectively select the feature bands related to Chl-a.; Combining gray wolf optimization algorithm with support vector regression can achieve better prediction accuracy.; The prediction method provides a new way for predicting Chl-a concentration in lake wetlands.;

Keywords