[发明专利]一种基于多任务学习及深度学习模型的水质预测方法

说明书

本发明提供了一种基于多任务学习及深度学习模型的水质预测方法，包括以下步骤：获取输入数据，输入数据包括水质数据和气象数据；将输入数据输入至预测模型中；获取输入数据的时间序列；根据时间序列和输入数据获取局部依赖特征和长时间依赖特征；根据局部依赖特征和长时间依赖特征获得水质指标预测结果。本发明所述的一种基于多任务学习及深度学习模型的水质预测方法能够适用于分析预测非线性的、复杂的水质数据，提高水质预测精度。

技术领域

本发明属于环境质量预测技术领域，具体涉及一种基于多任务学习及深度学习模型的水质预测方法。

背景技术

水是我们日常生活中不可或缺的资源，也是生态系统中赖以生存的物质。随着工业、经济的发展，水体污染愈发严重，水资源愈加短缺，因此保护水资源迫在眉睫。在水资源的保护过程中，需要使用水质预测方法预测水资源的健康情况，便于工作人员及时采取预防污染措施，避免水资源大面积污染。

目前水质预测方法大致分为两类。第一类是传统的统计学习方法，常用的是差分整合移动平均自回归模型(ARIMA)。ARIMA可以有效捕捉特征间的线性关系，但只能适用于平稳序列的预测，无法适用于当下复杂的非线性水质数据。第二类常用的水质预测方法是使用BP神经网络模型对水质进行预测。BP神经网络由于其能拟合任意连续函数及具有较强非线性映射能力的特性被成功用于水质预测领域。然而，评价水质健康的数据较多且复杂，BP神经网络较“浅”，无法适用于高度复杂的数据集。

基于上述水质预测存在的问题，亟需一种基于多任务学习的水质预测方法，能够解决现有的水质预测方法不能适用于当下非线性的、复杂的水质数据，导致水质预测结果不精确的问题。

发明内容

本发明旨在至少解决现有技术中存在的技术问题，提供一种基于多任务学习及深度学习模型的水质预测方法，适用于分析预测非线性的、复杂的水质数据，提高水质预测精度。

为了实现本发明的上述目的，根据本发明的第一个方面，本发明提供了一种基于多任务学习及深度学习模型的水质预测方法，包括以下步骤：获取输入数据，输入数据包括水质数据和气象数据；将输入数据输入至预测模型中；获取输入数据的时间序列；根据时间序列和输入数据获取局部依赖特征和长时间依赖特征；根据局部依赖特征和长时间依赖特征获得水质指标预测结果。

进一步地，预测模型的训练过程如下：创建初始模型；获取训练数据放入初始模型中，训练数据包括水质数据和气象数据，根据训练数据建立目标序列和时间序列，对训练数据进行预处理，通过滑动窗口将时间序列和目标序列作为初始模型的输入；初始模型提取时间序列和目标序列中的特征数据，对特征数据进行学习，获得特征数据之间的局部依赖特征和长时间依赖特征，向局部依赖特征和长时间依赖特征赋予动态权重获得初始模型的输出结果；基于多任务学习模型获取时间序列和目标序列中的特征数据的相关性，根据提取到的特征数据和多个任务间的相关性对初始模型的损失函数进行更新，直至初始模型的损失函数最小化，初始模型结束训练生成预测模型。

进一步地，向局部依赖特征和长时间依赖特征赋予动态权重的步骤中还包括加权求和步骤；加权求和步骤具体为：使用softmax函数确保动态权重的和为1。

进一步地，初始模型的损失函数如下：其中，L(W,σ₁,σ₂)表示初始模型的损失函数，L₁、L₂、L₃分别表示任务1的损失函数、任务2的损失函数、任务3的损失函数；W表示初始模型的模型参数；σ₁、σ₂、σ₃分别表示任务1的损失权重、任务2的损失权重、任务3的损失权重；L₁(W)、L₂(W)、L₃(W)分别表示任务1的损失值、任务2的损失值、任务3的损失值。

进一步地，初始模型包括卷积神经网络层、长短期记忆网络层、注意力机制层和多任务学习模型。