Neurosymbolic Learning for Advanced Persistent Threat Detection under Extreme Class Imbalance
Authors: Quhura Fathima, Neda Moghim, Mostafa Taghizade Firouzjaee, Christo K. Thomas, Ross Gore, Walid Saad
Organizations: Center for Secure & Intelligent Critical Systems and School of Cybersecurity, Old Dominion University, VA, USA · Department of Computer Engineering, University of Isfahan, Iran · Faculty of Engineering Modern Technologies, Amol University of Special Modern Technologies, Iran · Department of Electrical and Computer Engineering, Worcester Polytechnic Institute, MA, USA · Department of Electrical and Computer Engineering, Virginia Tech, VA, USA
The growing deployment of Internet of Things (IoT) devices in smart cities and industrial environments increases vulnerability to stealthy, multi-stage advanced persistent threats (APTs) that exploit wireless communication. Detection is challenging due to severe class imbalance in network traffic, which limits the effectiveness of traditional deep learning approaches and their lack of explainability in classification decisions. To address these challenges, this paper proposes a neurosymbolic architecture that integrates an optimized BERT model with logic tensor networks (LTN) for explainable APT detection in wireless IoT networks. The proposed method addresses the challenges of mobile IoT environments through efficient feature encoding that transforms network flow data into BERT-compatible sequences while preserving temporal dependencies critical for APT stage identification. Severe class imbalance is mitigated using focal loss, hierarchical classification that separates normal traffic detection from attack categorization, and adaptive sampling strategies. Evaluation on the SCVIC-APT2021 dataset demonstrates an operationally viable binary classification F1 score of 95.27% with a false positive rate of 0.14%, and a 76.75% macro F1 score for multi-class attack categorization. Furthermore, a novel explainability analysis statistically validates the importance of distinct network features. These results demonstrate that neurosymbolic learning enables high-performance, interpretable, and operationally viable APT detection for IoT network monitoring architectures.
Figures & tables
Class
Samples
Percentage
Normal Traffic
254,836
98.35%
Pivoting
2,122
0.82%
Reconnaissance
833
0.32%
Lateral Movement
729
0.28%
Data Exfiltration
527
0.20%
Initial Compromise
73
0.03%
TABLE I: Class distribution in SCVIC-APT2021 dataset
Fig. 1: System architecture showing parallel BERT and LTN pipelines, hierarchical classification heads, and multi-objective training integration.
Department of Computer Science, Birla Institute of Technology and Science, Pilani, Dubai Campus, Dubai International Academic City, Dubai, United Arab Emirates