Federated Learning for the Detection of Attacks on IoT Environment
摘要
The rapid expansion of Internet of Things (IoT) devices has significantly increased the attack surface, necessitating the development of robust and privacy-preserving intrusion detection systems. This research delves into the application of Federated Learning (FL) to detect network-based attacks in IoT environments, utilizing the NF-ToN-IoT dataset. We compare a centralized machine learning model with a federated counterpart, employing the Federated Averaging (FedAvg) algorithm. Both models employ a Multi-Layer Perceptron (MLP) architecture trained on NetFlow features for binary classification of benign versus malicious traffic. Our experimental results reveal that the federated model achieves comparable and even slightly superior performance across various metrics, including precision, recall, and F1-score, while preserving data privacy by decentralizing the training data. These findings underscore the potential of FL as a viable alternative to traditional intrusion detection systems in real-world, privacy-sensitive IoT scenarios.