Optimised_Lightweight_Federated_Learning__Revised_Submission_.pdf (391.88 kB)
Download file

Optimized Lightweight Federated Learning for Botnet Detection in Smart Critical Infrastructure

Download (391.88 kB)
posted on 2023-07-10, 07:23 authored by Segun PopoolaSegun Popoola, Ruth Ande, Aderemi Atayero, Mohammad Hammoudeh, Guan GuiGuan Gui, Bamidele Adebisi

In this paper, we propose an optimized lightweight Federated Deep Learning (FDL) method for botnet attack detection in smart critical infrastructure. First, an optimization method is developed to determine the most appropriate combination of model hyperparameters for local Deep Learning (DL) at the edge nodes. Then, an oversampling algorithm is combined with the optimal DL model to improve the classification performance when the training data is highly imbalanced, without a significant increase in the overall computation time. Furthermore, a feature dimensionality reduction method is used to reduce the amount of memory space required to store the network traffic data at the edge nodes.


Email Address of Submitting Author

Submitting Author's Institution

Manchester Metropolitan University

Submitting Author's Country

  • United Kingdom