关键词:
湘江流域
水质预测
机器学习
人工神经网络
模型性能
摘要:
利用现有水质数据对未来水质变化进行预测是实现区域规划与流域管理的有效工具.基于2022年4月至2022年5月湘江流域衡阳段水质监测数据,构建了反向传播神经网络(BPNN)与卷积神经网络(CNN)水质指标预测模型,并对高锰酸盐指数的预测结果进行了比较分析.数据显示,BPNN模型的预测值与真实水质情况基本吻合,但在拟合过程中出现了过拟合现象,利用粒子群算法(PSO)改进BPNN模型的参数选择方式,能够避免这一现象.而CNN模型拥有更复杂的结构、更科学的拟合方法,从而避免了模型在拟合过程中陷入局部极值,同时提高了模型预测结果的准确性.以不同模型的均方根误差(RMSE)、决定系数(R2)与平均绝对误差(MAE)作为评价参数,与传统的BPNN模型相比,PSO-BPNN模型中测试集的RMSE从0.2782 mg·L^(-1)降低到0.2109 mg·L^(-1),MAE从0.2223 mg·L^(-1)降低到0.1537 mg·L^(-1),R^(2)从0.8640提高到0.9218,PSO-BPNN模型拥有更加稳定的拟合效果.CNN模型测试集的RMSE、MAE和R^(2)分别为0.1220 mg·L^(-1)、0.0927 mg·L^(-1)和0.9705,显示CNN模型预测效果更好.