Self-Supervised Learning of Graph Representations for Network Intrusion Detection
By: Lorenzo Guerra , Thomas Chapuis , Guillaume Duc and more
Potential Business Impact:
Finds sneaky computer network attacks automatically.
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 up to 99.98% PR-AUC and 99.61% macro F1-score, outperforming baselines by 5-25 percentage points.
Similar Papers
Self-Supervised Transformer-based Contrastive Learning for Intrusion Detection Systems
Cryptography and Security
Finds computer attacks that are new and hidden.
AutoGraphAD: A novel approach using Variational Graph Autoencoders for anomalous network flow detection
Cryptography and Security
Finds computer network dangers without needing labels.
Are We There Yet? Unraveling the State-of-the-Art Graph Network Intrusion Detection Systems
Cryptography and Security
Makes computer security systems more trustworthy.