keyboard_arrow_up
Edge-Level Graph Neural Network Architectures for Network Intrusion Detection: Advancing Beyond Standard Edge-Level Classification

Authors

Mehmet Baris Yaman , Foundational Technology, Turkey

Abstract

Network intrusion detection in Internet of Things (IoT) environments presents unique challenges due to the scale, heterogeneity and dynamic nature of device interactions. Although graph neural networks (GNNs) have demonstrated promising results by modeling network topology, standard edge-level classification approaches leave substantial room for improvement. This paper introduces three GNN architectures that advance the state-of-the-art through distinct mechanisms: Prototype-GNN, which employs distance-based classification with learnable prototypes to capture diverse attack patterns; ContrastiveGNN, which optimizes embedding geometry through supervised contrastive learning; and GSL-GNN, which learns optimal graph structure adaptively from node features. Evaluated on the TON IoT dataset, our approaches achieve 94.24%, 94.71% and 96.66% accuracy, respectively, representing improvements of +2.37, +2.84, and +4.79 percentage points over the baseline EdgeLevelGCN (91.87%). GSL-GNN achieves near-perfect discrimination with 99.70% ROC-AUC and only 1.5% false positive rate. Our mechanisms generalize beyond network security to any edge classification task on graphs, convolutional neural networks, and knowledge graphs.

Keywords

Graph Neural Networks, Network Intrusion Detection, IoT, Security, Prototype Learning, Contrastive Learning, Graph Structure Learning, Deep Learning

Full Text  Volume 15, Number 25