arXiv:2509.16625cs.LGcs.CR2025-09NeurIPS被引 12

用自监督学习统一图表示与异常检测,提升网络入侵识别准确率。

Self-Supervised Learning of Graph Representations for Network Intrusion Detection

  • 通过掩码自编码器学习正常通信的局部图表示
  • 重建误差高的流量被标记为潜在攻击,最高达99.98% PR-AUC
  • 无需标注数据,适合实时入侵检测场景

网络流量入侵检测面临标注数据少、攻击模式持续演化等挑战。现有图神经网络方法常将表征学习与异常检测分离,限制了嵌入对攻击识别的效用。本文提出GraphIDS,一种自监督入侵检测模型,通过掩码自编码器统一两个阶段:利用归纳式图神经网络结合局部拓扑上下文嵌入每条流,捕捉典型网络行为;再通过基于Transformer的编码器-解码器重构这些嵌入,借助自注意力机制隐式学习全局共现模式,无需显式位置信息。推理时,重建误差显著偏高的流被标记为潜在入侵。该端到端框架使嵌入直接优化于下游任务,提升恶意流量识别能力。在多个NetFlow基准上表现优异,最高达到99.98%的PR-AUC和99.61%的宏平均F1分数。

原文摘要 · Abstract (English)

Detecting intrusions in network traffic is a challenging task, particularly under limited supervision and constantly evolving attack patterns. While recent works have leveraged graph neural networks for network intrusion detection, they often decouple representation learning from anomaly detection, limiting the utility of the embeddings for identifying attacks. We propose GraphIDS, a self-supervised intrusion detection model that unifies these two stages by learning local graph representations of normal communication patterns through a masked autoencoder. An inductive graph neural network embeds each flow with its local topological context to capture typical network behavior, while a Transformer-based encoder-decoder reconstructs these embeddings, implicitly learning global co-occurrence patterns via self-attention without requiring explicit positional information. During inference, flows with unusually high reconstruction errors are flagged as potential intrusions. This end-to-end framework ensures that embeddings are directly optimized for the downstream task, facilitating the recognition of malicious traffic. On diverse NetFlow benchmarks, GraphIDS achieves strong performance, reaching up to 99.98% PR-AUC and 99.61% macro F1-score.

图神经网络入侵检测自监督学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。