IDS for Industrial Applications: A Federated Learning Approach with Active Personalization
Creators
- 1. Department of Electrical and Computer Engineering, University of Western Macedonia, 501 31 Kozani, Greece
- 2. Department of Networks and Digital Media, Kingston University, London KT1 1LQ, UK
- 3. Department of Computer Science, Kavala Campus, International Hellenic University, 654 04 Kavala, Greece
- 4. Sidroco Holdings Ltd., Nicosia 1077, Cyprus
Description
Internet of Things (IoT) is a concept adopted in nearly every aspect of human life, leading to an explosive utilization of intelligent devices. Notably, such solutions are especially integrated in the industrial sector, to allow the remote monitoring and control of critical infrastructure. Such global integration of IoT solutions has led to an expanded attack surface against IoT-enabled infrastructures. Artificial intelligence and machine learning have demonstrated their ability to resolve issues that would have been impossible or difficult to address otherwise; thus, such solutions are closely associated with securing IoT. Classical collaborative and distributed machine learning approaches are known to compromise sensitive information. In our paper, we demonstrate the creation of a network flow-based Intrusion Detection System (IDS) aiming to protecting critical infrastructures, stemming from the pairing of two machine learning techniques, namely, federated learning and active learning. The former is utilized for privately training models in federation, while the latter is a semi-supervised approach applied for global model adaptation to each of the participant’s traffic. Experimental results indicate that global models perform significantly better for each participant, when locally personalized with just a few active learning queries. Specifically, we demonstrate how the accuracy increase can reach 7.07% in only 10 queries.
Files
sensors-21-06743-v2.pdf
Files
(676.2 kB)
Name | Size | Download all |
---|---|---|
md5:356651ded5fea3e6db84298ab28e7524
|
676.2 kB | Preview Download |