基于时空对比学习的动态图异常检测
Spatio-Temporal Contrastive Learning for Dynamic Graph Anomaly Detection
-
摘要: 为解决现有动态图异常检测方法难以准确建模不同时间尺度下异常模式的缺陷,提出一种基于时空对比学习的动态图异常检测(spatio-temporal contrastive learning for dynamic graph anomaly detection,STCL-DGAD)方法。该方法包括时空特征提取、时空对比学习和动态异常检测3个模块。首先,时空特征提取模块通过时间编码和基于Transformer的建模方法,提取含有时间动态性和全局结构信息的节点特征;其次,在时空对比学习模块中,设计一个全局-局部双粒度对比机制,其中全局对比通过指数衰减权重聚合历史交互信息,进而构建增强图以建模长期时空依赖,而局部对比则通过相邻时间片特征对齐来捕捉短时异常突变;最后,动态异常检测模块利用正则化策略增强方法的鲁棒性。实验结果表明,STCL-DGAD在4个真实数据集上的异常检测性能优于现有方法。Abstract: To address the limitation of existing dynamic graph anomaly detection methods in accurately modeling anomaly patterns at different time scales, a method of spatio-temporal contrastive learning for dynamic graph anomaly detection (STCL-DGAD) is proposed. The proposed method consists of three modules: spatio-temporal feature extraction, spatio-temporal contrastive learning, and dynamic anomaly detection. First, the spatio-temporal feature extraction module employed time encoding and a Transformer-based architecture to capture node features incorporating temporal dynamics and global structural information. Second, in the spatial-temporal contrastive learning module, a global-local dualgranularity contrastive mechanism was designed. The global contrastive part aggregated historical interaction information throuhg exponentially decaying weights, and further constructed an augmented graph to model long-term spatial-temporal dependencies. The local contrastive part captured short-term abnormal mutations by aligning features of adjacent time snapshots. Finally, the dynamic anomaly detection module enhanced the model robustness via a regularization strategy. Experimental results on four real-world datasets demonstrate that STCL-DGAD outperforms existing methods in anomaly detection tasks.
下载: