用节点可控性增强图神经网络,提升小样本异常检测能力
Robust Anomaly Detection with Graph Neural Networks using Controllability
- 将节点平均可控性作为边权重或独热编码特征
- 在真实与合成数据集上优于六种主流基线方法
- 适合处理数据稀疏且异常样本极少的场景
复杂领域中的异常检测面临标签数据不足和正负样本严重不平衡的挑战。基于图的机器学习模型能融合属性与关系信息,揭示复杂模式,但异常样本稀缺加剧了学习难度。本文提出,引入节点影响力度量——平均可控性,可显著提升异常检测性能。为此,我们设计两种新方法:(1) 将平均可控性作为边权重;(2) 将其编码为独热边特征向量。在六个真实与合成网络上,与六种先进基线对比,所提方法在异常识别上表现更优,验证了可控性度量对图学习模型的增益作用。本研究强调,在稀疏且不平衡的数据集中,融合平均可控性可有效应对异常检测难题。
原文摘要 · Abstract (English)
Anomaly detection in complex domains poses significant challenges due to the need for extensive labeled data and the inherently imbalanced nature of anomalous versus benign samples. Graph-based machine learning models have emerged as a promising solution that combines attribute and relational data to uncover intricate patterns. However, the scarcity of anomalous data exacerbates the challenge, which requires innovative strategies to enhance model learning with limited information. In this paper, we hypothesize that the incorporation of the influence of the nodes, quantified through average controllability, can significantly improve the performance of anomaly detection. We propose two novel approaches to integrate average controllability into graph-based frameworks: (1) using average controllability as an edge weight and (2) encoding it as a one-hot edge attribute vector. Through rigorous evaluation on real-world and synthetic networks with six state-of-the-art baselines, our proposed methods demonstrate improved performance in identifying anomalies, highlighting the critical role of controllability measures in enhancing the performance of graph machine learning models. This work underscores the potential of integrating average controllability as additional metrics to address the challenges of anomaly detection in sparse and imbalanced datasets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。