基于SG-RF-MTBLSTM模型的湿地水质预测研究
Research on Wetland Water Quality Prediction Based on the SG-RF-MTBLSTM Model
摘要: 为解决湿地水质预测受潮汐气象等环境、参数耦合及极端天气干扰,本文提出SG-RF-MTBLSTM混合预测模型。对近岸湿地溶解氧(DO)与原位硝酸盐(NO3)进行预测,两者受潮汐、气象与参数耦合共同影响:先对连续小时时序作Savitzky-Golay (SG)平滑,经随机森林(RF)得到精简协变量子集,再以共享BiLSTM骨干 + DO/NO3双头,在24 h滑动窗口上同步估计窗末时刻的溶解氧与硝酸盐。基于USGS连续探头站2020~2025年小时数据,站点为USGS-02292900,按时间划分为2020~2022年训练,2023~2024年验证,2025年测试。结果表明,模型测试集DO的R2为0.80,NO3的R2为0.69;以6~11月为飓风季分层后,飓风季DO的MAE由正常期0.59 mg/L升至0.94 mg/L、R2由0.74降至0.65,NO3同步轻度下降,存在季节性分布外泛化困难。
Abstract: To address the challenges of coastal wetland water quality prediction under tidal-meteorological forcing, cross-parameter coupling, and extreme-weather disturbances, this paper proposes the SG-RF-MTBLSTM hybrid prediction model. The targets are nearshore dissolved oxygen (DO) and in-situ nitrate (NO3), which are jointly driven by tidal, meteorological, and biogeochemical factors. Continuous hourly series are first smoothed with a Savitzky-Golay (SG) filter; a random forest (RF) then yields a compact covariate subset; and a shared bidirectional LSTM (BiLSTM) backbone with dual DO/NO₃ heads jointly estimates window-end DO and nitrate over a 24 h sliding window. Experiments use USGS continuous-probe hourly data for 2020~2025 at the primary site USGS-02292900, with a chronological split into training (2020~2022), validation (2023~2024), and testing (2025). On the test set, the model attains an R2 of 0.80 for DO and 0.69 for NO3. After a June-November hurricane-season stratification, hurricane-season DO MAE rises from 0.59 mg/L in the normal period to 0.94 mg/L and R2 falls from 0.74 to 0.65, with a mild concurrent decline for NO₃, indicating seasonal out-of-distribution generalization difficulty.
文章引用:马滨滨, 梁雅晴, 何雨薇, 叶晓霞. 基于SG-RF-MTBLSTM模型的湿地水质预测研究[J]. 人工智能与机器人研究, 2026, 15(6): 1358-1365. https://doi.org/10.12677/airr.2026.156123

参考文献

[1] Liang, Z., Zou, R., Chen, X., et al. (2020) Simulate the Forecast Capacity of a Complicated Water Quality Model Using the Long Short-Term Memory Approach. Journal of Hydrology, 581, Article 124432.
https://doi.org/10.1016/j.jhydrol.2019.124432
[2] Xu, J., Xu, Z., Kuang, J., et al. (2021) An Alternative to Laboratory Testing: Random Forest-Based Water Quality Prediction Framework for Inland and Nearshore Water Bodies. Water, 13, 3262.
https://doi.org/10.3390/w13223262
[3] 董玉萍, 于泉洲, 宋军, 等. 融合无人机与卫星遥感的南四湖菹草生物量制图及时空格局[J]. 湿地科学与管理, 2025, 21(5): 17-25.
[4] 许轲桐, 曾鸿滨, 陈丽萍, 等. 基于小波去噪的LSTM-Transformer模型在河流断面水质预测分析中的应用研究[J]. 环境工程, 2026, 44(3): 125-135.
[5] Ruder, S. (2017) An Overview of Multi-Task Learning in Deep Neural Networks. arXiv preprint arXiv:1706.05098.
[6] Liu, Y., Hu, T., Zhang, H., et al. (2024) iTransformer: Inverted Transformers Are Effective for Time Series Forecasting. International Conference on Learning Representations, Vienna, 7-11 May 2024, 11116-11140.