提出新方法检测无线传感器网络异常,融合对比学习与少样本学习。
A New Spatiotemporal Correlation Anomaly Detection Method that Integrates Contrastive Learning and Few-Shot Learning in Wireless Sensor Networks
- 用增强的RetNet和图注意力网络提取时空关联特征。
- 在真实数据集上达90.97%的F1分数,优于现有方法。
- 适合标签少、异常样本稀疏的传感器网络场景。
无线传感器网络(WSN)数据异常检测可为评估其可靠性与稳定性提供关键证据。现有方法常面临时空相关特征提取不足、标签缺失、异常样本少及样本分布不均等问题。为此,提出一种兼顾模型结构与两阶段训练策略的时空相关性检测模型MTAD-RD。模型结构上,采用增强交叉保留(CR)模块的RetNet,结合多粒度特征融合与图注意力网络,有效提取节点间关联信息及时间序列全局特征,且具备串行推理特性,显著降低推理开销。训练方面,设计两阶段策略:第一阶段利用带图结构信息的时间序列构建对比学习代理任务,通过无监督对比学习从无标签数据中学习可迁移特征;第二阶段引入基于缓存的采样器,将样本分为少样本与对比学习数据,并设计联合损失函数,协同训练双图判别网络以缓解样本不平衡问题。在真实公开数据集上的实验表明,所提MTAD-RD方法在异常检测任务中取得90.97%的F1分数,优于现有监督型检测方法。
原文摘要 · Abstract (English)
Detecting anomalies in the data collected by WSNs can provide crucial evidence for assessing the reliability and stability of WSNs. Existing methods for WSN anomaly detection often face challenges such as the limited extraction of spatiotemporal correlation features, the absence of sample labels, few anomaly samples, and an imbalanced sample distribution. To address these issues, a spatiotemporal correlation detection model (MTAD-RD) considering both model architecture and a two-stage training strategy perspective is proposed. In terms of model structure design, the proposed MTAD-RD backbone network includes a retentive network (RetNet) enhanced by a cross-retention (CR) module, a multigranular feature fusion module, and a graph attention network module to extract internode correlation information. This proposed model can integrate the intermodal correlation features and spatial features of WSN neighbor nodes while extracting global information from time series data. Moreover, its serialized inference characteristic can remarkably reduce inference overhead. For model training, a two-stage training approach was designed. First, a contrastive learning proxy task was designed for time series data with graph structure information in WSNs, enabling the backbone network to learn transferable features from unlabeled data using unsupervised contrastive learning methods, thereby addressing the issue of missing sample labels in the dataset. Then, a caching-based sample sampler was designed to divide samples into few-shot and contrastive learning data. A specific joint loss function was developed to jointly train the dual-graph discriminator network to address the problem of sample imbalance effectively. In experiments carried out on real public datasets, the designed MTAD-RD anomaly detection method achieved an F1 score of 90.97%, outperforming existing supervised WSN anomaly detection methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。