用Transformer模型提升网络异常检测,效率与精度双突破。
CyberGFM: Graph Foundation Models for Lateral Movement Detection in Enterprise Networks
- 将随机游走类比为语句,用Transformer建模网络行为
- 在3个数据集上平均精度提升最高达2倍
- 无需额外标注,适合企业级网络实时监控
将网络表示为图,并使用正常连接训练链接预测模型,是基于异常的入侵检测的有效方法。现有方法在时间图神经网络和基于随机游走的skip-gram方法上表现优异,但随机游走方法难以融入丰富的边信息,而GNN方法需要大量内存训练。本文提出将随机游走类比为语料库中句子的原始思想,扩展至现代Transformer基础模型。利用具备GPU优化的语言模型,可快速训练图基础模型,以预测网络计算机间随机游走中的缺失片段。该模型随后微调用于链接预测,作为网络异常检测器。此新方法结合了随机游走方法的高效性与深度学习的丰富语义表征能力。我们提出的系统CyberGFM在三个广泛使用的网络异常检测数据集上达到当前最佳性能,平均精度最高提升2倍。实验表明,在参数量相同的情况下,CyberGFM在无监督链接预测中优于所有先前工作,且训练效率与最优前代方法相当或更优。
原文摘要 · Abstract (English)
Representing networks as a graph and training a link prediction model using benign connections is an effective method of anomaly-based intrusion detection. Existing works using this technique have shown great success using temporal graph neural networks and skip-gram-based approaches on random walks. However, random walk-based approaches are unable to incorporate rich edge data, while the GNN-based approaches require large amounts of memory to train. In this work, we propose extending the original insight from random walk-based skip-grams--that random walks through a graph are analogous to sentences in a corpus--to the more modern transformer-based foundation models. Using language models that take advantage of GPU optimizations, we can quickly train a graph foundation model to predict missing tokens in random walks through a network of computers. The graph foundation model is then finetuned for link prediction and used as a network anomaly detector. This new approach allows us to combine the efficiency of random walk-based methods and the rich semantic representation of deep learning methods. This system, which we call CyberGFM, achieved state-of-the-art results on three widely used network anomaly detection datasets, delivering a up to 2$\times$ improvement in average precision. We found that CyberGFM outperforms all prior works in unsupervised link prediction for network anomaly detection, using the same number of parameters, and with equal or better efficiency than the previous best approaches.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。