基于注意力机制优化的LSTM河流溶解氧预测模型研究

溶解氧（DO）是水体中的重要水质指标，构建数据驱动模型，实现对溶解氧的准确预测，将为水环境管理提供科学有效的技术手段。考虑到溶解氧序列数据非线性强、非平稳性突出的特点，提出一种基于双阶段注意力权重优化机制的长短时记忆网络（long short-term memory，LSTM）的河流溶解氧预测模型（DAIW-LSTM模型），该模型的编码器包含双阶段权重优化的空间注意力机制，而解码器包含双阶段权重优化的时间注意力机制。将该模型应用于流溪河流域白云李溪坝、流溪河山庄、从化街口等水质监测站溶解氧日均值预测的研究，开展了该模型与DA-LSTM、LSTM、Bi-LSTM等基线模型的预测效果对比分析，探讨了特征权重优化机制及上游站点水质数据输入对模型预测性能的影响。结果表明：（1）通过与基线模型的预测效果对比，验证了DAIW-LSTM模型的精准性，其对白云李溪坝站溶解氧预测的对称平均绝对百分比误差（SMAPE）、平均绝对误差（MAE）、均方误差（MSE）分别为0.075、0.611、0.712，在所有模型中最优。（2）对于新的注意力权重优化机制，第二阶段会对第一阶段的初步权重进行优化修正；针对pH、电导率、水温、气温等影响溶解氧预测的重要特征，DAIW-LSTM模型会自适应调整其权重在时间序列上的分布，从而提高该模型的预测精度。（3）加入上游水质特征的输入影响，通过9个组合试验对比可知，DAIW-LSTM模型仍然为表现最佳的模型，该系列组合试验也证明上游站点及其特征变量选取的重要性。研究显示，注意力权重优化机制的引入使得该模型相较其他基线模型展现出更好的适用性和精准性，可为地表水水质预测研究提供新思路。