NGMD: next generation malware detection in federated server with deep neural network model for autonomous networks

Himanshi Babbar; Shalli Rani; Wadii Boulila

doi:10.1038/s41598-024-61298-7

NGMD: next generation malware detection in federated server with deep neural network model for autonomous networks

Sci Rep. 2024 May 13;14(1):10898. doi: 10.1038/s41598-024-61298-7.

Authors

Himanshi Babbar¹, Shalli Rani², Wadii Boulila^{3

4}

Affiliations

¹ Chitkara University Institute of Engineering and Technology, Chitkara University, Punjab, India.
² Chitkara University Institute of Engineering and Technology, Chitkara University, Punjab, India. shallir79@gmail.com.
³ Robotics and Internet-of-Things Laboratory, Prince Sultan University, 12435, Riyadh, Saudi Arabia.
⁴ RIADI Laboratory, National School of Computer Sciences, University of Manouba, 2010, Manouba, Tunisia.

Abstract

Distributed denial-of-service (DDoS) attacks persistently proliferate, impacting individuals and Internet Service Providers (ISPs). Deep learning (DL) models are paving the way to address these challenges and the dynamic nature of potential threats. Traditional detection systems, relying on signature-based techniques, are susceptible to next-generation malware. Integrating DL approaches in cloud-edge/federated servers enhances the resilience of these systems. In the Internet of Things (IoT) and autonomous networks, DL, particularly federated learning, has gained prominence for attack detection. Unlike conventional models (centralized and localized DL), federated learning does not require access to users' private data for attack detection. This approach is gaining much interest in academia and industry due to its deployment on local and global cloud-edge models. Recent advancements in DL enable training a quality cloud-edge model across various users (collaborators) without exchanging personal information. Federated learning, emphasizing privacy preservation at the cloud-edge terminal, holds significant potential for facilitating privacy-aware learning among collaborators. This paper addresses: (1) The deployment of an optimized deep neural network for network traffic classification. (2) The coordination of federated server model parameters with training across devices in IoT domains. A federated flowchart is proposed for training and aggregating local model updates. (3) The generation of a global model at the cloud-edge terminal after multiple rounds between domains and servers. (4) Experimental validation on the BoT-IoT dataset demonstrates that the federated learning model can reliably detect attacks with efficient classification, privacy, and confidentiality. Additionally, it requires minimal memory space for storing training data, resulting in minimal network delay. Consequently, the proposed framework outperforms both centralized and localized DL models, achieving superior performance.