Research map / Graph learning / Graph self-supervised learning
Self-Supervised Learning of Graph Representations for Network Intrusion Detection
Read the paperCheck your idea against 108,967 papers
AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.
Proposes GraphIDS, a self-supervised network intrusion detector that jointly trains an E-GraphSAGE flow encoder and a Transformer masked autoencoder on benign flows, flagging flows by reconstruction error.
Problem
Network intrusion detection under limited labels and evolving attacks, where graph representation learning is usually decoupled from anomaly detection and pretext tasks often need negative samples.
Approach
Jointly train a GNN encoder (E-GraphSAGE, 1-hop) and a Transformer-based masked autoencoder end-to-end on benign flow embeddings, so the GNN embeddings are optimized directly for reconstruction-based anomaly scoring; attention masking is used as structured regularization.
Key results
- On NF-UNSW-NB15-v3, GraphIDS reaches PR-AUC 0.9998 and macro F1 0.9961 across seeds.
- On NF-CSE-CIC-IDS2018-v3, GraphIDS reaches PR-AUC 0.8819 and macro F1 0.9447, versus 0.2555 PR-AUC for the best Anomal-E variant.
- On NF-UNSW-NB15-v2, GraphIDS achieves PR-AUC 0.8116, above Anomal-E's 0.7489, while Anomal-E retains higher macro F1 on CSE-CIC-IDS2018-v2 (0.9410 vs 0.9431 for GraphIDS is comparable).
Measured effects reported in the paper
| What was changed | Where | Measured | Effect |
|---|---|---|---|
| add graph-based flow representations (ablation) | T-MAE on NF-CSE-CIC-IDS2018-v3 · NF-CSE-CIC-IDS2018-v3 | PR-AUC: 0.7398 -> 0.8819 | Helped |
| add graph-based flow representations (ablation) | T-MAE on NF-CSE-CIC-IDS2018-v2 · NF-CSE-CIC-IDS2018-v2 | PR-AUC: 0.7375 -> 0.9201 | Helped |
| replace the MLP autoencoder with the Transformer (ablation) | GraphIDS on NF-CSE-CIC-IDS2018-v3 · NF-CSE-CIC-IDS2018-v3 | Macro F1: 0.9223 -> 0.9447 | Helped |
| replace the MLP autoencoder with the Transformer (ablation) | GraphIDS on NF-CSE-CIC-IDS2018-v2 · NF-CSE-CIC-IDS2018-v2 | PR-AUC: 0.9310 -> 0.9201 | No clear effect |
| exclude timestamp features (ablation) | GraphIDS on NF-CSE-CIC-IDS2018-v3 · NF-CSE-CIC-IDS2018-v3 | PR-AUC: 0.7909 -> 0.8523 | Helped |
| exclude timestamp features (ablation) | GraphIDS on NF-UNSW-NB15-v3 · NF-UNSW-NB15-v3 | PR-AUC: 0.9991 -> 0.9989 | No clear effect |
| omit positional encoding (ablation) | GraphIDS on NF-CSE-CIC-IDS2018-v3 · NF-CSE-CIC-IDS2018-v3 | PR-AUC: 0.8572 -> 0.8661 | No clear effect |
| increase GNN dropout from 0.0 to 0.6 (ablation) | GraphIDS on NF-UNSW-NB15-v3 · NF-UNSW-NB15-v3 | PR-AUC: 0.9773 -> 0.9998 | Helped |
| increase attention mask ratio from 0.0 to 0.15 (ablation) | GraphIDS on NF-CSE-CIC-IDS2018-v3 · NF-CSE-CIC-IDS2018-v3 | PR-AUC: 0.8659 -> 0.8772 | Helped |
| increase GNN neighborhood from 1-hop to 2-hop (ablation) | GraphIDS on NF-UNSW-NB15-v2 · NF-UNSW-NB15-v2 | PR-AUC: 0.8116 -> 0.8367 | Helped |
| increase GNN neighborhood from 1-hop to 2-hop (ablation) | GraphIDS on NF-CSE-CIC-IDS2018-v3 · NF-CSE-CIC-IDS2018-v3 | PR-AUC: 0.8819 -> 0.8556 | Hurt |
Surprising findings
- A SimpleAE ablation (MLP instead of Transformer) matches or beats GraphIDS on several PR-AUC numbers, suggesting the autoencoding objective drives much of the gain.
- Positional encodings had little effect, and the T-MAE ablation without graph context performs comparably to GraphIDS on NF-UNSW-NB15-v3.
Limitations
- Performance may degrade under abrupt distribution shifts and needs online adaptation.
- Limited topological context in single-host monitoring scenarios.
- Infiltration attacks resembling benign traffic are often missed on CSE-CIC-IDS2018; NF-UNSW-NB15-v2 has more false positives.
Experimental setup
Datasets: NF-UNSW-NB15-v2 NF-UNSW-NB15-v3 NF-CSE-CIC-IDS2018-v2 NF-CSE-CIC-IDS2018-v3
Benchmarks: NF-UNSW-NB15-v2 NF-UNSW-NB15-v3 NF-CSE-CIC-IDS2018-v2 NF-CSE-CIC-IDS2018-v3
Metrics: PR-AUC macro F1
Baselines: Anomal-E CBLOF SAFE T-MAE SimpleAE HBOS Isolation Forest PCA
Builds on: GraphSAGE E-GraphSAGE BERT GraphMAE Anomal-E
Closest papers by meaning
- Self-Discriminative Modeling for Anomalous Graph Detection (ICML 2025 · similarity 0.85)
- UniFORM: Towards Unified Framework for Anomaly Detection on Graphs (AAAI 2025 · similarity 0.84)
- Stealthy Yet Effective: Distribution-Preserving Backdoor Attacks on Graph Classification (NeurIPS 2025 · similarity 0.84)
- Node Identifiers: Compact, Discrete Representations for Efficient Graph Learning (ICLR 2025 · similarity 0.84)
- Clean-Label Graph Backdoor Attack in the Node Classification Task (AAAI 2025 · similarity 0.84)
- DiffGAD: A Diffusion-based Unsupervised Graph Anomaly Detector (ICLR 2025 · similarity 0.83)
Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.
