首页|期刊导航|长江科学院院报|基于SSA-IDBO-GRU-DCM的水质预测模型及应用

基于SSA-IDBO-GRU-DCM的水质预测模型及应用OA

Water Quality Prediction Model Based on Residual Correction and Optimization of Gated Recurrent Unit and Its Application

中文摘要英文摘要

水库水质预测是城市供水引水工程的关键.为提高水库水体溶解氧含量预测精度,提出一种基于奇异谱分析(SSA)、改进的蜣螂算法(IDBO)、优化门控循环单元(GRU)以及残差序列预测差值校正方法(DCM)的混合预测模型SSA-IDBO-GRU-DCM.该混合预测模型首先将溶解氧含量时间序列进行SSA后重构,将重构后的趋势、周期和残差的分量输入到GRU进行预测,对于复杂的残差分量提出VMD-GRU预测差值校正模型以提高预测精度,采用IDBO优化网络模型中的超参数,最后将3个分量进行叠加得到预测结果.利用该混合预测模型对河北唐山大黑汀水库溶解氧含量进行预测.结果表明,提出的混合预测模型均方根误差为0.580 2 mg/L、平均绝对误差为0.329 2 mg/L、决定系数R2为0.918 8.相较其他模型,混合预测模型的预测精度更高.

[Objective]The time series of reservoir water quality indices,especially dissolved oxygen content,ex-hibit strong nonlinearity,high complexity,and uncertainty,which lead to insufficient accuracy of single prediction models.This study aims to construct a high-precision hybrid prediction model that integrates time series decomposi-tion,intelligent optimization,and residual correction,thereby significantly improving the prediction accuracy of dis-solved oxygen(DO)content and providing reliable support for water environment management and pollution early warning.[Methods]The core procedures of the proposed hybrid prediction model are as follows.1)Data decom-position and reconstruction.Singular spectrum analysis(SSA)was applied to decompose the dissolved oxygen time series,and the series was reconstructed into trend components,periodic components,and residual components to reduce sequence complexity and highlight features at different frequencies.2)An improved dung beetle optimizer(IDBO)which integrates piecewise chaotic mapping and opposition-based learning strategies was designed to en-hance population diversity and initialization quality.The improved IDBO was used to optimize key hyperparameters of the GRU network,including the number of hidden layer neurons and the initial learning rate.3)Component pre-diction and residual correction.The GRU model optimized by IDBO was used to predict the trend component and periodic components separately.A residual series prediction difference correction method(DCM)was proposed.The residual component was first predicted using GRU,and the difference sequence between the predicted values and the observed values was calculated.Variational mode decomposition(VMD)was then applied to the difference sequence to fully extract high-frequency detail information.Each decomposed component was predicted using GRU and aggregated to obtain the predicted difference values.Finally,the predicted differences were compensated into the initial residual prediction to obtain the corrected residual prediction results.4)Model integration and validation.The prediction results of the three components were aggregated to obtain the final DO prediction values.Measured dissolved oxygen data from Daheiting Reservoir in Tangshan,Hebei Province were used for experiments.The data-set contained 2352 records with a sampling interval of four hours.Root mean square error(RMSE),mean absolute error(MAE),mean relative error(MRE),and the coefficient of determination(R2)were used as evaluation met-rics.The proposed model was compared with GRU,SSA-GRU,SSA-DBO-GRU,SSA-IDBO-GRU,and models re-ported in the literature such as LSTM and PSO-GRU.[Results]The proposed SSA-IDBO-GRU-DCM hybrid model achieved the best performance among all comparative models.The prediction errors were significantly reduced,with an RMSE of 0.580 2 mg/L,an MAE of 0.329 2 mg/L,an MRE of 0.0269,and an R2 of 0.918 8.Ablation experi-ments confirmed that the proposed IDBO improvement strategies effectively enhanced hyperparameter optimization accuracy.The residual difference correction method(DCM)significantly improved the prediction performance of the residual component and was the key factor contributing to the overall accuracy improvement.These results fully demonstrated the effectiveness and superiority of the"decomposition-optimization-correction"framework.[Conclusion]SSA effectively decouples the complex characteristics of water quality time series.IDBO efficiently and accurately optimizes GRU hyperparameters.The proposed VMD-GRU-based residual difference correction meth-od(DCM)is the key innovation for improving overall prediction accuracy.The proposed model significantly im-proves the prediction accuracy of dissolved oxygen content and provides an efficient and reliable new approach for reservoir dissolved oxygen prediction.Future work can extend this framework to the prediction of other key water quality parameters such as ammonia nitrogen and total phosphorus,and further explore the integration of natural ev-olutionary strategies to improve computational efficiency and generalization ability.

郭利进;陈剑铮

天津工业大学控制科学与工程学院,天津 300387||天津工业大学天津市电气装备智能控制重点实验室,天津 300387天津工业大学控制科学与工程学院,天津 300387||天津工业大学天津市电气装备智能控制重点实验室,天津 300387

信息技术与安全科学

水质预测奇异谱分析蜣螂算法门控循环单元溶解氧

water quality predictionsingular spectrum analysisdung beetle optimizergated recurrent unitdis-solved oxygen

《长江科学院院报》 2026 (3)

46-54,87,10

国家自然科学基金项目(52077155)

10.11988/ckyyb.20250127

评论