用极少攻击样本训练图神经网络,精准识别真实攻击而非误报。
Few Edges Are Enough: Few-Shot Network Attack Detection with Graph Neural Networks
- 结合对比与重建的自监督学习,提升少样本攻击检测能力。
- 每类攻击仅需1条标记样本,性能媲美甚至超越部分有监督方法。
- 适合缺乏标注数据的网络安全场景,尤其适用于新型攻击快速响应。
使用图神经网络(GNN)检测网络攻击近年来取得显著进展。但现有主流模型多依赖大量标注样本,而实际场景中标签数据难以获取。为此,无监督学习和自监督学习(SSL)成为减少标签依赖的可行方向,但往往导致更多误报,难以区分异常与真实攻击。本文提出少边足够(FEAE)框架,基于自监督学习与少样本学习(FSL)构建GNN模型,有效区分虚假异常与真实攻击。通过融合对比式与重构式自监督目标,最大化利用少量已知攻击样本(即攻击边)。实验表明,仅需每类攻击1条标记事件,即可在两个主流网络数据集上实现优异性能。相比自监督基线,FEAE表现更优;在某数据集上甚至超越部分有监督方法。
原文摘要 · Abstract (English)
Detecting cyberattacks using Graph Neural Networks (GNNs) has seen promising results recently. Most of the state-of-the-art models that leverage these techniques require labeled examples, hard to obtain in many real-world scenarios. To address this issue, unsupervised learning and Self-Supervised Learning (SSL) have emerged as interesting approaches to reduce the dependency on labeled data. Nonetheless, these methods tend to yield more anomalous detection algorithms rather than effective attack detection systems. This paper introduces Few Edges Are Enough (FEAE), a GNN-based architecture trained with SSL and Few-Shot Learning (FSL) to better distinguish between false positive anomalies and actual attacks. To maximize the potential of few-shot examples, our model employs a hybrid self-supervised objective that combines the advantages of contrastive-based and reconstruction-based SSL. By leveraging only a minimal number of labeled attack events, represented as attack edges, FEAE achieves competitive performance on two well-known network datasets compared to both supervised and unsupervised methods. Remarkably, our experimental results unveil that employing only 1 malicious event for each attack type in the dataset is sufficient to achieve substantial improvements. FEAE not only outperforms self-supervised GNN baselines but also surpasses some supervised approaches on one of the datasets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。