本研究提出了一种用于水质异常检测的注意力门控液体神经网络,该方法通过连续时间建模和基于注意力的门控机制,实现了更高的准确性和可解释性,从而增强了环境监测的可靠性,并支持可持续的水资源管理。
方法文章
* These authors contributed equally
本研究提出了一种用于水质异常检测的注意力门控液体神经网络,该方法通过连续时间建模和基于注意力的门控机制,实现了更高的准确性和可解释性,从而增强了环境监测的可靠性,并支持可持续的水资源管理。
随着水质监测网络的迅速扩展,时间序列异常检测在保护水生环境和确保其可持续管理方面变得日益关键。然而,传统模型在实际应用中常常难以应对不规则的采样间隔、多变量相关性以及模型可解释性等问题。为解决这些挑战,本研究提出了一种注意力门控液态神经网络(Attention Gated-Liquid Neural Network, AG-LNN),该模型将液态神经网络(Liquid Neural Network, LNN)的动态建模能力与基于注意力机制的门控结构相结合。该模型引入了输入注意力门,以突出溶解氧(DO)和高锰酸盐指数(CODMn)等与异常相关的变量;同时设计了时间常数门,可自适应地调节模型的时间记忆特性。基于中国国家环境监测中心(CNEMC)在2019年至2024年间覆盖13个省份的监测数据,AG-LNN在多种基线模型上表现出更优性能,包括长短期记忆网络(LSTM)、时间卷积网络(TCN)、Transformer以及图神经网络(GNN)架构。该模型实现了0.95的精确率-召回率曲线下面积(PR-AUC)和0.90的F1分数,并在跨区域和跨时间评估中保持稳定。其轻量化版本AG-LNN-light在精度损失极小的情况下将参数量减少了62%,支持高效的边缘端部署。结果表明,基于注意力门控的连续时间建模为大规模水质异常检测提供了一种鲁棒且可解释的解决方案。
水质监测对于保护公众健康、维持水生生态系统以及确保符合环境法规至关重要。传统的水质模型,如土壤与水资源评估工具(SWAT)、机理型QUAL模型以及统计预测方法,往往难以捕捉受化学、生物、水文和气象因素共同影响的水体系统所具有的高度非线性、多变量以及时序不规则的特征1,2,3,4,5,6。因此,提升水质数据中异常检测的准确性仍是一项紧迫的挑战。
从实际应用角度来看,所提出的框架适用于多参数水质传感网络,用于测量氢离子浓度指数(pH)、溶解氧(DO)、总氮(TN)、总磷(TP)、高锰酸盐指数(CODMn)、浊度和电导率等变量,典型采样间隔为15至60分钟。该模型能够容忍不规则采样和缺失观测数据,因此适用于地表水和水库监测系统的实际部署。
机器学习在水质异常检测中的应用
近年来,机器学习(ML)和深度学习(DL)方法在水质相关应用中得到了广泛采用。例如,Wang 等人7提出了一种结合机制约束的带注意力机制的长短期记忆自编码器(LSTMA-AE),用于提高注水泵运行异常检测的准确性并减少误报,其性能显著优于插值法、随机森林或LSTM-AE方法。
同样,Zhao 等人8提出了一种融合物理信息神经网络的门控循环单元(GRU-PINN)模型,该模型将物理约束引入损失函数,在水质异常检测任务中显著提升了可解释性与F1分数。ElShafeiy 等人9开发了一种多变量卷积网络-长短期记忆网络(MCN-LSTM)架构,将多变量卷积网络与LSTM相结合,用于水质传感器数据的实时异常检测,在实地环境中表现出更强的检测能力。另一项进展是门控-液态神经网络(Gated-LNN)模型10,该模型将门控机制融入液态神经网络,可精确预测水质指数(WQI)并对水质进行分类,在印度数据集上实现了R2 ≈ 0.9995和99.74%的分类准确率。
时间序列异常检测
除了深度学习特定的应用外,时间序列异常检测在各个领域均取得了快速发展。时间序列异常检测已成为提升水质监测与早期预警能力的关键技术。传统的统计方法往往难以捕捉水质数据的非线性与动态特征,从而推动了深度学习模型的应用。例如,Zhang 等人11将经验模态分解(Empirical Mode Decomposition, EMD)与长短期记忆网络(LSTM)相结合,以更好地处理非平稳信号,显著提高了化学需氧量(COD)、五日生化需氧量(BOD₅)、总磷(TP)、总氮(TN)和氨氮(NH₃-N)的预测精度。在此基础上,Wang 等人12提出了一种基于LSTM并结合近似熵的波动分析方法,进一步提升了实时异常检测的性能。
近期进展强调了注意力机制和混合框架的应用。Arepalli 等人13提出了一种轻量级的空间共享注意力 LSTM 模型,用于水产养殖中的缺氧检测,准确率达到 99.8%;而 Zhang 等人14表明,空间和时间注意力机制显著提升了 CNN-LSTM 对溶解氧(DO)和氨氮(NH₃-N)的预测性能。类似地,Long 等人15将自适应噪声完全集合经验模态分解-变分模态分解(CEEMDAN-VMD)与注意力增强型 LSTM 相结合,在多个指标上实现了高达 0.99 的纳什-斯图克莱夫效率系数(NSE)。在应用层面,Xie 等人16展示了一种基于移动 LSTM 的序列到序列(Seq2Seq)系统,可用于流域范围内的业务化实时水质预测。
总体而言,这些研究凸显了基于LSTM的基线模型、结合注意力机制的混合模型以及系统级应用中的明确发展趋势。尽管准确性已大幅提升,但在可解释性、噪声条件下的鲁棒性,以及在不同监测网络间的泛化能力方面,仍然存在挑战。
图神经网络与时空模型在环境监测中的应用
图神经网络(GNN)和时空深度学习模型在环境监测中展现出巨大潜力,能够同时捕捉监测站点之间的空间相关性以及水质动态变化中的时间依赖性。例如,Wu 等人17提出了一种预训练增强的时空图神经网络(PT-STGNN),用于污水处理厂,通过结合基于 Transformer 的预训练和图结构学习,显著提升了对化学需氧量(COD)、氨氮(NH₃-N)、总磷(TP)、总氮(TN)、pH 值和流量的长期预测能力。类似地,Wan 等人18开发了一种时空特征图神经网络(STF-GNN),该模型融合了图卷积、门控循环单元(GRU)和注意力机制,在显著提高溶解氧(DO)和总氮(TN)预测精度的同时,表现出强大的跨流域泛化能力。
混合框架已超越纯数据驱动的设计,以增强泛化能力。Mu 等人19提出了时空图物理信息神经网络(ST-GPINN),该方法将水力学原理嵌入基于图神经网络(GNN)的模型中,用于供水系统。通过将图表示与物理信息约束相结合,ST-GPINN 在从小型到大型网络的扩展中实现了最先进的预测精度。这些研究共同凸显了时空图神经网络在环境监测中的日益重要作用,推动了预测准确性、可解释性和鲁棒性的提升,并为下一代智能水管理系统奠定了基础。
液态神经网络的优势与潜力
液体神经网络(LNNs),也称为液体时间常数网络,通过依赖输入且可学习的时间常数,独特地建模连续时间动态。Hasani 等人20 提出了 LNNs,并展示了其在时间序列任务中的表达能力、稳定性和高效性。这种受大脑启发的适应性使其成为建模水质等动态环境现象的有前景的候选模型,然而其与注意力机制的结合仍鲜有研究。
注意力机制,尤其是注意力门控,是聚焦相关特征并实现可解释性的强大工具。尽管在医学影像领域已得到广泛应用,注意力U-Net通过可学习的注意力门控为选择性特征优化奠定了基础。注意力机制还已被应用于基于传感器的异常检测框架中,例如LSTMA-AE7和GRU-PINN8,增强了对关键模式的敏感性,并提高了模型的透明度。
综上所述,这些观察结果表明需要一种混合架构,将连续时间适应性与可解释的注意力机制相结合。因此,我们提出了一种注意力门控液态神经网络(Attention-Gated Liquid Neural Network, AG-LNN),该网络将液态神经网络(Liquid Neural Networks, LNNs)的动态建模能力与注意力门机制相集成,以聚焦于显著的输入或时间片段。该架构旨在增强对传感器噪声和不规则采样的鲁棒性,提高对短时异常的敏感性,并通过注意力热图和自适应时间常数轨迹提供可解释性。AG-LNN 在真实世界的多变量水质数据集上进行了评估,并与 LSTM、Transformer、纯 LNN 以及传统机器学习基线模型进行了对比。定量分析(精确率、召回率、F1 分数、AUC)和定性分析(注意力可视化、时间常数动态)均证明了所提出模型的有效性与透明性。
访问受限。请登录或开始试用以查看此内容。
注意:本研究的整体工作流程见图1。
1. 数据采集
2. 变量选择与标记
(1)
(2)
为变量 i 在季节 s 中的下限和上限分位数阈值(第1和第99百分位)。若数据点超出这些界限,则标记为异常,如公式(3)所示。
(3)
(4)
(5)
(6)
(7)3. 数据质量控制与预处理
(8)
(9)
(10)
(11)
使用第一个可用的非缺失观测值,并设定初始协方差 P0|0 到
若缺失时间间隔超过24小时(96个记录),则应删除整个片段,以防止长期外推导致误差累积。
(12)
(13)4. 注意力门控液体神经网络(AG-LNN)
注意:所提出的注意力门控液态神经网络(AG-LNN)的整体架构如图2所示。该模型集成了多源水质输入、基于注意力的门控机制、具有自适应时间常数的液态神经网络(LNN),以及具备可解释性输出的读出解码阶段。该设计使网络能够动态捕捉复杂的时序依赖关系,同时对噪声和非均匀采样保持鲁棒性。
(14)
(15)
(16)
表示在时间步 τ 上 C 个通道的平均注意力权重;β 是一个敏感性系数,用于决定注意力对时间响应性的影响力强弱。
较高时,λτ 减小,促使网络加快响应速度,从而更好地捕捉短暂的异常。对 λτ 施加边界约束以稳定训练过程。
(17)
(18)
(20)
(21)
(22)
(23)
(24)5. 训练程序
访问受限。请登录或开始试用以查看此内容。
总体性能比较
表2展示了不同基线模型与所提出的AG-LNN在水质异常检测任务中的对比结果。在传统的无监督基线方法中,孤立森林22和一类支持向量机23仅取得了中等水平的精确率和召回率,反映出它们在捕捉多源水质数据中固有的时间依赖性方面能力有限。局部离群因子24通过利用局部密度结构表现稍好,但其性能与深度学习模型相比仍不理想。
对于基于神经网络的方法,LSTM25 和 Bi-GRU26 由于其循环结构,在召回率方面表现出提升,但在处理长序列和不规则采样时仍存在困难。TCN27 通过有效建模短期波动,在精确率方面取得了具有竞争力的性能。基于 Transformer 的架构在长距离依赖...
访问受限。请登录或开始试用以查看此内容。
本研究提出了一种注意力门控液体神经网络(Attention-Gated Liquid Neural Network, AG-LNN),这是一种用于水质异常检测的新型架构,结合了连续时间液体动力学与基于注意力的门控机制。AG-LNN中的液体神经组件灵感来源于液体时间常数网络(Liquid Time-constant Networks, LTCs),后者通过可学习的、依赖输入的时间常数对时间序列进行建模20。然而,AG-LNN在标准LTC的基础上进一步引入了两个额外的门控机制:输入注意力门,用于突出与异常相关的变量(如溶解氧DO和高锰酸盐指数CODMn);时间常数调制门,可根据上下文动态加速或减缓液体动力学过程。这种双重门控机制确保AG-LNN在继承LTC表达能力强的连续时间特性的同时,提升了模型的可解释性与对异常的敏感性。
该框架在多个方面推动了科学领域的发展。首先,AG-LNN 解决了大规模环境监测网络中常见的不规则采样多变量传感器数据的处理难题。与需要固定时间间隔的离散时间RNN不同,AG-L...
访问受限。请登录或开始试用以查看此内容。
本研究由2024年广东省高校特色创新项目“基于无线传感器网络的水质监测与预警系统”(项目编号:2024KTSCX304)、广州南洋理工职业学院2024年校级科研项目“基于物联网的水箱管理系统”(项目编号:NY-2024KYZD-01)、2022年广东省重点领域专项(新一代电子信息)“基于HarmonyOS的水产养殖在线预测、预警与联动防控系统”(项目编号:2022ZDZX1081)以及2021年广东省高职院校高水平专业群建设项目“大数据技术专业群”(项目编号:GSPZYQ2020089)资助。
访问受限。请登录或开始试用以查看此内容。
| 姓名 | 公司 | 目录编号 | 评论 |
|---|---|---|---|
| 256 GB DDR4 内存 | Samsung | M393A4K40DB3 | 用于处理大规模多变量时间序列数据集的高容量内存 |
| CUDA 11.6 | NVIDIA | N/A | PyTorch 的 GPU 加速工具包 |
| Intel Xeon Gold 6330 CPU (2.0GHz, 28 核) | Intel | BX80708-6330 | 用作模型训练的主计算服务器 |
| Matplotlib 3.4, Seaborn 0.11 | 开源 | N/A | 实验结果的可视化 |
| NumPy 1.21, Pandas 1.3 | 开源 | N/A | 数据预处理与特征工程 |
| NVIDIA A100 GPU (40GB) | NVIDIA | 900-21001-0000-001 | 支持 CUDA 的 AG-LNN 加速训练 |
| Python 3.9 | Python 软件基金会 | N/A | 实现所使用的主要编程语言 |
| PyTorch 1.12 | Meta AI | N/A | 用于构建 AG-LNN 的深度学习框架 |
| Scikit-learn 0.24 | 开源 | N/A | 评估指标与基线模型 |
| Ubuntu 20.04 LTS 操作系统 | Canonical | N/A | 计算环境的操作系统 |
访问受限。请登录或开始试用以查看此内容。
申请许可以重复使用本 JoVE 文章的文本或图表
申请许可