Novel federated learning-based approach for network attack detection
摘要
Network attack detection and prevention are critical aspects of ensuring cybersecurity in today’s interconnected world. Traditional centralized models for detecting malicious activity often face challenges like data privacy concerns and inefficiencies in handling distributed data sources. To address these challenges, this research leverages federated learning. Being a decentralized approach that offers advantageous benefits of collaborative model training across different devices while keeping sensitive data localized. Using the NF-UQ-NIDS-v2 benchmark dataset, we developed a robust machine-learning pipeline. This included thorough data preparation and applying advanced feature engineering techniques to reduce noise and enhance the dataset’s quality. We build several advanced machine learning approaches and hybrid models. Building on this, we proposed an innovative federated learning model designed to improve attack detection accuracy while preserving data privacy. With the help of the proposed federated learning architecture, the client model logistic regression (LR) achieved high-performance accuracy scores of 99% for multi-class network attack detection. Also, we quantify FL communication efficiency by measuring per-round latency, bandwidth usage, and the impact of compression techniques such as sparsification, 8-bit quantization, and top-